https://github.com/akkartik/mu/blob/master/035lookup.cc
  1 //: Go from an address to the payload it points at using /lookup.
  2 //:
  3 //: The tests in this layer use unsafe operations so as to stay decoupled from
  4 //: 'new'.
  5 
  6 :(scenario copy_indirect)
  7 def main [
  8   # skip alloc id for 10:&:num
  9   11:num <- copy 20
 10   # skip alloc id for payload
 11   21:num <- copy 94
 12   # Treat locations 10 and 11 as an address to look up, pointing at the
 13   # payload in locations 20 and 21.
 14   30:num <- copy 10:&:num/lookup
 15 ]
 16 +mem: storing 94 in location 30
 17 
 18 :(before "End Preprocess read_memory(x)")
 19 canonize(x);
 20 
 21 //: similarly, write to addresses pointing at other locations using the
 22 //: 'lookup' property
 23 :(scenario store_indirect)
 24 def main [
 25   # skip alloc id for 10:&:num
 26   11:num <- copy 10
 27   10:&:num/lookup <- copy 94
 28 ]
 29 +mem: storing 94 in location 11
 30 
 31 :(before "End Preprocess write_memory(x, data)")
 32 canonize(x);
 33 
 34 //: writes to address 0 always loudly fail
 35 :(scenario store_to_0_fails)
 36 % Hide_errors = true;
 37 def main [
 38   10:&:num <- copy null
 39   10:&:num/lookup <- copy 94
 40 ]
 41 -mem: storing 94 in location 0
 42 +error: main: tried to lookup 0 in '10:&:num/lookup <- copy 94'
 43 
 44 //: attempts to /lookup address 0 always loudly fail
 45 :(scenario lookup_0_fails)
 46 % Hide_errors = true;
 47 def main [
 48   10:&:num <- copy null
 49   20:num <- copy 10:&:num/lookup
 50 ]
 51 +error: main: tried to lookup 0 in '20:num <- copy 10:&:num/lookup'
 52 
 53 :(scenario lookup_0_dumps_callstack)
 54 % Hide_errors = true;
 55 def main [
 56   foo null
 57 ]
 58 def foo [
 59   10:&:num <- next-input
 60   20:num <- copy 10:&:num/lookup
 61 ]
 62 +error: foo: tried to lookup 0 in '20:num <- copy 10:&:num/lookup'
 63 +error:   called from main: foo null
 64 
 65 :(code)
 66 void canonize(reagent& x) {
 67   if (is_literal(x)) return;
 68   // Begin canonize(x) Lookups
 69   while (has_property(x, "lookup"))
 70     lookup_memory(x);
 71 }
 72 
 73 void lookup_memory(reagent& x) {
 74   if (!x.type || x.type->atom || x.type->left->value != Address_type_ordinal) {
 75     raise << maybe(current_recipe_name()) << "tried to lookup '" << x.original_string << "' but it isn't an address\n" << end();
 76     dump_callstack();
 77     return;
 78   }
 79   // compute value
 80   if (x.value == 0) {
 81     raise << maybe(current_recipe_name()) << "tried to lookup 0\n" << end();
 82     dump_callstack();
 83     return;
 84   }
 85   lookup_memory_core(x, /*check_for_null*/true);
 86 }
 87 
 88 void lookup_memory_core(reagent& x, bool check_for_null) {
 89   double address = x.value + /*skip alloc id in address*/1;
 90   double new_value = get_or_insert(Memory, address);
 91   trace("mem") << "location " << address << " contains " << no_scientific(new_value) << end();
 92   // check for null
 93   if (check_for_null && new_value == 0) {
 94     if (Current_routine) {
 95       raise << maybe(current_recipe_name()) << "tried to lookup 0 in '" << to_original_string(current_instruction()) << "'\n" << end();
 96       dump_callstack();
 97     }
 98     else {
 99       raise << "tried to lookup 0\n" << end();
100     }
101   }
102   // validate alloc-id
103   double alloc_id_in_address = get_or_insert(Memory, x.value);
104   double alloc_id_in_payload = get_or_insert(Memory, new_value);
105 //?   cerr << x.value << ": " << alloc_id_in_address << " vs " << new_value << ": " << alloc_id_in_payload << '\n';
106   if (alloc_id_in_address != alloc_id_in_payload) {
107       raise << maybe(current_recipe_name()) << "address is already abandoned in '" << to_original_string(current_instruction()) << "'\n" << end();
108       dump_callstack();
109   }
110   // all well; complete the lookup
111   x.set_value(new_value+/*skip alloc id in payload*/1);
112   drop_from_type(x, "address");
113   drop_one_lookup(x);
114 }
115 
116 :(after "Begin types_coercible(reagent to, reagent from)")
117 if (!canonize_type(to)) return false;
118 if (!canonize_type(from)) return false;
119 :(after "Begin types_match(reagent to, reagent from)")
120 if (!canonize_type(to)) return false;
121 if (!canonize_type(from)) return false;
122 :(after "Begin types_strictly_match(reagent to, reagent from)")
123 if (!canonize_type(to)) return false;
124 if (!canonize_type(from)) return false;
125 
126 :(before "End Preprocess is_mu_array(reagent r)")
127 if (!canonize_type(r)) return false;
128 
129 :(before "End Preprocess is_mu_address(reagent r)")
130 if (!canonize_type(r)) return false;
131 
132 :(before "End Preprocess is_mu_number(reagent r)")
133 if (!canonize_type(r)) return false;
134 :(before "End Preprocess is_mu_boolean(reagent r)")
135 if (!canonize_type(r)) return false;
136 :(before "End Preprocess is_mu_character(reagent r)")
137 if (!canonize_type(r)) return false;
138 
139 :(after "Update product While Type-checking Merge")
140 if (!canonize_type(product)) continue;
141 
142 :(before "End Compute Call Ingredient")
143 canonize_type(ingredient);
144 :(before "End Preprocess NEXT_INGREDIENT product")
145 canonize_type(product);
146 :(before "End Check RETURN Copy(lhs, rhs)
147 canonize_type(lhs);
148 canonize_type(rhs);
149 
150 :(code)
151 bool canonize_type(reagent& r) {
152   while (has_property(r, "lookup")) {
153     if (!r.type || r.type->atom || !r.type->left || !r.type->left->atom || r.type->left->value != Address_type_ordinal) {
154       raise << "cannot perform lookup on '" << r.name << "' because it has non-address type " << to_string(r.type) << '\n' << end();
155       return false;
156     }
157     drop_from_type(r, "address");
158     drop_one_lookup(r);
159   }
160   return true;
161 }
162 
163 void drop_one_lookup(reagent& r) {
164   for (vector<pair<string, string_tree*> >::iterator p = r.properties.begin();  p != r.properties.end();  ++p) {
165     if (p->first == "lookup") {
166       r.properties.erase(p);
167       return;
168     }
169   }
170   assert(false);
171 }
172 
173 //: Tedious fixup to support addresses in container/array instructions of previous layers.
174 //: Most instructions don't require fixup if they use the 'ingredients' and
175 //: 'products' variables in run_current_routine().
176 
177 :(scenario get_indirect)
178 def main [
179   # skip alloc id for 10:&:point
180   11:num <- copy 20
181   # skip alloc id for payload
182   21:num <- copy 94
183   22:num <- copy 95
184   30:num <- get 10:&:point/lookup, 0:offset
185 ]
186 +mem: storing 94 in location 30
187 
188 :(scenario get_indirect2)
189 def main [
190   # skip alloc id for 10:&:point
191   11:num <- copy 20
192   # skip alloc id for payload
193   21:num <- copy 94
194   22:num <- copy 95
195   # skip alloc id for destination
196   31:num <- copy 40
197   30:&:num/lookup <- get 10:&:point/lookup, 0:offset
198 ]
199 +mem: storing 94 in location 41
200 
201 :(scenario include_nonlookup_properties)
202 def main [
203   # skip alloc id for 10:&:point
204   11:num <- copy 20
205   # skip alloc id for payload
206   21:num <- copy 94
207   22:num <- copy 95
208   30:num <- get 10:&:point/lookup/foo, 0:offset
209 ]
210 +mem: storing 94 in location 30
211 
212 :(after "Update GET base in Check")
213 if (!canonize_type(base)) break;
214 :(after "Update GET product in Check")
215 if (!canonize_type(product)) break;
216 :(after "Update GET base in Run")
217 canonize(base);
218 
219 :(scenario put_indirect)
220 def main [
221   # skip alloc id for 10:&:point
222   11:num <- copy 20
223   # skip alloc id for payload
224   21:num <- copy 94
225   22:num <- copy 95
226   10:&:point/lookup <- put 10:&:point/lookup, 0:offset, 96
227 ]
228 +mem: storing 96 in location 21
229 
230 :(after "Update PUT base in Check")
231 if (!canonize_type(base)) break;
232 :(after "Update PUT offset in Check")
233 if (!canonize_type(offset)) break;
234 :(after "Update PUT base in Run")
235 canonize(base);
236 
237 :(scenario put_product_error_with_lookup)
238 % Hide_errors = true;
239 def main [
240   # skip alloc id for 10:&:point
241   11:num <- copy 20
242   # skip alloc id for payload
243   21:num <- copy 94
244   22:num <- copy 95
245   10:&:point <- put 10:&:point/lookup, x:offset, 96
246 ]
247 +error: main: product of 'put' must be first ingredient '10:&:point/lookup', but got '10:&:point'
248 
249 :(before "End PUT Product Checks")
250 reagent/*copy*/ p = inst.products.at(0);
251 if (!canonize_type(p)) break;  // error raised elsewhere
252 reagent/*copy*/ i = inst.ingredients.at(0);
253 if (!canonize_type(i)) break;  // error raised elsewhere
254 if (!types_strictly_match(p, i)) {
255   raise << maybe(get(Recipe, r).name) << "product of 'put' must be first ingredient '" << inst.ingredients.at(0).original_string << "', but got '" << inst.products.at(0).original_string << "'\n" << end();
256   break;
257 }
258 
259 :(scenario new_error)
260 % Hide_errors = true;
261 def main [
262   1:num/raw <- new num:type
263 ]
264 +error: main: product of 'new' has incorrect type: '1:num/raw <- new num:type'
265 
266 :(after "Update NEW product in Check")
267 canonize_type(product);
268 
269 :(scenario copy_array_indirect)
270 def main [
271   # skip alloc id for 10:&:@:num
272   11:num <- copy 20
273   # skip alloc id for payload
274   21:num <- copy 3  # array length
275   22:num <- copy 94
276   23:num <- copy 95
277   24:num <- copy 96
278   30:@:num <- copy 10:&:@:num/lookup
279 ]
280 +mem: storing 3 in location 30
281 +mem: storing 94 in location 31
282 +mem: storing 95 in location 32
283 +mem: storing 96 in location 33
284 
285 :(scenario create_array_indirect)
286 def main [
287   # skip alloc id for 10:&:@:num:3
288   11:num <- copy 3000
289   10:&:array:num:3/lookup <- create-array
290 ]
291 +mem: storing 3 in location 3001
292 
293 :(after "Update CREATE_ARRAY product in Check")
294 if (!canonize_type(product)) break;
295 :(after "Update CREATE_ARRAY product in Run")
296 canonize(product);
297 
298 :(scenario index_indirect)
299 def main [
300   # skip alloc id for 10:&:@:num
301   11:num <- copy 20
302   # skip alloc id for payload
303   21:num <- copy 3  # array length
304   22:num <- copy 94
305   23:num <- copy 95
306   24:num <- copy 96
307   30:num <- index 10:&:@:num/lookup, 1
308 ]
309 +mem: storing 95 in location 30
310 
311 :(before "Update INDEX base in Check")
312 if (!canonize_type(base)) break;
313 :(before "Update INDEX index in Check")
314 if (!canonize_type(index)) break;
315 :(before "Update INDEX product in Check")
316 if (!canonize_type(product)) break;
317 
318 :(before "Update INDEX base in Run")
319 canonize(base);
320 :(before "Update INDEX index in Run")
321 canonize(index);
322 
323 :(scenario put_index_indirect)
324 def main [
325   # skip alloc id for 10:&:@:num
326   11:num <- copy 20
327   # skip alloc id for payload
328   21:num <- copy 3  # array length
329   22:num <- copy 94
330   23:num <- copy 95
331   24:num <- copy 96
332   10:&:@:num/lookup <- put-index 10:&:@:num/lookup, 1, 97
333 ]
334 +mem: storing 97 in location 23
335 
336 :(scenario put_index_indirect_2)
337 def main [
338   10:num <- copy 3  # array length
339   11:num <- copy 94
340   12:num <- copy 95
341   13:num <- copy 96
342   # skip alloc id for address
343   21:num <- copy 30
344   # skip alloc id for payload
345   31:num <- copy 1  # index
346   10:@:num <- put-index 10:@:num, 20:&:num/lookup, 97
347 ]
348 +mem: storing 97 in location 12
349 
350 :(scenario put_index_product_error_with_lookup)
351 % Hide_errors = true;
352 def main [
353   # skip alloc id for 10:&:@:num
354   11:num <- copy 20
355   # skip alloc id for payload
356   21:num <- copy 3  # array length
357   22:num <- copy 94
358   23:num <- copy 95
359   24:num <- copy 96
360   10:&:@:num <- put-index 10:&:@:num/lookup, 1, 34
361 ]
362 +error: main: product of 'put-index' must be first ingredient '10:&:@:num/lookup', but got '10:&:@:num'
363 
364 :(before "End PUT_INDEX Product Checks")
365 reagent/*copy*/ p = inst.products.at(0);
366 if (!canonize_type(p)) break;  // error raised elsewhere
367 reagent/*copy*/ i = inst.ingredients.at(0);
368 if (!canonize_type(i)) break;  // error raised elsewhere
369 if (!types_strictly_match(p, i)) {
370   raise << maybe(get(Recipe, r).name) << "product of 'put-index' must be first ingredient '" << inst.ingredients.at(0).original_string << "', but got '" << inst.products.at(0).original_string << "'\n" << end();
371   break;
372 }
373 
374 :(scenario dilated_reagent_in_static_array)
375 def main [
376   {1: (array (& num) 3)} <- create-array
377   10:&:num <- new num:type
378   {1: (array (& num) 3)} <- put-index {1: (array (& num) 3)}, 0, 10:&:num
379   *10:&:num <- copy 94
380   20:num <- copy *10:&:num
381 ]
382 +run: creating array from 7 locations
383 +mem: storing 94 in location 20
384 
385 :(before "Update PUT_INDEX base in Check")
386 if (!canonize_type(base)) break;
387 :(before "Update PUT_INDEX index in Check")
388 if (!canonize_type(index)) break;
389 :(before "Update PUT_INDEX value in Check")
390 if (!canonize_type(value)) break;
391 
392 :(before "Update PUT_INDEX base in Run")
393 canonize(base);
394 :(before "Update PUT_INDEX index in Run")
395 canonize(index);
396 
397 :(scenario length_indirect)
398 def main [
399   # skip alloc id for 10:&:@:num
400   11:num <- copy 20
401   # skip alloc id for payload
402   21:num <- copy 3  # array length
403   22:num <- copy 94
404   23:num <- copy 95
405   24:num <- copy 96
406   30:num <- length 10:&:array:num/lookup
407 ]
408 +mem: storing 3 in location 30
409 
410 :(before "Update LENGTH array in Check")
411 if (!canonize_type(array)) break;
412 :(before "Update LENGTH array in Run")
413 canonize(array);
414 
415 :(scenario maybe_convert_indirect)
416 def main [
417   # skip alloc id for 10:&:number-or-point
418   11:num <- copy 20
419   # skip alloc id for payload
420   21:number-or-point <- merge 0/number, 94
421   30:num, 31:bool <- maybe-convert 10:&:number-or-point/lookup, i:variant
422 ]
423 +mem: storing 1 in location 31
424 +mem: storing 94 in location 30
425 
426 :(scenario maybe_convert_indirect_2)
427 def main [
428   # skip alloc id for 10:&:number-or-point
429   11:num <- copy 20
430   # skip alloc id for payload
431   21:number-or-point <- merge 0/number, 94
432   # skip alloc id for 30:&:num
433   31:num <- copy 40
434   30:&:num/lookup, 50:bool <- maybe-convert 10:&:number-or-point/lookup, i:variant
435 ]
436 +mem: storing 1 in location 50
437 +mem: storing 94 in location 41
438 
439 :(scenario maybe_convert_indirect_3)
440 def main [
441   # skip alloc id for 10:&:number-or-point
442   11:num <- copy 20
443   # skip alloc id for payload
444   21:number-or-point <- merge 0/number, 94
445   # skip alloc id for 30:&:bool
446   31:num <- copy 40
447   50:num, 30:&:bool/lookup <- maybe-convert 10:&:number-or-point/lookup, i:variant
448 ]
449 +mem: storing 1 in location 41
450 +mem: storing 94 in location 50
451 
452 :(before "Update MAYBE_CONVERT base in Check")
453 if (!canonize_type(base)) break;
454 :(before "Update MAYBE_CONVERT product in Check")
455 if (!canonize_type(product)) break;
456 :(before "Update MAYBE_CONVERT status in Check")
457 if (!canonize_type(status)) break;
458 
459 :(before "Update MAYBE_CONVERT base in Run")
460 canonize(base);
461 :(before "Update MAYBE_CONVERT product in Run")
462 canonize(product);
463 :(before "Update MAYBE_CONVERT status in Run")
464 canonize(status);
465 
466 :(scenario merge_exclusive_container_indirect)
467 def main [
468   # skip alloc id for 10:&:number-or-point
469   11:num <- copy 20
470   10:&:number-or-point/lookup <- merge 0/number, 34
471 ]
472 # skip alloc id
473 +mem: storing 0 in location 21
474 +mem: storing 34 in location 22
475 
476 :(before "Update size_mismatch Check for MERGE(x)
477 canonize(x);
478 
479 //: abbreviation for '/lookup': a prefix '*'
480 
481 :(scenario lookup_abbreviation)
482 def main [
483   # skip alloc id for 10:&:num
484   11:num <- copy 20
485   # skip alloc id for payload
486   21:num <- copy 94
487   30:num <- copy *10:&:num
488 ]
489 +parse: ingredient: {10: ("&" "num"), "lookup": ()}
490 +mem: storing 94 in location 30
491 
492 :(before "End Parsing reagent")
493 {
494   while (starts_with(name, "*")) {
495     name.erase(0, 1);
496     properties.push_back(pair<string, string_tree*>("lookup", NULL));
497   }
498   if (name.empty())
499     raise << "illegal name '" << original_string << "'\n" << end();
500 }
501 
502 //:: helpers for debugging
503 
504 :(before "End Primitive Recipe Declarations")
505 _DUMP,
506 :(before "End Primitive Recipe Numbers")
507 put(Recipe_ordinal, "$dump", _DUMP);
508 :(before "End Primitive Recipe Implementations")
509 case _DUMP: {
510   reagent/*copy*/ after_canonize = current_instruction().ingredients.at(0);
511   canonize(after_canonize);
512   cerr << maybe(current_recipe_name()) << current_instruction().ingredients.at(0).name << ' ' << no_scientific(current_instruction().ingredients.at(0).value) << " => " << no_scientific(after_canonize.value) << " => " << no_scientific(get_or_insert(Memory, after_canonize.value)) << '\n';
513   break;
514 }
515 
516 //: grab an address, and then dump its value at intervals
517 //: useful for tracking down memory corruption (writing to an out-of-bounds address)
518 :(before "End Globals")
519 int Bar = -1;
520 :(before "End Primitive Recipe Declarations")
521 _BAR,
522 :(before "End Primitive Recipe Numbers")
523 put(Recipe_ordinal, "$bar", _BAR);
524 :(before "End Primitive Recipe Implementations")
525 case _BAR: {
526   if (current_instruction().ingredients.empty()) {
527     if (Bar != -1) cerr << Bar << ": " << no_scientific(get_or_insert(Memory, Bar)) << '\n';
528     else cerr << '\n';
529   }
530   else {
531     reagent/*copy*/ tmp = current_instruction().ingredients.at(0);
532     canonize(tmp);
533     Bar = tmp.value;
534   }
535   break;
536 }