blob: e7ff54b915206dc7165a0aac91f0fcbef5319be6 (
plain)
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
|
/*
* Copyright 2021 WebAssembly Community Group participants
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef wasm_ir_element_h
#define wasm_ir_element_h
#include "wasm-builder.h"
#include "wasm.h"
namespace wasm::ElementUtils {
// iterate over functions referenced in an element segment
template<typename T>
inline void iterElementSegmentFunctionNames(ElementSegment* segment,
T visitor) {
if (!segment->type.isFunction()) {
return;
}
for (Index i = 0; i < segment->data.size(); i++) {
if (auto* get = segment->data[i]->dynCast<RefFunc>()) {
visitor(get->func, i);
}
}
}
// iterate over functions referenced in all element segments of a module
template<typename T>
inline void iterAllElementFunctionNames(const Module* wasm, T visitor) {
for (auto& segment : wasm->elementSegments) {
iterElementSegmentFunctionNames(
segment.get(), [&](Name& name, Index i) { visitor(name); });
}
}
} // namespace wasm::ElementUtils
#endif // wasm_ir_element_h
|
