git ssb

1
const {wasm2json, json2wasm} = require('wasm-json-toolkit')
2
const annotations = require('primea-annotations')
3
const wasmMetering = require('wasm-metering')
4
const ReferanceMap = require('reference-map')
5
const Message = require('./message.js')
6
const injectGlobals = require('./injectGlobals.js')
7
const typeCheckWrapper = require('./typeCheckWrapper.js')
8
const {FunctionRef, ModuleRef, DEFAULTS} = require('./systemObjects.js')
9
10
const nativeTypes = new Set(['i32', 'i64', 'f32', 'f64'])
11
const FUNC_INDEX_OFFSET = 1
12
13
function fromMetaJSON (json, id) {
14
  const exports = {}
15
  for (const ex in json.exports) {
16
    const type = json.types[json.indexes[json.exports[ex].toString()]].params
17
    exports[ex] = type
18
  }
19
  return new ModuleRef(exports, id)
20
}
21
22
function generateWrapper (funcRef, container) {
23
  let wrapper = typeCheckWrapper(funcRef.params)
24
  const wasm = json2wasm(wrapper)
25
  const mod = WebAssembly.Module(wasm)
26
  const self = funcRef
27
  wrapper = WebAssembly.Instance(mod, {
28
    'env': {
29
      'checkTypes': function () {
30
        const args = [...arguments]
31
        const checkedArgs = []
32
        while (args.length) {
33
          const type = annotations.LANGUAGE_TYPES_BIN[args.shift()]
34
          let arg = args.shift()
35
          if (!nativeTypes.has(type)) {
36
            arg = container.refs.get(arg, type)
37
          }
38
          checkedArgs.push(arg)
39
        }
40
        const message = new Message({
41
          funcRef: self,
42
          funcArguments: checkedArgs
43
        })
44
        container.actor.send(message)
45
      }
46
    }
47
  })
48
  wrapper.exports.check.object = funcRef
49
  return wrapper
50
}
51
52
module.exports = class WasmContainer {
53
  constructor (actor) {
54
    this.actor = actor
55
    this.refs = new ReferanceMap()
56
  }
57
58
  static createModule (wasm, id) {
59
    if (!WebAssembly.validate(wasm)) {
60
      throw new Error('invalid wasm binary')
61
    }
62
63
    let moduleJSON = wasm2json(wasm)
64
    const json = annotations.mergeTypeSections(moduleJSON)
65
    moduleJSON = wasmMetering.meterJSON(moduleJSON, {
66
      meterType: 'i32'
67
    })
68
69
    // initialize the globals
70
    if (json.persist.length) {
71
      moduleJSON = injectGlobals(moduleJSON, json.persist)
72
    }
73
    // recompile the wasm
74
    wasm = json2wasm(moduleJSON)
75
    const modRef = fromMetaJSON(json, id)
76
    return {
77
      wasm,
78
      json,
79
      modRef
80
    }
81
  }
82
83
  static onCreation (unverifiedWasm, id, tree) {
84
    const {modRef} = this.createModule(unverifiedWasm, id)
85
    return modRef
86
  }
87
88
  getInterface (funcRef) {
89
    const self = this
90
    return {
91
      func: {
92
        externalize: index => {
93
          const func = this.instance.exports.table.get(index)
94
          const object = func.object
95
          if (object) {
96
            // externalize a pervously internalized function
97
            return self.refs.add(object)
98
          } else {
99
            const params = self.json.types[self.json.indexes[func.name - FUNC_INDEX_OFFSET]].params
100
            const ref = new FunctionRef({
101
              identifier: [true, func.tableIndex],
102
              params,
103
              id: self.actor.id
104
            })
105
            return self.refs.add(ref, 'func')
106
          }
107
        },
108
        internalize: (index, ref) => {
109
          const funcRef = self.refs.get(ref, 'func')
110
          const wrapper = generateWrapper(funcRef, self)
111
          this.instance.exports.table.set(index, wrapper.exports.check)
112
        },
113
        catch: (ref, catchRef) => {
114
          const {funcRef} = self.refs.get(ref, FunctionRef)
115
          const {funcRef: catchFunc} = self.refs.get(ref, FunctionRef)
116
          funcRef.catch = catchFunc
117
        },
118
        get_gas_budget: (funcRef) => {
119
          const func = self.refs.get(funcRef, 'func')
120
          return func.gas
121
        },
122
        set_gas_budget: (funcRef, amount) => {
123
          const func = self.refs.get(funcRef, 'func')
124
          func.gas = amount
125
        }
126
      },
127
      link: {
128
        wrap: ref => {
129
          const obj = this.refs.get(ref)
130
          const link = {'/': obj}
131
          return this.refs.add(link, 'link')
132
        },
133
        unwrap: async (ref, cb) => {
134
          const obj = this.refs.get(ref, 'link')
135
          const promise = this.actor.tree.dataStore.get(obj)
136
          await this._opsQueue.push(promise)
137
        }
138
      },
139
      module: {
140
        new: dataRef => {
141
          const mod = this.actor.createActor(dataRef)
142
          return this.refs.add(mod, 'mod')
143
        },
144
        export: (modRef, dataRef) => {
145
          const mod = this.refs.get(modRef, 'mod')
146
          let name = this.refs.get(dataRef, 'data')
147
          name = Buffer.from(name).toString()
148
          const funcRef = mod.getFuncRef(name)
149
          return this.refs.add(funcRef, 'func')
150
        },
151
        self: () => {
152
          return this.refs.add(this.modSelf, 'mod')
153
        }
154
      },
155
      memory: {
156
        externalize: (index, length) => {
157
          const data = Buffer.from(this.get8Memory(index, length))
158
          return this.refs.add(data, 'data')
159
        },
160
        internalize: (dataRef, srcOffset, sinkOffset, length) => {
161
          let data = this.refs.get(dataRef, 'data')
162
          data = data.subarray(srcOffset, length)
163
          const mem = this.get8Memory(sinkOffset, data.length)
164
          mem.set(data)
165
        },
166
        length (dataRef) {
167
          let data = this.refs.get(dataRef, 'data')
168
          return data.length
169
        }
170
      },
171
      table: {
172
        externalize: (index, length) => {
173
          const mem = Buffer.from(this.get8Memory(index, length * 4))
174
          const objects = []
175
          while (length--) {
176
            const ref = mem.readUInt32LE(length * 4)
177
            const obj = this.refs.get(ref)
178
            objects.unshift(obj)
179
          }
180
          return this.refs.add(objects, 'elem')
181
        },
182
        internalize: (elemRef, srcOffset, sinkOffset, length) => {
183
          let table = this.refs.get(elemRef, 'elem')
184
          const buf = table.slice(srcOffset, srcOffset + length).map(obj => this.refs.add(obj))
185
          const mem = this.get32Memory(sinkOffset, length)
186
          mem.set(buf)
187
        },
188
        length (elemRef) {
189
          let elem = this.refs.get(elemRef, 'elem')
190
          return elem.length
191
        }
192
      },
193
      metering: {
194
        usegas: amount => {
195
          this.actor.incrementTicks(amount)
196
          funcRef.gas -= amount
197
          if (funcRef.gas < 0) {
198
            throw new Error('out of gas! :(')
199
          }
200
        }
201
      }
202
    }
203
  }
204
205
  async onMessage (message) {
206
    const funcRef = message.funcRef
207
    const intef = this.getInterface(funcRef)
208
    this.instance = WebAssembly.Instance(this.mod, intef)
209
    // map table indexes
210
    const table = this.instance.exports.table
211
    if (table) {
212
      let length = table.length
213
      while (length--) {
214
        const func = table.get(length)
215
        if (func) {
216
          func.tableIndex = length
217
        }
218
      }
219
    }
220
    // import references
221
    const args = message.funcArguments.map((arg, index) => {
222
      const type = funcRef.params[index]
223
      if (nativeTypes.has(type)) {
224
        return arg
225
      } else {
226
        return this.refs.add(arg, type)
227
      }
228
    })
229
230
    // setup globals
231
    let numOfGlobals = this.json.persist.length
232
    if (numOfGlobals) {
233
      const refs = []
234
      while (numOfGlobals--) {
235
        const obj = this.actor.storage[numOfGlobals] || DEFAULTS[this.json.persist[numOfGlobals].type]
236
        refs.push(this.refs.add(obj, this.json.persist[numOfGlobals].type))
237
      }
238
      this.instance.exports.setter_globals(...refs)
239
    }
240
241
    // call entrypoint function
242
    if (funcRef.identifier[0]) {
243
      this.instance.exports.table.get(funcRef.identifier[1])(...args)
244
    } else {
245
      this.instance.exports[funcRef.identifier[1]](...args)
246
    }
247
    await this.onDone()
248
249
    // store globals
250
    numOfGlobals = this.json.persist.length
251
    if (numOfGlobals) {
252
      const storage = []
253
      this.instance.exports.getter_globals()
254
      const mem = this.get32Memory(0, numOfGlobals)
255
      while (numOfGlobals--) {
256
        const ref = mem[numOfGlobals]
257
        storage.push(this.refs.get(ref, this.json.persist[numOfGlobals].type))
258
      }
259
      this.actor.storage = storage
260
    }
261
262
    this.refs.clear()
263
  }
264
265
  /**
266
   * returns a promise that resolves when the wasm instance is done running
267
   * @returns {Promise}
268
   */
269
  async onDone () {
270
    let prevOps
271
    while (prevOps !== this._opsQueue) {
272
      prevOps = this._opsQueue
273
      await prevOps
274
    }
275
  }
276
277
  /**
278
   * Pushed an async operation to the a promise queue that
279
   * @returns {Promise} the returned promise resolves in the order the intail
280
   * operation was pushed to the queue
281
   */
282
  pushOpsQueue (promise) {
283
    this._opsQueue = Promise.all([this._opsQueue, promise])
284
    return this._opsQueue
285
  }
286
287
  async onStartup () {
288
    const code = this.actor.code
289
    const {json, wasm, modRef} = WasmContainer.createModule(code, this.actor.id)
290
    this.mod = WebAssembly.Module(wasm)
291
    this.json = json
292
    this.modSelf = modRef
293
  }
294
295
  get8Memory (offset, length) {
296
    return new Uint8Array(this.instance.exports.memory.buffer, offset, length)
297
  }
298
299
  get32Memory (offset, length) {
300
    return new Uint32Array(this.instance.exports.memory.buffer, offset, length)
301
  }
302
303
  static get typeId () {
304
    return 9
305
  }
306
}
307