simgear/nasal/lib.c

   1 #include <math.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <stdarg.h>
   5 #include <string.h>
   6
   7 #ifdef _MSC_VER // sigh...
   8 #define snprintf _snprintf
   9 #define vsnprintf _vsnprintf
  10 #endif
  11
  12 #include "nasal.h"
  13 #include "code.h"
  14
  15 #define NEWSTR(c, s, l) naStr_fromdata(naNewString(c), s, l)
  16 #define NEWCSTR(c, s) NEWSTR(c, s, strlen(s))
  17
  18 // Generic argument error, assumes that the symbol "c" is a naContext,
  19 // and that the __FUNCTION__ string is of the form "f_NASALSYMBOL".
  20 #define ARGERR() \
  21     naRuntimeError(c, "bad/missing argument to %s()", (__FUNCTION__ + 2))
  22
  23 static naRef f_size(naContext c, naRef me, int argc, naRef* args)
  24 {
  25     if(argc == 0) ARGERR();
  26     if(naIsString(args[0])) return naNum(naStr_len(args[0]));
  27     if(naIsVector(args[0])) return naNum(naVec_size(args[0]));
  28     if(naIsHash(args[0])) return naNum(naHash_size(args[0]));
  29     naRuntimeError(c, "object has no size()");
  30     return naNil();
  31 }
  32
  33 static naRef f_keys(naContext c, naRef me, int argc, naRef* args)
  34 {
  35     naRef v, h = argc > 0 ? args[0] : naNil();
  36     if(!naIsHash(h)) ARGERR();
  37     v = naNewVector(c);
  38     naHash_keys(v, h);
  39     return v;
  40 }
  41
  42 static naRef f_append(naContext c, naRef me, int argc, naRef* args)
  43 {
  44     int i;
  45     if(argc < 2 || !naIsVector(args[0])) ARGERR();
  46     for(i=1; i<argc; i++) naVec_append(args[0], args[i]);
  47     return args[0];
  48 }
  49
  50 static naRef f_pop(naContext c, naRef me, int argc, naRef* args)
  51 {
  52     if(argc < 1 || !naIsVector(args[0])) ARGERR();
  53     return naVec_removelast(args[0]);
  54 }
  55
  56 static naRef f_setsize(naContext c, naRef me, int argc, naRef* args)
  57 {
  58     if(argc < 2 || !naIsVector(args[0])) ARGERR();
  59     naVec_setsize(args[0], (int)naNumValue(args[1]).num);
  60     return args[0];
  61 }
  62
  63 static naRef f_subvec(naContext c, naRef me, int argc, naRef* args)
  64 {
  65     int i;
  66     naRef nlen, result, v = args[0];
  67     int len = 0, start = (int)naNumValue(args[1]).num;
  68     if(argc < 2) return naNil();
  69     nlen = argc > 2 ? naNumValue(args[2]) : naNil();
  70     if(!naIsNil(nlen))
  71         len = (int)nlen.num;
  72     if(!naIsVector(v) || start < 0 || start > naVec_size(v) || len < 0)
  73         ARGERR();
  74     if(naIsNil(nlen) || len > naVec_size(v) - start)
  75         len = naVec_size(v) - start;
  76     result = naNewVector(c);
  77     naVec_setsize(result, len);
  78     for(i=0; i<len; i++)
  79         naVec_set(result, i, naVec_get(v, start + i));
  80     return result;
  81 }
  82
  83 static naRef f_delete(naContext c, naRef me, int argc, naRef* args)
  84 {
  85     if(argc < 2 || !naIsHash(args[0])) ARGERR();
  86     naHash_delete(args[0], args[1]);
  87     return args[0];
  88 }
  89
  90 static naRef f_int(naContext c, naRef me, int argc, naRef* args)
  91 {
  92     if(argc > 0) {
  93         naRef n = naNumValue(args[0]);
  94         if(naIsNil(n)) return n;
  95         if(n.num < 0) n.num = -floor(-n.num);
  96         else n.num = floor(n.num);
  97         return n;
  98     } else ARGERR();
  99     return naNil();
 100 }
 101
 102 static naRef f_num(naContext c, naRef me, int argc, naRef* args)
 103 {
 104     return argc > 0 ? naNumValue(args[0]) : naNil();
 105 }
 106
 107 static naRef f_streq(naContext c, naRef me, int argc, naRef* args)
 108 {
 109     return argc > 1 ? naNum(naStrEqual(args[0], args[1])) : naNil();
 110 }
 111
 112 static naRef f_cmp(naContext c, naRef me, int argc, naRef* args)
 113 {
 114     char *a, *b;
 115     int i, alen, blen;
 116     if(argc < 2 || !naIsString(args[0]) || !naIsString(args[1]))
 117         ARGERR();
 118     a = naStr_data(args[0]);
 119     alen = naStr_len(args[0]);
 120     b = naStr_data(args[1]);
 121     blen = naStr_len(args[1]);
 122     for(i=0; i<alen && i<blen; i++) {
 123         int diff = a[i] - b[i];
 124         if(diff) return naNum(diff < 0 ? -1 : 1);
 125     }
 126     return naNum(alen == blen ? 0 : (alen < blen ? -1 : 1));
 127 }
 128
 129 static naRef f_substr(naContext c, naRef me, int argc, naRef* args)
 130 {
 131     int start, len, srclen;
 132     naRef src = argc > 0 ? args[0] : naNil();
 133     naRef startr = argc > 1 ? naNumValue(args[1]) : naNil();
 134     naRef lenr = argc > 2 ? naNumValue(args[2]) : naNil();
 135     if(!naIsString(src)) ARGERR();
 136     if(naIsNil(startr) || !naIsNum(startr)) ARGERR();
 137     if(!naIsNil(lenr) && !naIsNum(lenr)) ARGERR();
 138     srclen = naStr_len(src);
 139     start = (int)startr.num;
 140     len = naIsNum(lenr) ? (int)lenr.num : (srclen - start);
 141     if(start < 0) start += srclen;
 142     if(start < 0) start = len = 0;
 143     if(start >= srclen) start = len = 0;
 144     if(len < 0) len = 0;
 145     if(len > srclen - start) len = srclen - start;
 146     return naStr_substr(naNewString(c), src, start, len);
 147 }
 148
 149 static naRef f_chr(naContext c, naRef me, int argc, naRef* args)
 150 {
 151     char chr[1];
 152     naRef cr = argc > 0 ? naNumValue(args[0]) : naNil();
 153     if(IS_NIL(cr)) ARGERR();
 154     chr[0] = (char)cr.num;
 155     return NEWSTR(c, chr, 1);
 156 }
 157
 158 static naRef f_contains(naContext c, naRef me, int argc, naRef* args)
 159 {
 160     naRef hash = argc > 0 ? args[0] : naNil();
 161     naRef key = argc > 1 ? args[1] : naNil();
 162     if(naIsNil(hash) || naIsNil(key)) ARGERR();
 163     if(!naIsHash(hash)) return naNil();
 164     return naHash_get(hash, key, &key) ? naNum(1) : naNum(0);
 165 }
 166
 167 static naRef f_typeof(naContext c, naRef me, int argc, naRef* args)
 168 {
 169     naRef r = argc > 0 ? args[0] : naNil();
 170     char* t = "unknown";
 171     if(naIsNil(r)) t = "nil";
 172     else if(naIsNum(r)) t = "scalar";
 173     else if(naIsString(r)) t = "scalar";
 174     else if(naIsVector(r)) t = "vector";
 175     else if(naIsHash(r)) t = "hash";
 176     else if(naIsFunc(r)) t = "func";
 177     else if(naIsGhost(r)) t = "ghost";
 178     return NEWCSTR(c, t);
 179 }
 180
 181 static naRef f_ghosttype(naContext c, naRef me, int argc, naRef* args)
 182 {
 183     naRef g = argc > 0 ? args[0] : naNil();
 184     if(!naIsGhost(g)) return naNil();
 185     if(naGhost_type(g)->name) {
 186         return NEWCSTR(c, (char*)naGhost_type(g)->name);
 187     } else {
 188         char buf[32];
 189         sprintf(buf, "%p", naGhost_type(g));
 190         return NEWCSTR(c, buf);
 191     }
 192 }
 193
 194 static naRef f_compile(naContext c, naRef me, int argc, naRef* args)
 195 {
 196     int errLine;
 197     naRef script, code, fname;
 198     script = argc > 0 ? args[0] : naNil();
 199     fname = argc > 1 ? args[1] : NEWCSTR(c, "<compile>");
 200     if(!naIsString(script) || !naIsString(fname)) return naNil();
 201     code = naParseCode(c, fname, 1,
 202                        naStr_data(script), naStr_len(script), &errLine);
 203     if(naIsNil(code)) {
 204         char buf[256];
 205         snprintf(buf, sizeof(buf), "Parse error: %s at line %d",
 206                  naGetError(c), errLine);
 207         c->dieArg = NEWCSTR(c, buf);
 208         naRuntimeError(c, "__die__");
 209     }
 210     return naBindToContext(c, code);
 211 }
 212
 213 // FIXME: need a place to save the current IP when we get an error so
 214 // that it can be reset if we get a die()/naRethrowError() situation
 215 // later.  Right now, the IP on the stack trace is the line of the
 216 // die() call, when it should be this one...
 217 static naRef f_call(naContext c, naRef me, int argc, naRef* args)
 218 {
 219     naContext subc;
 220     naRef callargs, callme, callns, result;
 221     struct VecRec* vr;
 222     callargs = argc > 1 ? args[1] : naNil();
 223     callme = argc > 2 ? args[2] : naNil(); // Might be nil, that's OK
 224     callns = argc > 3 ? args[3] : naNil(); // ditto
 225     if(!IS_HASH(callme)) callme = naNil();
 226     if(!IS_HASH(callns)) callns = naNil();
 227     if(!IS_FUNC(args[0]) || (!IS_NIL(callargs) && !IS_VEC(callargs)))
 228         ARGERR();
 229
 230     subc = naSubContext(c);
 231     vr = IS_NIL(callargs) ? 0 : PTR(callargs).vec->rec;
 232     result = naCall(subc, args[0], vr ? vr->size : 0, vr ? vr->array : 0,
 233                     callme, callns);
 234     if(!naGetError(subc)) {
 235         naFreeContext(subc);
 236         return result;
 237     }
 238
 239     // Error handling. Note that we don't free the subcontext after an
 240     // error, in case the user re-throws the same error or calls
 241     // naContinue()
 242     if(argc <= 2 || !IS_VEC(args[argc-1])) {
 243         naRethrowError(subc);
 244     } else {
 245         int i, sd;
 246         naRef errv = args[argc-1];
 247         if(!IS_NIL(subc->dieArg)) naVec_append(errv, subc->dieArg);
 248         else naVec_append(errv, NEWCSTR(subc, naGetError(subc)));
 249         sd = naStackDepth(subc);
 250         for(i=0; i<sd; i++) {
 251             naVec_append(errv, naGetSourceFile(subc, i));
 252             naVec_append(errv, naNum(naGetLine(subc, i)));
 253         }
 254     }
 255     return naNil();
 256 }
 257
 258 static naRef f_die(naContext c, naRef me, int argc, naRef* args)
 259 {
 260     naRef darg = argc > 0 ? args[0] : naNil();
 261     if(!naIsNil(darg) && c->callChild && IDENTICAL(c->callChild->dieArg, darg))
 262         naRethrowError(c->callChild);
 263     c->dieArg = darg;
 264     naRuntimeError(c, "__die__");
 265     return naNil(); // never executes
 266 }
 267
 268 // Wrapper around vsnprintf, iteratively increasing the buffer size
 269 // until it fits.  Returned buffer should be freed by the caller.
 270 static char* dosprintf(char* f, ...)
 271 {
 272     char* buf;
 273     va_list va;
 274     int olen, len = 16;
 275     while(1) {
 276         buf = naAlloc(len);
 277         va_start(va, f);
 278         olen = vsnprintf(buf, len, f, va);
 279         if(olen >= 0 && olen < len) {
 280             va_end(va);
 281             return buf;
 282         }
 283         va_end(va);
 284         naFree(buf);
 285         len *= 2;
 286     }
 287 }
 288
 289 // Inspects a printf format string f, and finds the next "%..." format
 290 // specifier.  Stores the start of the specifier in out, the length in
 291 // len, and the type in type.  Returns a pointer to the remainder of
 292 // the format string, or 0 if no format string was found.  Recognizes
 293 // all of ANSI C's syntax except for the "length modifier" feature.
 294 // Note: this does not validate the format character returned in
 295 // "type". That is the caller's job.
 296 static char* nextFormat(naContext c, char* f, char** out, int* len, char* type)
 297 {
 298     // Skip to the start of the format string
 299     while(*f && *f != '%') f++;
 300     if(!*f) return 0;
 301     *out = f++;
 302
 303     while(*f && (*f=='-' || *f=='+' || *f==' ' || *f=='0' || *f=='#')) f++;
 304
 305     // Test for duplicate flags.  This is pure pedantry and could
 306     // be removed on all known platforms, but just to be safe...
 307     {   char *p1, *p2;
 308         for(p1 = *out + 1; p1 < f; p1++)
 309             for(p2 = p1+1; p2 < f; p2++)
 310                 if(*p1 == *p2)
 311                     naRuntimeError(c, "duplicate flag in format string"); }
 312
 313     while(*f && *f >= '0' && *f <= '9') f++;
 314     if(*f && *f == '.') f++;
 315     while(*f && *f >= '0' && *f <= '9') f++;
 316     if(!*f) naRuntimeError(c, "invalid format string");
 317
 318     *type = *f++;
 319     *len = f - *out;
 320     return f;
 321 }
 322
 323 #define ERR(m) naRuntimeError(c, m)
 324 #define APPEND(r) result = naStr_concat(naNewString(c), result, r)
 325 static naRef f_sprintf(naContext c, naRef me, int argc, naRef* args)
 326 {
 327     char t, nultmp, *fstr, *next, *fout=0, *s;
 328     int flen, argn=1;
 329     naRef format, arg, result = naNewString(c);
 330
 331     if(argc < 1) ERR("not enough arguments to sprintf()");
 332     format = naStringValue(c, argc > 0 ? args[0] : naNil());
 333     if(naIsNil(format)) ERR("bad format string in sprintf()");
 334     s = naStr_data(format);
 335
 336     while((next = nextFormat(c, s, &fstr, &flen, &t))) {
 337         APPEND(NEWSTR(c, s, fstr-s)); // stuff before the format string
 338         if(flen == 2 && fstr[1] == '%') {
 339             APPEND(NEWSTR(c, "%", 1));
 340             s = next;
 341             continue;
 342         }
 343         if(argn >= argc) ERR("not enough arguments to sprintf()");
 344         arg = args[argn++];
 345         nultmp = fstr[flen]; // sneaky nul termination...
 346         fstr[flen] = 0;
 347         if(t == 's') {
 348             arg = naStringValue(c, arg);
 349             if(naIsNil(arg)) fout = dosprintf(fstr, "nil");
 350             else             fout = dosprintf(fstr, naStr_data(arg));
 351         } else {
 352             arg = naNumValue(arg);
 353             if(naIsNil(arg))
 354                 fout = dosprintf(fstr, "nil");
 355             else if(t=='d' || t=='i' || t=='c')
 356                 fout = dosprintf(fstr, (int)naNumValue(arg).num);
 357             else if(t=='o' || t=='u' || t=='x' || t=='X')
 358                 fout = dosprintf(fstr, (unsigned int)naNumValue(arg).num);
 359             else if(t=='e' || t=='E' || t=='f' || t=='F' || t=='g' || t=='G')
 360                 fout = dosprintf(fstr, naNumValue(arg).num);
 361             else
 362                 ERR("invalid sprintf format type");
 363         }
 364         fstr[flen] = nultmp;
 365         APPEND(NEWSTR(c, fout, strlen(fout)));
 366         naFree(fout);
 367         s = next;
 368     }
 369     APPEND(NEWSTR(c, s, strlen(s)));
 370     return result;
 371 }
 372
 373 // FIXME: needs to honor subcontext list
 374 static naRef f_caller(naContext c, naRef me, int argc, naRef* args)
 375 {
 376     int fidx;
 377     struct Frame* frame;
 378     naRef result, fr = argc ? naNumValue(args[0]) : naNum(1);
 379     if(IS_NIL(fr)) ARGERR();
 380     fidx = (int)fr.num;
 381     if(fidx > c->fTop - 1) return naNil();
 382     frame = &c->fStack[c->fTop - 1 - fidx];
 383     result = naNewVector(c);
 384     naVec_append(result, frame->locals);
 385     naVec_append(result, frame->func);
 386     naVec_append(result, PTR(PTR(frame->func).func->code).code->srcFile);
 387     naVec_append(result, naNum(naGetLine(c, fidx)));
 388     return result;
 389 }
 390
 391 static naRef f_closure(naContext c, naRef me, int argc, naRef* args)
 392 {
 393     int i;
 394     struct naFunc* f;
 395     naRef func = argc > 0 ? args[0] : naNil();
 396     naRef idx = argc > 1 ? naNumValue(args[1]) : naNum(0);
 397     if(!IS_FUNC(func) || IS_NIL(idx)) ARGERR();
 398     i = (int)idx.num;
 399     f = PTR(func).func;
 400     while(i > 0 && f) { i--; f = PTR(f->next).func; }
 401     if(!f) return naNil();
 402     return f->namespace;
 403 }
 404
 405 static int match(unsigned char* a, unsigned char* b, int l)
 406 {
 407     int i;
 408     for(i=0; i<l; i++) if(a[i] != b[i]) return 0;
 409     return 1;
 410 }
 411
 412 static int find(unsigned char* a, int al, unsigned char* s, int sl, int start)
 413 {
 414     int i;
 415     if(al == 0) return 0;
 416     for(i=start; i<sl-al+1; i++) if(match(a, s+i, al)) return i;
 417     return -1;
 418 }
 419
 420 static naRef f_find(naContext c, naRef me, int argc, naRef* args)
 421 {
 422     int start = 0;
 423     if(argc < 2 || !IS_STR(args[0]) || !IS_STR(args[1])) ARGERR();
 424     if(argc > 2) start = (int)(naNumValue(args[2]).num);
 425     return naNum(find(PTR(args[0]).str->data, PTR(args[0]).str->len,
 426                       PTR(args[1]).str->data, PTR(args[1]).str->len,
 427                       start));
 428 }
 429
 430 static naRef f_split(naContext c, naRef me, int argc, naRef* args)
 431 {
 432     int sl, dl, i;
 433     char *s, *d, *s0;
 434     naRef result;
 435     if(argc < 2 || !IS_STR(args[0]) || !IS_STR(args[1])) ARGERR();
 436     d = naStr_data(args[0]); dl = naStr_len(args[0]);
 437     s = naStr_data(args[1]); sl = naStr_len(args[1]);
 438     result = naNewVector(c);
 439     if(dl == 0) { // special case zero-length delimiter
 440         for(i=0; i<sl; i++) naVec_append(result, NEWSTR(c, s+i, 1));
 441         return result;
 442     }
 443     s0 = s;
 444     for(i=0; i <= sl-dl; i++) {
 445         if(match((unsigned char*)(s+i), (unsigned char*)d, dl)) {
 446             naVec_append(result, NEWSTR(c, s0, s+i-s0));
 447             s0 = s + i + dl;
 448             i += dl - 1;
 449         }
 450     }
 451     if(s0 - s <= sl) naVec_append(result, NEWSTR(c, s0, s+sl-s0));
 452     return result;
 453 }
 454
 455 // This is a comparatively weak RNG, based on the C library's rand()
 456 // function, which is usually not threadsafe and often of limited
 457 // precision.  The 5x loop guarantees that we get a full double worth
 458 // of precision even for 15 bit (Win32...) rand() implementations.
 459 static naRef f_rand(naContext c, naRef me, int argc, naRef* args)
 460 {
 461     int i;
 462     double r = 0;
 463     if(argc) {
 464         if(!IS_NUM(args[0])) naRuntimeError(c, "rand() seed not number");
 465         srand((unsigned int)args[0].num);
 466         return naNil();
 467     }
 468     for(i=0; i<5; i++) r = (r + rand()) * (1.0/(RAND_MAX+1.0));
 469     return naNum(r);
 470 }
 471
 472 static naRef f_bind(naContext c, naRef me, int argc, naRef* args)
 473 {
 474     naRef func = argc > 0 ? args[0] : naNil();
 475     naRef hash = argc > 1 ? args[1] : naNewHash(c);
 476     naRef next = argc > 2 ? args[2] : naNil();
 477     if(!IS_FUNC(func) || (!IS_NIL(next) && !IS_FUNC(next)) || !IS_HASH(hash))
 478         ARGERR();
 479     func = naNewFunc(c, PTR(func).func->code);
 480     PTR(func).func->namespace = hash;
 481     PTR(func).func->next = next;
 482     return func;
 483 }
 484
 485 /* We use the "SortRec" gadget for two reasons: first, because ANSI
 486  * qsort() doesn't give us a mechanism for passing a "context" pointer
 487  * to the comparison routine we have to store one in every sorted
 488  * record.  Second, using an index into the original vector here
 489  * allows us to make the sort stable in the event of a zero returned
 490  * from the Nasal comparison function. */
 491 struct SortData { naContext ctx, subc; struct SortRec* recs;
 492                   naRef* elems; int n; naRef fn; };
 493 struct SortRec { struct SortData* sd; int i; };
 494
 495 static int sortcmp(struct SortRec* a, struct SortRec* b)
 496 {
 497     struct SortData* sd = a->sd;
 498     naRef args[2], d;
 499     args[0] = sd->elems[a->i];
 500     args[1] = sd->elems[b->i];
 501     d = naCall(sd->subc, sd->fn, 2, args, naNil(), naNil());
 502     if(naGetError(sd->subc)) {
 503         naFree(sd->recs);
 504         naRethrowError(sd->subc);
 505     } else if(!naIsNum(d = naNumValue(d))) {
 506         naFree(sd->recs);
 507         naRuntimeError(sd->ctx, "sort() comparison returned non-number");
 508     }
 509     return (d.num > 0) ? 1 : ((d.num < 0) ? -1 : (a->i - b->i));
 510 }
 511
 512 static naRef f_sort(naContext c, naRef me, int argc, naRef* args)
 513 {
 514     int i;
 515     struct SortData sd;
 516     naRef out;
 517     if(argc != 2 || !naIsVector(args[0]) || !naIsFunc(args[1]))
 518         naRuntimeError(c, "bad/missing argument to sort()");
 519     sd.subc = naSubContext(c);
 520     if(!PTR(args[0]).vec->rec) return naNewVector(c);
 521     sd.elems = PTR(args[0]).vec->rec->array;
 522     sd.n = PTR(args[0]).vec->rec->size;
 523     sd.fn = args[1];
 524     sd.recs = naAlloc(sizeof(struct SortRec) * sd.n);
 525     for(i=0; i<sd.n; i++) {
 526         sd.recs[i].sd = &sd;
 527         sd.recs[i].i = i;
 528     }
 529     qsort(sd.recs, sd.n, sizeof(sd.recs[0]),
 530           (int(*)(const void*,const void*))sortcmp);
 531     out = naNewVector(c);
 532     naVec_setsize(out, sd.n);
 533     for(i=0; i<sd.n; i++)
 534         PTR(out).vec->rec->array[i] = sd.elems[sd.recs[i].i];
 535     naFree(sd.recs);
 536     naFreeContext(sd.subc);
 537     return out;
 538 }
 539
 540 static naCFuncItem funcs[] = {
 541     { "size", f_size },
 542     { "keys", f_keys },
 543     { "append", f_append },
 544     { "pop", f_pop },
 545     { "setsize", f_setsize },
 546     { "subvec", f_subvec },
 547     { "delete", f_delete },
 548     { "int", f_int },
 549     { "num", f_num },
 550     { "streq", f_streq },
 551     { "cmp", f_cmp },
 552     { "substr", f_substr },
 553     { "chr", f_chr },
 554     { "contains", f_contains },
 555     { "typeof", f_typeof },
 556     { "ghosttype", f_ghosttype },
 557     { "compile", f_compile },
 558     { "call", f_call },
 559     { "die", f_die },
 560     { "sprintf", f_sprintf },
 561     { "caller", f_caller },
 562     { "closure", f_closure },
 563     { "find", f_find },
 564     { "split", f_split },
 565     { "rand", f_rand },
 566     { "bind", f_bind },
 567     { "sort", f_sort },
 568     { 0 }
 569 };
 570
 571 naRef naInit_std(naContext c)
 572 {
 573     return naGenLib(c, funcs);
 574 }