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

Most Liked

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

Where Next?

Popular in Questions Top

_russellb
I want to try my hand at web scraping. What tools/libraries do I need to use. I’m hoping to turn this into something professional so don’...
New
9mm
I am constructing a JSON object (map) and I need to conditionally set a field. I’m trying to write proper elixir-way code… and I’m at a l...
New
chrisalley
ExUnit now has describe blocks which is a welcome addition coming from RSpec. In the docs, it states that nested hierarchies of describe ...
New
Fl4m3Ph03n1x
About me? ( if you have nothing better to do than reading about some random guy in the internet :stuck_out_tongue: ) Hello all, this is ...
New
dokuzbir
I want to highlight html closing tags when i click a html tag. That works in .html files but doesnt work for html.eex templates. How can...
New
sergio_101
I am VERY much an elixir newbie. I have taken one elixir course and one phoenix course on Udemy. During that course, I saw the instructor...
New
script
If I have a string “1000 cfu/ml” . I want to remove the characters and / and space . So the string is like this "1000" What is the ...
New
rms.mrcs
Hi, I need to transform a list of numbers into a map where the keys are the indexes and the values are the original values of the list. ...
New
hariharasudhan94
I would like to know what is the best IDE for elixir development?
New
WestKeys
Currently suffering from paralysis by [HTTP client] analysis. This is rather unusual in Elixirland as there tends to be consensus on the ...
New

Other popular topics Top

New
minhajuddin
I have seen a lot of code which picks the first element from a list using Enum.at(0) instead of List.first. Is there a reason why people ...
New
baxterw3b
Hi guys, i’m new in the Elixir world, and i have to say, that i love it! i’m having some problem to understand anonymous functions with ...
New
pmjoe
I have a relationship of love and hate with Elixir. Lots of things are just absolutely right, but there are some things that are kind of ...
New
boundedvariable
I am going through the kafka architecture. All the features what the kafka is providing are already in Erlang. I would like hear your opi...
New
hariharasudhan94
I would like to know what is the best IDE for elixir development?
New
marick
I had some trouble figuring out how to make many-to-many associations work. Once I got it working, I wrote a blog post. Because I’m a nov...
New
openscript
Hello! Sorry for this astonishing simple question, but I’m really stuck. I try to set up the intellij-elixir plugin, but I don’t know ho...
New
PeterCarter
There are pre-rolled solutions for other frameworks that do work. However, Phoenix does not seem to have these. Have people had good expe...
New
vonH
In asking this question I am more interested about the expressiveness of the language itself and less concerned about the availability of...
New

We're in Beta

About us Mission Statement