1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
(* SPDX-License-Identifier: AGPL-3.0-or-later *)
(* Copyright © 2021-2024 OCamlPro *)
(* Written by the Owi programmers *)
open Text
open Syntax
type env =
{ start : bool
; memory : bool
; imported_memory : bool
; funcs : bool
; tables : bool
; globals : bool
}
let empty_env () =
{ start = false
; memory = false
; imported_memory = false
; funcs = false
; tables = false
; globals = false
}
let modul m =
Log.info (fun m -> m "checking ...");
let add_global =
let seen = Hashtbl.create 512 in
function
| None -> Ok ()
| Some id ->
if Hashtbl.mem seen id then Error (`Duplicate_global id)
else Ok (Hashtbl.replace seen id ())
in
let add_table =
let seen = Hashtbl.create 512 in
function
| None -> Ok ()
| Some id ->
if Hashtbl.mem seen id then Error (`Duplicate_table id)
else Ok (Hashtbl.replace seen id ())
in
let add_memory =
let seen = Hashtbl.create 512 in
function
| None -> Ok ()
| Some id ->
if Hashtbl.mem seen id then Error (`Duplicate_memory id)
else Ok (Hashtbl.add seen id ())
in
let+ (_env : env) =
let open Module in
list_fold_left
(fun env ->
let open Field in
function
| Export _e -> Ok env
| Func _f -> Ok { env with funcs = true }
| Start _start ->
if env.start then Error `Multiple_start_sections
else Ok { env with start = true }
| Import i ->
if env.funcs then Error `Import_after_function
else if env.memory then Error `Import_after_memory
else if env.tables then Error `Import_after_table
else if env.globals then Error `Import_after_global
else begin
match i.typ with
| Mem (id, _) ->
let* () = add_memory id in
if env.imported_memory then Error `Multiple_memories
else Ok { env with imported_memory = true }
| Func _ -> Ok env
| Global (id, _) ->
let+ () = add_global id in
env
| Table (id, _) ->
let+ () = add_table id in
env
end
| Data _d -> Ok env
| Elem _e -> Ok env
| Mem (id, _) ->
let* () = add_memory id in
if env.memory || env.imported_memory then Error `Multiple_memories
else Ok { env with memory = true }
| Typedef _t -> Ok env
| Global { id; _ } ->
let+ () = add_global id in
{ env with globals = true }
| Table (id, _) ->
let+ () = add_table id in
{ env with tables = true } )
(empty_env ()) m.fields
in
m