1323530Savg/* 2323530Savg** $Id: lobject.c,v 2.58.1.1 2013/04/12 18:48:47 roberto Exp $ 3323530Savg** Some generic functions over Lua objects 4323530Savg** See Copyright Notice in lua.h 5323530Savg*/ 6323530Savg 7323530Savg#include <sys/zfs_context.h> 8323530Savg 9323530Savg#define lobject_c 10323530Savg#define LUA_CORE 11323530Savg 12323530Savg#include "lua.h" 13323530Savg 14323530Savg#include "lctype.h" 15323530Savg#include "ldebug.h" 16323530Savg#include "ldo.h" 17323530Savg#include "lmem.h" 18323530Savg#include "lobject.h" 19323530Savg#include "lstate.h" 20323530Savg#include "lstring.h" 21323530Savg#include "lvm.h" 22323530Savg 23323530Savg 24323530Savg 25323530SavgLUAI_DDEF const TValue luaO_nilobject_ = {NILCONSTANT}; 26323530Savg 27323530Savg 28323530Savg/* 29323530Savg** converts an integer to a "floating point byte", represented as 30323530Savg** (eeeeexxx), where the real value is (1xxx) * 2^(eeeee - 1) if 31323530Savg** eeeee != 0 and (xxx) otherwise. 32323530Savg*/ 33323530Savgint luaO_int2fb (unsigned int x) { 34323530Savg int e = 0; /* exponent */ 35323530Savg if (x < 8) return x; 36323530Savg while (x >= 0x10) { 37323530Savg x = (x+1) >> 1; 38323530Savg e++; 39323530Savg } 40323530Savg return ((e+1) << 3) | (cast_int(x) - 8); 41323530Savg} 42323530Savg 43323530Savg 44323530Savg/* converts back */ 45323530Savgint luaO_fb2int (int x) { 46323530Savg int e = (x >> 3) & 0x1f; 47323530Savg if (e == 0) return x; 48323530Savg else return ((x & 7) + 8) << (e - 1); 49323530Savg} 50323530Savg 51323530Savg 52323530Savgint luaO_ceillog2 (unsigned int x) { 53323530Savg static const lu_byte log_2[256] = { 54323530Savg 0,1,2,2,3,3,3,3,4,4,4,4,4,4,4,4,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5, 55323530Savg 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6, 56323530Savg 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, 57323530Savg 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, 58323530Savg 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 59323530Savg 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 60323530Savg 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 61323530Savg 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8 62323530Savg }; 63323530Savg int l = 0; 64323530Savg x--; 65323530Savg while (x >= 256) { l += 8; x >>= 8; } 66323530Savg return l + log_2[x]; 67323530Savg} 68323530Savg 69323530Savg 70323530Savglua_Number luaO_arith (int op, lua_Number v1, lua_Number v2) { 71323530Savg switch (op) { 72323530Savg case LUA_OPADD: return luai_numadd(NULL, v1, v2); 73323530Savg case LUA_OPSUB: return luai_numsub(NULL, v1, v2); 74323530Savg case LUA_OPMUL: return luai_nummul(NULL, v1, v2); 75323530Savg case LUA_OPDIV: return luai_numdiv(NULL, v1, v2); 76323530Savg case LUA_OPMOD: return luai_nummod(NULL, v1, v2); 77323530Savg case LUA_OPPOW: return luai_numpow(NULL, v1, v2); 78323530Savg case LUA_OPUNM: return luai_numunm(NULL, v1); 79323530Savg default: lua_assert(0); return 0; 80323530Savg } 81323530Savg} 82323530Savg 83323530Savg 84323530Savgint luaO_hexavalue (int c) { 85323530Savg if (lisdigit(c)) return c - '0'; 86323530Savg else return ltolower(c) - 'a' + 10; 87323530Savg} 88323530Savg 89323530Savg 90323530Savg#if !defined(lua_strx2number) 91323530Savg 92323530Savg 93323530Savg 94323530Savgstatic int isneg (const char **s) { 95323530Savg if (**s == '-') { (*s)++; return 1; } 96323530Savg else if (**s == '+') (*s)++; 97323530Savg return 0; 98323530Savg} 99323530Savg 100323530Savg 101323530Savgstatic lua_Number readhexa (const char **s, lua_Number r, int *count) { 102323530Savg for (; lisxdigit(cast_uchar(**s)); (*s)++) { /* read integer part */ 103323530Savg r = (r * cast_num(16.0)) + cast_num(luaO_hexavalue(cast_uchar(**s))); 104323530Savg (*count)++; 105323530Savg } 106323530Savg return r; 107323530Savg} 108323530Savg 109323530Savg 110323530Savg/* 111323530Savg** convert an hexadecimal numeric string to a number, following 112323530Savg** C99 specification for 'strtod' 113323530Savg*/ 114323530Savgstatic lua_Number lua_strx2number (const char *s, char **endptr) { 115323530Savg lua_Number r = 0.0; 116323530Savg int e = 0, i = 0; 117323530Savg int neg = 0; /* 1 if number is negative */ 118323530Savg *endptr = cast(char *, s); /* nothing is valid yet */ 119323530Savg while (lisspace(cast_uchar(*s))) s++; /* skip initial spaces */ 120323530Savg neg = isneg(&s); /* check signal */ 121323530Savg if (!(*s == '0' && (*(s + 1) == 'x' || *(s + 1) == 'X'))) /* check '0x' */ 122323530Savg return 0.0; /* invalid format (no '0x') */ 123323530Savg s += 2; /* skip '0x' */ 124323530Savg r = readhexa(&s, r, &i); /* read integer part */ 125323530Savg if (*s == '.') { 126323530Savg s++; /* skip dot */ 127323530Savg r = readhexa(&s, r, &e); /* read fractional part */ 128323530Savg } 129323530Savg if (i == 0 && e == 0) 130323530Savg return 0.0; /* invalid format (no digit) */ 131323530Savg e *= -4; /* each fractional digit divides value by 2^-4 */ 132323530Savg *endptr = cast(char *, s); /* valid up to here */ 133323530Savg if (*s == 'p' || *s == 'P') { /* exponent part? */ 134323530Savg int exp1 = 0; 135323530Savg int neg1; 136323530Savg s++; /* skip 'p' */ 137323530Savg neg1 = isneg(&s); /* signal */ 138323530Savg if (!lisdigit(cast_uchar(*s))) 139323530Savg goto ret; /* must have at least one digit */ 140323530Savg while (lisdigit(cast_uchar(*s))) /* read exponent */ 141323530Savg exp1 = exp1 * 10 + *(s++) - '0'; 142323530Savg if (neg1) exp1 = -exp1; 143323530Savg e += exp1; 144323530Savg } 145323530Savg *endptr = cast(char *, s); /* valid up to here */ 146323530Savg ret: 147323530Savg if (neg) r = -r; 148323530Savg return (r * (1 << e)); 149323530Savg} 150323530Savg 151323530Savg#endif 152323530Savg 153323530Savg 154323530Savgint luaO_str2d (const char *s, size_t len, lua_Number *result) { 155323530Savg char *endptr; 156323530Savg if (strpbrk(s, "nN")) /* reject 'inf' and 'nan' */ 157323530Savg return 0; 158323530Savg else if (strpbrk(s, "xX")) /* hexa? */ 159323530Savg *result = lua_strx2number(s, &endptr); 160323530Savg else 161323530Savg *result = lua_str2number(s, &endptr); 162323530Savg if (endptr == s) return 0; /* nothing recognized */ 163323530Savg while (lisspace(cast_uchar(*endptr))) endptr++; 164323530Savg return (endptr == s + len); /* OK if no trailing characters */ 165323530Savg} 166323530Savg 167323530Savg 168323530Savg 169323530Savgstatic void pushstr (lua_State *L, const char *str, size_t l) { 170323530Savg setsvalue2s(L, L->top++, luaS_newlstr(L, str, l)); 171323530Savg} 172323530Savg 173323530Savg 174323530Savg/* this function handles only `%d', `%c', %f, %p, and `%s' formats */ 175323530Savgconst char *luaO_pushvfstring (lua_State *L, const char *fmt, va_list argp) { 176323530Savg int n = 0; 177323530Savg for (;;) { 178323530Savg const char *e = strchr(fmt, '%'); 179323530Savg if (e == NULL) break; 180323530Savg luaD_checkstack(L, 2); /* fmt + item */ 181323530Savg pushstr(L, fmt, e - fmt); 182323530Savg switch (*(e+1)) { 183323530Savg case 's': { 184323530Savg const char *s = va_arg(argp, char *); 185323530Savg if (s == NULL) s = "(null)"; 186323530Savg pushstr(L, s, strlen(s)); 187323530Savg break; 188323530Savg } 189323530Savg case 'c': { 190323530Savg char buff; 191323530Savg buff = cast(char, va_arg(argp, int)); 192323530Savg pushstr(L, &buff, 1); 193323530Savg break; 194323530Savg } 195323530Savg case 'd': { 196323530Savg setnvalue(L->top++, cast_num(va_arg(argp, int))); 197323530Savg break; 198323530Savg } 199323530Savg case 'f': { 200323530Savg setnvalue(L->top++, cast_num(va_arg(argp, l_uacNumber))); 201323530Savg break; 202323530Savg } 203323530Savg case 'p': { 204323530Savg char buff[4*sizeof(void *) + 8]; /* should be enough space for a `%p' */ 205323530Savg int l = lcompat_sprintf(buff, "%p", va_arg(argp, void *)); 206323530Savg pushstr(L, buff, l); 207323530Savg break; 208323530Savg } 209323530Savg case '%': { 210323530Savg pushstr(L, "%", 1); 211323530Savg break; 212323530Savg } 213323530Savg default: { 214323530Savg luaG_runerror(L, 215323530Savg "invalid option " LUA_QL("%%%c") " to " LUA_QL("lua_pushfstring"), 216323530Savg *(e + 1)); 217323530Savg } 218323530Savg } 219323530Savg n += 2; 220323530Savg fmt = e+2; 221323530Savg } 222323530Savg luaD_checkstack(L, 1); 223323530Savg pushstr(L, fmt, strlen(fmt)); 224323530Savg if (n > 0) luaV_concat(L, n + 1); 225323530Savg return svalue(L->top - 1); 226323530Savg} 227323530Savg 228323530Savg 229323530Savgconst char *luaO_pushfstring (lua_State *L, const char *fmt, ...) { 230323530Savg const char *msg; 231323530Savg va_list argp; 232323530Savg va_start(argp, fmt); 233323530Savg msg = luaO_pushvfstring(L, fmt, argp); 234323530Savg va_end(argp); 235323530Savg return msg; 236323530Savg} 237323530Savg 238323530Savg 239323530Savg/* number of chars of a literal string without the ending \0 */ 240323530Savg#define LL(x) (sizeof(x)/sizeof(char) - 1) 241323530Savg 242323530Savg#define RETS "..." 243323530Savg#define PRE "[string \"" 244323530Savg#define POS "\"]" 245323530Savg 246323530Savg#define addstr(a,b,l) ( memcpy(a,b,(l) * sizeof(char)), a += (l) ) 247323530Savg 248323530Savgvoid luaO_chunkid (char *out, const char *source, size_t bufflen) { 249323530Savg size_t l = strlen(source); 250323530Savg if (*source == '=') { /* 'literal' source */ 251323530Savg if (l <= bufflen) /* small enough? */ 252323530Savg memcpy(out, source + 1, l * sizeof(char)); 253323530Savg else { /* truncate it */ 254323530Savg addstr(out, source + 1, bufflen - 1); 255323530Savg *out = '\0'; 256323530Savg } 257323530Savg } 258323530Savg else if (*source == '@') { /* file name */ 259323530Savg if (l <= bufflen) /* small enough? */ 260323530Savg memcpy(out, source + 1, l * sizeof(char)); 261323530Savg else { /* add '...' before rest of name */ 262323530Savg addstr(out, RETS, LL(RETS)); 263323530Savg bufflen -= LL(RETS); 264323530Savg memcpy(out, source + 1 + l - bufflen, bufflen * sizeof(char)); 265323530Savg } 266323530Savg } 267323530Savg else { /* string; format as [string "source"] */ 268323530Savg const char *nl = strchr(source, '\n'); /* find first new line (if any) */ 269323530Savg addstr(out, PRE, LL(PRE)); /* add prefix */ 270323530Savg bufflen -= LL(PRE RETS POS) + 1; /* save space for prefix+suffix+'\0' */ 271323530Savg if (l < bufflen && nl == NULL) { /* small one-line source? */ 272323530Savg addstr(out, source, l); /* keep it */ 273323530Savg } 274323530Savg else { 275323530Savg if (nl != NULL) l = nl - source; /* stop at first newline */ 276323530Savg if (l > bufflen) l = bufflen; 277323530Savg addstr(out, source, l); 278323530Savg addstr(out, RETS, LL(RETS)); 279323530Savg } 280323530Savg memcpy(out, POS, (LL(POS) + 1) * sizeof(char)); 281323530Savg } 282323530Savg} 283323530Savg 284