Analise uploads de arquivos com uma função de firewall do AI Inference

Crie uma função que envia um upload a um modelo através de AI Inference e execute-a em um firewall para que um upload malicioso nunca chegue à origem.

Última atualização

Ver como Markdown Configurar agente

Você pode analisar um upload com um modelo e negar um arquivo malicioso, a partir de Azion Console. A função é executada em Firewall e chama o modelo através de AI Inference. Para conhecer o design que esta página implementa, consulte Inspecionar uploads de arquivos em busca de conteúdo malicioso.

Pré-requisitos

  • Uma conta Azion. Para criar uma, consulte Como criar uma conta na Azion.
  • Um firewall com o módulo Functions ativado. Para saber o que uma função faz em um firewall, consulte Functions no Firewall.
  • Um domínio associado a esse firewall, para que o tráfego de upload chegue até ele.
  • As rotas de upload a proteger, como /upload ou /documents/upload.

Crie a função de firewall

Uma função é executada em um firewall quando registra um handler firewall. Esta função lê o corpo da requisição, envia-o ao modelo e aplica o veredito na mesma requisição. Para criá-la:

Abra a página de Functions

Acesse Azion Console > Products Menu > Libraries > Functions.

Selecione + Function
Nomeie a função

Insira um nome único e descritivo para a função.

Cole o código na aba Code

Na aba Code, cole a função abaixo.

Cole os argumentos na aba Arguments

Na aba Arguments, cole o JSON abaixo.

Selecione Save

A aba Code recebe esta função:

JavaScript
01async function handleRequest(event) {
02 try {
03 const requestBody = await event.request.text();
04 const { model, action, prompt } = event.args;
05 const contentType = event.request.headers.get("content-type");
06
07 // Allows images
08 if (contentType && contentType.startsWith("image/")) {
09 event.continue();
10 return;
11 }
12
13 // Decodes the content if it's base64
14 const decodedBody = contentType && contentType.includes("base64")
15 ? atob(requestBody.split(",")[1])
16 : requestBody;
17
18 // Calls the model via Azion.AI.run
19 const modelResponse = await Azion.AI.run(model, {
20 stream: false,
21 seed: 42,
22 temperature: 0,
23 max_tokens: 1024,
24 messages: [
25 {
26 role: "system",
27 content: `${prompt}`
28 },
29 {
30 role: "user",
31 content: decodedBody
32 }
33 ]
34 });
35
36 // Extracts the model's response
37 const result = modelResponse?.choices?.[0]?.message?.content?.trim();
38
39 if (result === "true") {
40 event.console.warn(`[AI] ${result}`);
41 event[action]();
42 return;
43 }
44 } catch (err) {
45 event.console.error("Error handling request:", err.message);
46 event.continue();
47 return;
48 }
49
50 event.continue(); // If not malicious, let it pass
51}
52
53addEventListener("firewall", event => {
54 event.waitUntil(handleRequest(event));
55});

A aba Arguments recebe este JSON:

JSON
01{
02 "model": "casperhansen-mistral-small-24b-instruct-2501-awq",
03 "prompt": "You are a security assistant specialized in detecting malicious PDF files. Analyze the provided content carefully and return 'true' only if you identify malicious content, such as embedded scripts, suspicious patterns, or known vulnerabilities. Do not classify a file as malicious based solely on its structure or the presence of a PDF header. If the content is safe or does not contain clear malicious indicators, return 'false'. Do not provide any additional explanation or output.",
04 "action": "deny"
05}

A função está salva e disponível para ser instanciada em um firewall.

A função lê o corpo da requisição e deixa um content type de imagem passar sem alteração. Ela decodifica conteúdo em base64, envia o conteúdo ao modelo com o prompt dos argumentos e aplica a ação. A ação, deny neste exemplo, é executada somente quando o modelo responde true. Em caso de erro, a função registra a mensagem em log e deixa a requisição continuar.

Os argumentos selecionam Mistral 3 Small, cujo id é casperhansen-mistral-small-24b-instruct-2501-awq. Certifique-se de que o modelo escolhido está disponível e configurado na sua conta. Para conhecer todos os ids que você pode passar, consulte Modelos de AI.

O handler é executado em Functions. Para conhecer todos os campos que Azion.AI.run aceita, consulte Invocação de modelos. Para saber o que acontece depois da chamada, consulte Como AI Inference funciona.

Ative os módulos do firewall

A função é executada em um firewall, e um firewall só executa uma função quando o módulo Functions está ativo. Para preparar o firewall:

Abra a página de Firewall

Em Azion Console, vá para Products Menu > Firewall.

Selecione o firewall que protege as rotas de upload
Ative os módulos que esta arquitetura precisa

Na aba Main Settings, ative os módulos DDoS Protection, Functions, Network Shield e Web Application Firewall.

Selecione Save

O firewall agora executa os módulos de que esta arquitetura depende, e pode conter uma instância de função.

Instancie a função no firewall

Uma instância de função vincula a função a um firewall. Para criar a instância:

Vá para a aba Functions Instances
Selecione + Function Instance
Insira um nome para a instância
Selecione a função que você criou
Selecione Save

A instância aparece na aba Functions Instances. Ela não é executada até que uma regra do Rules Engine a selecione.

Adicione a regra que executa a função

Uma regra do Rules Engine define os critérios que acionam a instância. Para executá-la nos uploads que sua aplicação recebe:

Vá para a aba Rules Engine
Selecione + Rule
Insira um nome para a regra
Selecione a variável Request URI

Na seção Criteria, selecione a variável Request URI.

Defina o operador de comparação como starts with
Insira a rota de upload

Insira uma rota de upload como argumento, por exemplo /upload.

Selecione Run Function

Na seção Behaviors, selecione Run Function.

Selecione a instância que você criou
Selecione Save

A regra executa a função em toda requisição cuja URI começa com essa rota. Um arquivo que o modelo classifica como malicioso é negado antes que a origem o receba. As alterações podem levar alguns minutos para se propagar. Aguarde antes de enviar um upload que corresponda ao critério.

Monitore e ajuste a política

Um modelo decide o que a política bloqueia, então o registro dele é a única forma de verificá-la. Acompanhe estes sinais em Real-Time Metrics e Real-Time Events:

  • A taxa de uploads bloqueados, permitidos e em quarentena.
  • Padrões por país, IP, ASN, rota e horário.
  • Eventos correlacionados com os incidentes de segurança que sua equipe reporta.

Depois, ajuste a política de acordo com o que você lê:

  • O prompt na aba Arguments, que define com que rigor o modelo responde.
  • Se um upload suspeito é negado ou retido para revisão em Object Storage.
  • As regras de Web Application Firewall e Network Shield no mesmo firewall.
  • O volume que a função envia ao modelo, dentro dos limites do AI Inference.
  • Os falsos positivos e falsos negativos que seus analistas rotulam, realimentados no prompt.

Os sinais mostram o que a função decidiu, e cada ajuste muda o que a próxima leitura mostra.

Próximos passos

Inspecionar uploads de arquivos em busca de conteúdo maliciosoO design que esta página implementa, com seu fluxo de dados e os componentes que ele coloca no caminho da requisição.Boas práticas do AI InferenceAs decisões que a função toma antes da chamada: o id do modelo, a entrada que ela envia e como ela lê a resposta.