nhpip

nhpip

Identifying function calls in AST

Hi,

As the title states, I want to reject an AST with function calls. This seems to work, is there a better way?

  @invalid_ast_ops [:., :import, :apply, :__block__]

  defp sanitize_args(args) do
    Macro.prewalk(args, fn {call, _, _} when call in @invalid_ast_ops ->
                              raise "Invalid arguments"

                           {call, _, args} when is_atom(call) and is_list(args) ->
                              Macro.special_form?(call, length(args)) || raise "Invalid arguments"

                           v -> v
    end)

Thanks

First Post! Switch mode

fuelen

fuelen

A couple of years ago, I wrote this code that works a bit differently – it allows only a whitelist and collects all invalid nodes for error reporting:

def ensure_ast_safe(ast) do
  Macro.prewalk(
    ast,
    [],
    fn
      {fn_name, _, _} = node, acc
      when fn_name in [:%{}, :sigil_D, :=, :sigil_U, :<<>>, :{}, :<>] ->
        {node, acc}

      {atom, _} = node, acc when is_atom(atom) ->
        {node, acc}

      node, acc
      when is_atom(node) or is_binary(node) or is_integer(node) or is_list(node) or
             is_float(node) ->
        {node, acc}

      node, acc ->
        {nil, [node | acc]}
    end
  )
  |> elem(1)
  |> case do
    [] ->
      :ok

    expressions ->
      {:error, {:unsupported_expressions, expressions}}
  end
end

Hope it helps

— All posts loaded —

Where Next?

Trending in Questions Top

stjefim
Hello! Suppose you are building workflow (order / task / payment) processing system with the following requirements: Each workflow con...
New
jonnycharles
I’m in search of an Elixir library that offers PDF generation capabilities similar to Ruby’s Prawn. While there have been discussions abo...
New
spammy
I’m looking to build a personal workflow to quickly deploy web applications written in elixir/phoenix, for local consumption (ie not on t...
New
silverdr
Using Phoenix.LiveView.TagEngine as an EEx.Engine is deprecated! To compile HEEx, use Phoenix.LiveView.TagEngine.compile/2 instead. Sta...
New
dli
Before I dive in myself, did anyone successfully sprinkle Hologram into their existing LiveView app? Looking for hints regarding: Addi...
New
bottlenecked
Hi all, I wanted to ask how the community is dealing with post-release steps. Today we have Ecto migrations, which make sure that the db...
New
michallepicki
I am using Oban and occasionally, shortly after a deployment, a handful of jobs can fail because of dependency on other parts of the syst...
New

Other Trending Topics Top

JesseHerrick
Hey, I’m Jesse and I’m the main contributor behind Dexter, a full-featured, lightning-fast Elixir LSP optimized for large codebases. It s...
New
jimsynz
Beam Bots (or just BB for short) is a framework for building fault-tolerant robotics applications in Elixir using familiar OTP patterns. ...
New
Damirados
Hello everyone. After busy few months I am happy to announce v0.1.0 of Emerge &amp; Solve. They are GUI (Emerge) and State management (S...
New
ausimian
Emily is an Elixir library that runs Nx computations on Apple’s MLX. Install it as the default Nx backend and Nx, defn, Axon, Nx.Serving,...
New
type1fool
I just stumbled on a newly redesigned elixir-lang.org. :tada: It looks like @Software_Mansion did the work, and I think it is generally a...
New
akoutmos
@hugobarauna and I (Alex Koutmos) have been hard at work on writing a book on Nerves that takes you from simply blinking LEDs to building...
New

We're in Beta

About us Mission Statement