| 1 | /* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- | |
| 2 | * | |
| 3 | * ***** BEGIN LICENSE BLOCK ***** | |
| 4 | * Version: MPL 1.1/GPL 2.0/LGPL 2.1 | |
| 5 | * | |
| 6 | * The contents of this file are subject to the Mozilla Public License Version | |
| 7 | * 1.1 (the "License"); you may not use this file except in compliance with | |
| 8 | * the License. You may obtain a copy of the License at | |
| 9 | * http://www.mozilla.org/MPL/ | |
| 10 | * | |
| 11 | * Software distributed under the License is distributed on an "AS IS" basis, | |
| 12 | * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License | |
| 13 | * for the specific language governing rights and limitations under the | |
| 14 | * License. | |
| 15 | * | |
| 16 | * The Original Code is Mozilla Communicator client code, released | |
| 17 | * March 31, 1998. | |
| 18 | * | |
| 19 | * The Initial Developer of the Original Code is | |
| 20 | * Netscape Communications Corporation. | |
| 21 | * Portions created by the Initial Developer are Copyright (C) 1998 | |
| 22 | * the Initial Developer. All Rights Reserved. | |
| 23 | * | |
| 24 | * Contributor(s): | |
| 25 | * | |
| 26 | * Alternatively, the contents of this file may be used under the terms of | |
| 27 | * either of the GNU General Public License Version 2 or later (the "GPL"), | |
| 28 | * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), | |
| 29 | * in which case the provisions of the GPL or the LGPL are applicable instead | |
| 30 | * of those above. If you wish to allow use of your version of this file only | |
| 31 | * under the terms of either the GPL or the LGPL, and not to allow others to | |
| 32 | * use your version of this file under the terms of the MPL, indicate your | |
| 33 | * decision by deleting the provisions above and replace them with the notice | |
| 34 | * and other provisions required by the GPL or the LGPL. If you do not delete | |
| 35 | * the provisions above, a recipient may use your version of this file under | |
| 36 | * the terms of any one of the MPL, the GPL or the LGPL. | |
| 37 | * | |
| 38 | * ***** END LICENSE BLOCK ***** */ | |
| 39 | #include "jsstddef.h" | |
| 40 | #include "jsconfig.h" | |
| 41 | ||
| 42 | #if JS_HAS_XDR | |
| 43 | ||
| 44 | #include <string.h> | |
| 45 | #include "jstypes.h" | |
| 46 | #include "jsutil.h" /* Added by JSIFY */ | |
| 47 | #include "jsdhash.h" | |
| 48 | #include "jsprf.h" | |
| 49 | #include "jsapi.h" | |
| 50 | #include "jscntxt.h" | |
| 51 | #include "jsnum.h" | |
| 52 | #include "jsobj.h" /* js_XDRObject */ | |
| 53 | #include "jsscript.h" /* js_XDRScript */ | |
| 54 | #include "jsstr.h" | |
| 55 | #include "jsxdrapi.h" | |
| 56 | ||
| 57 | #ifdef DEBUG | |
| 58 | #define DBG(x) x | |
| 59 | #else | |
| 60 | #define DBG(x) ((void)0) | |
| 61 | #endif | |
| 62 | ||
| 63 | typedef struct JSXDRMemState { | |
| 64 | JSXDRState state; | |
| 65 | char *base; | |
| 66 | uint32 count; | |
| 67 | uint32 limit; | |
| 68 | } JSXDRMemState; | |
| 69 | ||
| 70 | #define MEM_BLOCK 8192 | |
| 71 | #define MEM_PRIV(xdr) ((JSXDRMemState *)(xdr)) | |
| 72 | ||
| 73 | #define MEM_BASE(xdr) (MEM_PRIV(xdr)->base) | |
| 74 | #define MEM_COUNT(xdr) (MEM_PRIV(xdr)->count) | |
| 75 | #define MEM_LIMIT(xdr) (MEM_PRIV(xdr)->limit) | |
| 76 | ||
| 77 | #define MEM_LEFT(xdr, bytes) \ | |
| 78 | JS_BEGIN_MACRO \ | |
| 79 | if ((xdr)->mode == JSXDR_DECODE && \ | |
| 80 | MEM_COUNT(xdr) + bytes > MEM_LIMIT(xdr)) { \ | |
| 81 | JS_ReportErrorNumber((xdr)->cx, js_GetErrorMessage, NULL, \ | |
| 82 | JSMSG_END_OF_DATA); \ | |
| 83 | return 0; \ | |
| 84 | } \ | |
| 85 | JS_END_MACRO | |
| 86 | ||
| 87 | #define MEM_NEED(xdr, bytes) \ | |
| 88 | JS_BEGIN_MACRO \ | |
| 89 | if ((xdr)->mode == JSXDR_ENCODE) { \ | |
| 90 | if (MEM_LIMIT(xdr) && \ | |
| 91 | MEM_COUNT(xdr) + bytes > MEM_LIMIT(xdr)) { \ | |
| 92 | uint32 limit_ = JS_ROUNDUP(MEM_COUNT(xdr) + bytes, MEM_BLOCK);\ | |
| 93 | void *data_ = JS_realloc((xdr)->cx, MEM_BASE(xdr), limit_); \ | |
| 94 | if (!data_) \ | |
| 95 | return 0; \ | |
| 96 | MEM_BASE(xdr) = data_; \ | |
| 97 | MEM_LIMIT(xdr) = limit_; \ | |
| 98 | } \ | |
| 99 | } else { \ | |
| 100 | MEM_LEFT(xdr, bytes); \ | |
| 101 | } \ | |
| 102 | JS_END_MACRO | |
| 103 | ||
| 104 | #define MEM_DATA(xdr) ((void *)(MEM_BASE(xdr) + MEM_COUNT(xdr))) | |
| 105 | #define MEM_INCR(xdr,bytes) (MEM_COUNT(xdr) += (bytes)) | |
| 106 | ||
| 107 | static JSBool | |
| 108 | mem_get32(JSXDRState *xdr, uint32 *lp) | |
| 109 | 0 | { |
| 110 | 0 | MEM_LEFT(xdr, 4); |
| 111 | 0 | *lp = *(uint32 *)MEM_DATA(xdr); |
| 112 | 0 | MEM_INCR(xdr, 4); |
| 113 | 0 | return JS_TRUE; |
| 114 | } | |
| 115 | ||
| 116 | static JSBool | |
| 117 | mem_set32(JSXDRState *xdr, uint32 *lp) | |
| 118 | 0 | { |
| 119 | 0 | MEM_NEED(xdr, 4); |
| 120 | 0 | *(uint32 *)MEM_DATA(xdr) = *lp; |
| 121 | 0 | MEM_INCR(xdr, 4); |
| 122 | 0 | return JS_TRUE; |
| 123 | } | |
| 124 | ||
| 125 | static JSBool | |
| 126 | mem_getbytes(JSXDRState *xdr, char *bytes, uint32 len) | |
| 127 | 0 | { |
| 128 | 0 | MEM_LEFT(xdr, len); |
| 129 | 0 | memcpy(bytes, MEM_DATA(xdr), len); |
| 130 | 0 | MEM_INCR(xdr, len); |
| 131 | 0 | return JS_TRUE; |
| 132 | } | |
| 133 | ||
| 134 | static JSBool | |
| 135 | mem_setbytes(JSXDRState *xdr, char *bytes, uint32 len) | |
| 136 | 0 | { |
| 137 | 0 | MEM_NEED(xdr, len); |
| 138 | 0 | memcpy(MEM_DATA(xdr), bytes, len); |
| 139 | 0 | MEM_INCR(xdr, len); |
| 140 | 0 | return JS_TRUE; |
| 141 | } | |
| 142 | ||
| 143 | static void * | |
| 144 | mem_raw(JSXDRState *xdr, uint32 len) | |
| 145 | 0 | { |
| 146 | void *data; | |
| 147 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 148 | 0 | MEM_NEED(xdr, len); |
| 149 | 0 | } else if (xdr->mode == JSXDR_DECODE) { |
| 150 | 0 | MEM_LEFT(xdr, len); |
| 151 | } | |
| 152 | 0 | data = MEM_DATA(xdr); |
| 153 | 0 | MEM_INCR(xdr, len); |
| 154 | 0 | return data; |
| 155 | } | |
| 156 | ||
| 157 | static JSBool | |
| 158 | mem_seek(JSXDRState *xdr, int32 offset, JSXDRWhence whence) | |
| 159 | 0 | { |
| 160 | 0 | switch (whence) { |
| 161 | case JSXDR_SEEK_CUR: | |
| 162 | 0 | if ((int32)MEM_COUNT(xdr) + offset < 0) { |
| 163 | 0 | JS_ReportErrorNumber(xdr->cx, js_GetErrorMessage, NULL, |
| 164 | JSMSG_SEEK_BEYOND_START); | |
| 165 | 0 | return JS_FALSE; |
| 166 | } | |
| 167 | 0 | if (offset > 0) |
| 168 | 0 | MEM_NEED(xdr, offset); |
| 169 | 0 | MEM_COUNT(xdr) += offset; |
| 170 | 0 | return JS_TRUE; |
| 171 | case JSXDR_SEEK_SET: | |
| 172 | 0 | if (offset < 0) { |
| 173 | 0 | JS_ReportErrorNumber(xdr->cx, js_GetErrorMessage, NULL, |
| 174 | JSMSG_SEEK_BEYOND_START); | |
| 175 | 0 | return JS_FALSE; |
| 176 | } | |
| 177 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 178 | 0 | if ((uint32)offset > MEM_COUNT(xdr)) |
| 179 | 0 | MEM_NEED(xdr, offset - MEM_COUNT(xdr)); |
| 180 | 0 | MEM_COUNT(xdr) = offset; |
| 181 | } else { | |
| 182 | 0 | if ((uint32)offset > MEM_LIMIT(xdr)) { |
| 183 | 0 | JS_ReportErrorNumber(xdr->cx, js_GetErrorMessage, NULL, |
| 184 | JSMSG_SEEK_BEYOND_END); | |
| 185 | 0 | return JS_FALSE; |
| 186 | } | |
| 187 | 0 | MEM_COUNT(xdr) = offset; |
| 188 | } | |
| 189 | 0 | return JS_TRUE; |
| 190 | case JSXDR_SEEK_END: | |
| 191 | 0 | if (offset >= 0 || |
| 192 | xdr->mode == JSXDR_ENCODE || | |
| 193 | (int32)MEM_LIMIT(xdr) + offset < 0) { | |
| 194 | 0 | JS_ReportErrorNumber(xdr->cx, js_GetErrorMessage, NULL, |
| 195 | JSMSG_END_SEEK); | |
| 196 | 0 | return JS_FALSE; |
| 197 | } | |
| 198 | 0 | MEM_COUNT(xdr) = MEM_LIMIT(xdr) + offset; |
| 199 | 0 | return JS_TRUE; |
| 200 | default: { | |
| 201 | char numBuf[12]; | |
| 202 | 0 | JS_snprintf(numBuf, sizeof numBuf, "%d", whence); |
| 203 | 0 | JS_ReportErrorNumber(xdr->cx, js_GetErrorMessage, NULL, |
| 204 | JSMSG_WHITHER_WHENCE, numBuf); | |
| 205 | 0 | return JS_FALSE; |
| 206 | } | |
| 207 | } | |
| 208 | } | |
| 209 | ||
| 210 | static uint32 | |
| 211 | mem_tell(JSXDRState *xdr) | |
| 212 | 0 | { |
| 213 | 0 | return MEM_COUNT(xdr); |
| 214 | } | |
| 215 | ||
| 216 | static void | |
| 217 | mem_finalize(JSXDRState *xdr) | |
| 218 | 0 | { |
| 219 | 0 | JS_free(xdr->cx, MEM_BASE(xdr)); |
| 220 | 0 | } |
| 221 | ||
| 222 | static JSXDROps xdrmem_ops = { | |
| 223 | mem_get32, mem_set32, mem_getbytes, mem_setbytes, | |
| 224 | mem_raw, mem_seek, mem_tell, mem_finalize | |
| 225 | }; | |
| 226 | ||
| 227 | JS_PUBLIC_API(void) | |
| 228 | JS_XDRInitBase(JSXDRState *xdr, JSXDRMode mode, JSContext *cx) | |
| 229 | 0 | { |
| 230 | 0 | xdr->mode = mode; |
| 231 | 0 | xdr->cx = cx; |
| 232 | 0 | xdr->registry = NULL; |
| 233 | 0 | xdr->numclasses = xdr->maxclasses = 0; |
| 234 | 0 | xdr->reghash = NULL; |
| 235 | 0 | xdr->userdata = NULL; |
| 236 | 0 | } |
| 237 | ||
| 238 | JS_PUBLIC_API(JSXDRState *) | |
| 239 | JS_XDRNewMem(JSContext *cx, JSXDRMode mode) | |
| 240 | 0 | { |
| 241 | 0 | JSXDRState *xdr = (JSXDRState *) JS_malloc(cx, sizeof(JSXDRMemState)); |
| 242 | 0 | if (!xdr) |
| 243 | 0 | return NULL; |
| 244 | 0 | JS_XDRInitBase(xdr, mode, cx); |
| 245 | 0 | if (mode == JSXDR_ENCODE) { |
| 246 | 0 | if (!(MEM_BASE(xdr) = JS_malloc(cx, MEM_BLOCK))) { |
| 247 | 0 | JS_free(cx, xdr); |
| 248 | 0 | return NULL; |
| 249 | } | |
| 250 | } else { | |
| 251 | /* XXXbe ok, so better not deref MEM_BASE(xdr) if not ENCODE */ | |
| 252 | 0 | MEM_BASE(xdr) = NULL; |
| 253 | } | |
| 254 | 0 | xdr->ops = &xdrmem_ops; |
| 255 | 0 | MEM_COUNT(xdr) = 0; |
| 256 | 0 | MEM_LIMIT(xdr) = MEM_BLOCK; |
| 257 | 0 | return xdr; |
| 258 | } | |
| 259 | ||
| 260 | JS_PUBLIC_API(void *) | |
| 261 | JS_XDRMemGetData(JSXDRState *xdr, uint32 *lp) | |
| 262 | 0 | { |
| 263 | 0 | if (xdr->ops != &xdrmem_ops) |
| 264 | 0 | return NULL; |
| 265 | 0 | *lp = MEM_COUNT(xdr); |
| 266 | 0 | return MEM_BASE(xdr); |
| 267 | } | |
| 268 | ||
| 269 | JS_PUBLIC_API(void) | |
| 270 | JS_XDRMemSetData(JSXDRState *xdr, void *data, uint32 len) | |
| 271 | 0 | { |
| 272 | 0 | if (xdr->ops != &xdrmem_ops) |
| 273 | 0 | return; |
| 274 | 0 | MEM_LIMIT(xdr) = len; |
| 275 | 0 | MEM_BASE(xdr) = data; |
| 276 | 0 | MEM_COUNT(xdr) = 0; |
| 277 | } | |
| 278 | ||
| 279 | JS_PUBLIC_API(uint32) | |
| 280 | JS_XDRMemDataLeft(JSXDRState *xdr) | |
| 281 | 0 | { |
| 282 | 0 | if (xdr->ops != &xdrmem_ops) |
| 283 | 0 | return 0; |
| 284 | 0 | return MEM_LIMIT(xdr) - MEM_COUNT(xdr); |
| 285 | } | |
| 286 | ||
| 287 | JS_PUBLIC_API(void) | |
| 288 | JS_XDRMemResetData(JSXDRState *xdr) | |
| 289 | 0 | { |
| 290 | 0 | if (xdr->ops != &xdrmem_ops) |
| 291 | 0 | return; |
| 292 | 0 | MEM_COUNT(xdr) = 0; |
| 293 | } | |
| 294 | ||
| 295 | JS_PUBLIC_API(void) | |
| 296 | JS_XDRDestroy(JSXDRState *xdr) | |
| 297 | 0 | { |
| 298 | 0 | JSContext *cx = xdr->cx; |
| 299 | 0 | xdr->ops->finalize(xdr); |
| 300 | 0 | if (xdr->registry) { |
| 301 | 0 | JS_free(cx, xdr->registry); |
| 302 | 0 | if (xdr->reghash) |
| 303 | 0 | JS_DHashTableDestroy(xdr->reghash); |
| 304 | } | |
| 305 | 0 | JS_free(cx, xdr); |
| 306 | 0 | } |
| 307 | ||
| 308 | JS_PUBLIC_API(JSBool) | |
| 309 | JS_XDRUint8(JSXDRState *xdr, uint8 *b) | |
| 310 | 0 | { |
| 311 | 0 | uint32 l = *b; |
| 312 | 0 | if (!JS_XDRUint32(xdr, &l)) |
| 313 | 0 | return JS_FALSE; |
| 314 | 0 | *b = (uint8) l; |
| 315 | 0 | return JS_TRUE; |
| 316 | } | |
| 317 | ||
| 318 | JS_PUBLIC_API(JSBool) | |
| 319 | JS_XDRUint16(JSXDRState *xdr, uint16 *s) | |
| 320 | 0 | { |
| 321 | 0 | uint32 l = *s; |
| 322 | 0 | if (!JS_XDRUint32(xdr, &l)) |
| 323 | 0 | return JS_FALSE; |
| 324 | 0 | *s = (uint16) l; |
| 325 | 0 | return JS_TRUE; |
| 326 | } | |
| 327 | ||
| 328 | JS_PUBLIC_API(JSBool) | |
| 329 | JS_XDRUint32(JSXDRState *xdr, uint32 *lp) | |
| 330 | 0 | { |
| 331 | 0 | JSBool ok = JS_TRUE; |
| 332 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 333 | 0 | uint32 xl = JSXDR_SWAB32(*lp); |
| 334 | 0 | ok = xdr->ops->set32(xdr, &xl); |
| 335 | 0 | } else if (xdr->mode == JSXDR_DECODE) { |
| 336 | 0 | ok = xdr->ops->get32(xdr, lp); |
| 337 | 0 | *lp = JSXDR_SWAB32(*lp); |
| 338 | } | |
| 339 | 0 | return ok; |
| 340 | } | |
| 341 | ||
| 342 | JS_PUBLIC_API(JSBool) | |
| 343 | JS_XDRBytes(JSXDRState *xdr, char *bytes, uint32 len) | |
| 344 | 0 | { |
| 345 | uint32 padlen; | |
| 346 | static char padbuf[JSXDR_ALIGN-1]; | |
| 347 | ||
| 348 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 349 | 0 | if (!xdr->ops->setbytes(xdr, bytes, len)) |
| 350 | 0 | return JS_FALSE; |
| 351 | } else { | |
| 352 | 0 | if (!xdr->ops->getbytes(xdr, bytes, len)) |
| 353 | 0 | return JS_FALSE; |
| 354 | } | |
| 355 | 0 | len = xdr->ops->tell(xdr); |
| 356 | 0 | if (len % JSXDR_ALIGN) { |
| 357 | 0 | padlen = JSXDR_ALIGN - (len % JSXDR_ALIGN); |
| 358 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 359 | 0 | if (!xdr->ops->setbytes(xdr, padbuf, padlen)) |
| 360 | 0 | return JS_FALSE; |
| 361 | } else { | |
| 362 | 0 | if (!xdr->ops->seek(xdr, padlen, JSXDR_SEEK_CUR)) |
| 363 | 0 | return JS_FALSE; |
| 364 | } | |
| 365 | } | |
| 366 | 0 | return JS_TRUE; |
| 367 | } | |
| 368 | ||
| 369 | /** | |
| 370 | * Convert between a C string and the XDR representation: | |
| 371 | * leading 32-bit count, then counted vector of chars, | |
| 372 | * then possibly \0 padding to multiple of 4. | |
| 373 | */ | |
| 374 | JS_PUBLIC_API(JSBool) | |
| 375 | JS_XDRCString(JSXDRState *xdr, char **sp) | |
| 376 | 0 | { |
| 377 | uint32 len; | |
| 378 | ||
| 379 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 380 | 0 | len = strlen(*sp); |
| 381 | 0 | JS_XDRUint32(xdr, &len); |
| 382 | 0 | if (xdr->mode == JSXDR_DECODE) { |
| 383 | 0 | if (!(*sp = (char *) JS_malloc(xdr->cx, len + 1))) |
| 384 | 0 | return JS_FALSE; |
| 385 | } | |
| 386 | 0 | if (!JS_XDRBytes(xdr, *sp, len)) { |
| 387 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 388 | 0 | JS_free(xdr->cx, *sp); |
| 389 | 0 | return JS_FALSE; |
| 390 | } | |
| 391 | 0 | if (xdr->mode == JSXDR_DECODE) { |
| 392 | 0 | (*sp)[len] = '\0'; |
| 393 | 0 | } else if (xdr->mode == JSXDR_FREE) { |
| 394 | 0 | JS_free(xdr->cx, *sp); |
| 395 | 0 | *sp = NULL; |
| 396 | } | |
| 397 | 0 | return JS_TRUE; |
| 398 | } | |
| 399 | ||
| 400 | JS_PUBLIC_API(JSBool) | |
| 401 | JS_XDRCStringOrNull(JSXDRState *xdr, char **sp) | |
| 402 | 0 | { |
| 403 | 0 | uint32 null = (*sp == NULL); |
| 404 | 0 | if (!JS_XDRUint32(xdr, &null)) |
| 405 | 0 | return JS_FALSE; |
| 406 | 0 | if (null) { |
| 407 | 0 | *sp = NULL; |
| 408 | 0 | return JS_TRUE; |
| 409 | } | |
| 410 | 0 | return JS_XDRCString(xdr, sp); |
| 411 | } | |
| 412 | ||
| 413 | /* | |
| 414 | * Convert between a JS (Unicode) string and the XDR representation. | |
| 415 | */ | |
| 416 | JS_PUBLIC_API(JSBool) | |
| 417 | JS_XDRString(JSXDRState *xdr, JSString **strp) | |
| 418 | 0 | { |
| 419 | uint32 i, len, padlen, nbytes; | |
| 420 | 0 | jschar *chars = NULL, *raw; |
| 421 | ||
| 422 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 423 | 0 | len = JSSTRING_LENGTH(*strp); |
| 424 | 0 | if (!JS_XDRUint32(xdr, &len)) |
| 425 | 0 | return JS_FALSE; |
| 426 | 0 | nbytes = len * sizeof(jschar); |
| 427 | ||
| 428 | 0 | if (xdr->mode == JSXDR_DECODE) { |
| 429 | 0 | if (!(chars = (jschar *) JS_malloc(xdr->cx, nbytes + sizeof(jschar)))) |
| 430 | 0 | return JS_FALSE; |
| 431 | } else { | |
| 432 | 0 | chars = JSSTRING_CHARS(*strp); |
| 433 | } | |
| 434 | ||
| 435 | 0 | padlen = nbytes % JSXDR_ALIGN; |
| 436 | 0 | if (padlen) { |
| 437 | 0 | padlen = JSXDR_ALIGN - padlen; |
| 438 | 0 | nbytes += padlen; |
| 439 | } | |
| 440 | 0 | if (!(raw = (jschar *) xdr->ops->raw(xdr, nbytes))) |
| 441 | 0 | goto bad; |
| 442 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 443 | 0 | for (i = 0; i < len; i++) |
| 444 | 0 | raw[i] = JSXDR_SWAB16(chars[i]); |
| 445 | 0 | if (padlen) |
| 446 | 0 | memset((char *)raw + nbytes - padlen, 0, padlen); |
| 447 | 0 | } else if (xdr->mode == JSXDR_DECODE) { |
| 448 | 0 | for (i = 0; i < len; i++) |
| 449 | 0 | chars[i] = JSXDR_SWAB16(raw[i]); |
| 450 | 0 | chars[len] = 0; |
| 451 | ||
| 452 | 0 | if (!(*strp = JS_NewUCString(xdr->cx, chars, len))) |
| 453 | 0 | goto bad; |
| 454 | } | |
| 455 | 0 | return JS_TRUE; |
| 456 | ||
| 457 | 0 | bad: |
| 458 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 459 | 0 | JS_free(xdr->cx, chars); |
| 460 | 0 | return JS_FALSE; |
| 461 | } | |
| 462 | ||
| 463 | JS_PUBLIC_API(JSBool) | |
| 464 | JS_XDRStringOrNull(JSXDRState *xdr, JSString **strp) | |
| 465 | 0 | { |
| 466 | 0 | uint32 null = (*strp == NULL); |
| 467 | 0 | if (!JS_XDRUint32(xdr, &null)) |
| 468 | 0 | return JS_FALSE; |
| 469 | 0 | if (null) { |
| 470 | 0 | *strp = NULL; |
| 471 | 0 | return JS_TRUE; |
| 472 | } | |
| 473 | 0 | return JS_XDRString(xdr, strp); |
| 474 | } | |
| 475 | ||
| 476 | JS_PUBLIC_API(JSBool) | |
| 477 | JS_XDRDouble(JSXDRState *xdr, jsdouble **dp) | |
| 478 | 0 | { |
| 479 | jsdpun u; | |
| 480 | ||
| 481 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 482 | 0 | u.d = **dp; |
| 483 | 0 | if (!JS_XDRUint32(xdr, &u.s.lo) || !JS_XDRUint32(xdr, &u.s.hi)) |
| 484 | 0 | return JS_FALSE; |
| 485 | 0 | if (xdr->mode == JSXDR_DECODE) { |
| 486 | 0 | *dp = JS_NewDouble(xdr->cx, u.d); |
| 487 | 0 | if (!*dp) |
| 488 | 0 | return JS_FALSE; |
| 489 | } | |
| 490 | 0 | return JS_TRUE; |
| 491 | } | |
| 492 | ||
| 493 | /* These are magic pseudo-tags: see jsapi.h, near the top, for real tags. */ | |
| 494 | #define JSVAL_XDRNULL 0x8 | |
| 495 | #define JSVAL_XDRVOID 0xA | |
| 496 | ||
| 497 | JS_PUBLIC_API(JSBool) | |
| 498 | JS_XDRValue(JSXDRState *xdr, jsval *vp) | |
| 499 | 0 | { |
| 500 | uint32 type; | |
| 501 | ||
| 502 | 0 | if (xdr->mode == JSXDR_ENCODE) { |
| 503 | 0 | if (JSVAL_IS_NULL(*vp)) |
| 504 | 0 | type = JSVAL_XDRNULL; |
| 505 | 0 | else if (JSVAL_IS_VOID(*vp)) |
| 506 | 0 | type = JSVAL_XDRVOID; |
| 507 | else | |
| 508 | 0 | type = JSVAL_TAG(*vp); |
| 509 | } | |
| 510 | 0 | if (!JS_XDRUint32(xdr, &type)) |
| 511 | 0 | return JS_FALSE; |
| 512 | ||
| 513 | 0 | switch (type) { |
| 514 | case JSVAL_XDRNULL: | |
| 515 | 0 | *vp = JSVAL_NULL; |
| 516 | 0 | break; |
| 517 | case JSVAL_XDRVOID: | |
| 518 | 0 | *vp = JSVAL_VOID; |
| 519 | 0 | break; |
| 520 | case JSVAL_STRING: { | |
| 521 | JSString *str; | |
| 522 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 523 | 0 | str = JSVAL_TO_STRING(*vp); |
| 524 | 0 | if (!JS_XDRString(xdr, &str)) |
| 525 | 0 | return JS_FALSE; |
| 526 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 527 | 0 | *vp = STRING_TO_JSVAL(str); |
| 528 | 0 | break; |
| 529 | } | |
| 530 | case JSVAL_DOUBLE: { | |
| 531 | jsdouble *dp; | |
| 532 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 533 | 0 | dp = JSVAL_TO_DOUBLE(*vp); |
| 534 | 0 | if (!JS_XDRDouble(xdr, &dp)) |
| 535 | 0 | return JS_FALSE; |
| 536 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 537 | 0 | *vp = DOUBLE_TO_JSVAL(dp); |
| 538 | 0 | break; |
| 539 | } | |
| 540 | case JSVAL_OBJECT: { | |
| 541 | JSObject *obj; | |
| 542 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 543 | 0 | obj = JSVAL_TO_OBJECT(*vp); |
| 544 | 0 | if (!js_XDRObject(xdr, &obj)) |
| 545 | 0 | return JS_FALSE; |
| 546 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 547 | 0 | *vp = OBJECT_TO_JSVAL(obj); |
| 548 | 0 | break; |
| 549 | } | |
| 550 | case JSVAL_BOOLEAN: { | |
| 551 | uint32 b; | |
| 552 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 553 | 0 | b = (uint32) JSVAL_TO_BOOLEAN(*vp); |
| 554 | 0 | if (!JS_XDRUint32(xdr, &b)) |
| 555 | 0 | return JS_FALSE; |
| 556 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 557 | 0 | *vp = BOOLEAN_TO_JSVAL((JSBool) b); |
| 558 | 0 | break; |
| 559 | } | |
| 560 | default: { | |
| 561 | uint32 i; | |
| 562 | ||
| 563 | JS_ASSERT(type & JSVAL_INT); | |
| 564 | 0 | if (xdr->mode == JSXDR_ENCODE) |
| 565 | 0 | i = (uint32) JSVAL_TO_INT(*vp); |
| 566 | 0 | if (!JS_XDRUint32(xdr, &i)) |
| 567 | 0 | return JS_FALSE; |
| 568 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 569 | 0 | *vp = INT_TO_JSVAL((int32) i); |
| 570 | break; | |
| 571 | } | |
| 572 | } | |
| 573 | 0 | return JS_TRUE; |
| 574 | } | |
| 575 | ||
| 576 | JS_PUBLIC_API(JSBool) | |
| 577 | JS_XDRScript(JSXDRState *xdr, JSScript **scriptp) | |
| 578 | 0 | { |
| 579 | 0 | if (!js_XDRScript(xdr, scriptp, NULL)) |
| 580 | 0 | return JS_FALSE; |
| 581 | 0 | if (xdr->mode == JSXDR_DECODE) |
| 582 | 0 | js_CallNewScriptHook(xdr->cx, *scriptp, NULL); |
| 583 | 0 | return JS_TRUE; |
| 584 | } | |
| 585 | ||
| 586 | #define CLASS_REGISTRY_MIN 8 | |
| 587 | #define CLASS_INDEX_TO_ID(i) ((i)+1) | |
| 588 | #define CLASS_ID_TO_INDEX(id) ((id)-1) | |
| 589 | ||
| 590 | typedef struct JSRegHashEntry { | |
| 591 | JSDHashEntryHdr hdr; | |
| 592 | const char *name; | |
| 593 | uint32 index; | |
| 594 | } JSRegHashEntry; | |
| 595 | ||
| 596 | JS_PUBLIC_API(JSBool) | |
| 597 | JS_XDRRegisterClass(JSXDRState *xdr, JSClass *clasp, uint32 *idp) | |
| 598 | 0 | { |
| 599 | uintN numclasses, maxclasses; | |
| 600 | JSClass **registry; | |
| 601 | ||
| 602 | 0 | numclasses = xdr->numclasses; |
| 603 | 0 | maxclasses = xdr->maxclasses; |
| 604 | 0 | if (numclasses == maxclasses) { |
| 605 | 0 | maxclasses = (maxclasses == 0) ? CLASS_REGISTRY_MIN : maxclasses << 1; |
| 606 | 0 | registry = (JSClass **) |
| 607 | JS_realloc(xdr->cx, xdr->registry, maxclasses * sizeof(JSClass *)); | |
| 608 | 0 | if (!registry) |
| 609 | 0 | return JS_FALSE; |
| 610 | 0 | xdr->registry = registry; |
| 611 | 0 | xdr->maxclasses = maxclasses; |
| 612 | } else { | |
| 613 | JS_ASSERT(numclasses && numclasses < maxclasses); | |
| 614 | 0 | registry = xdr->registry; |
| 615 | } | |
| 616 | ||
| 617 | 0 | registry[numclasses] = clasp; |
| 618 | 0 | if (xdr->reghash) { |
| 619 | JSRegHashEntry *entry = (JSRegHashEntry *) | |
| 620 | 0 | JS_DHashTableOperate(xdr->reghash, clasp->name, JS_DHASH_ADD); |
| 621 | 0 | if (!entry) { |
| 622 | 0 | JS_ReportOutOfMemory(xdr->cx); |
| 623 | 0 | return JS_FALSE; |
| 624 | } | |
| 625 | 0 | entry->name = clasp->name; |
| 626 | 0 | entry->index = numclasses; |
| 627 | } | |
| 628 | 0 | *idp = CLASS_INDEX_TO_ID(numclasses); |
| 629 | 0 | xdr->numclasses = ++numclasses; |
| 630 | 0 | return JS_TRUE; |
| 631 | } | |
| 632 | ||
| 633 | JS_PUBLIC_API(uint32) | |
| 634 | JS_XDRFindClassIdByName(JSXDRState *xdr, const char *name) | |
| 635 | 0 | { |
| 636 | uintN i, numclasses; | |
| 637 | ||
| 638 | 0 | numclasses = xdr->numclasses; |
| 639 | 0 | if (numclasses >= 10) { |
| 640 | JSRegHashEntry *entry; | |
| 641 | ||
| 642 | /* Bootstrap reghash from registry on first overpopulated Find. */ | |
| 643 | 0 | if (!xdr->reghash) { |
| 644 | 0 | xdr->reghash = JS_NewDHashTable(JS_DHashGetStubOps(), NULL, |
| 645 | sizeof(JSRegHashEntry), | |
| 646 | numclasses); | |
| 647 | 0 | if (xdr->reghash) { |
| 648 | 0 | for (i = 0; i < numclasses; i++) { |
| 649 | 0 | JSClass *clasp = xdr->registry[i]; |
| 650 | 0 | entry = (JSRegHashEntry *) |
| 651 | JS_DHashTableOperate(xdr->reghash, clasp->name, | |
| 652 | JS_DHASH_ADD); | |
| 653 | 0 | entry->name = clasp->name; |
| 654 | 0 | entry->index = i; |
| 655 | } | |
| 656 | } | |
| 657 | } | |
| 658 | ||
| 659 | /* If we managed to create reghash, use it for O(1) Find. */ | |
| 660 | 0 | if (xdr->reghash) { |
| 661 | 0 | entry = (JSRegHashEntry *) |
| 662 | JS_DHashTableOperate(xdr->reghash, name, JS_DHASH_LOOKUP); | |
| 663 | 0 | if (JS_DHASH_ENTRY_IS_BUSY(&entry->hdr)) |
| 664 | 0 | return CLASS_INDEX_TO_ID(entry->index); |
| 665 | } | |
| 666 | } | |
| 667 | ||
| 668 | /* Only a few classes, or we couldn't malloc reghash: use linear search. */ | |
| 669 | 0 | for (i = 0; i < numclasses; i++) { |
| 670 | 0 | if (!strcmp(name, xdr->registry[i]->name)) |
| 671 | 0 | return CLASS_INDEX_TO_ID(i); |
| 672 | } | |
| 673 | 0 | return 0; |
| 674 | } | |
| 675 | ||
| 676 | JS_PUBLIC_API(JSClass *) | |
| 677 | JS_XDRFindClassById(JSXDRState *xdr, uint32 id) | |
| 678 | 0 | { |
| 679 | 0 | uintN i = CLASS_ID_TO_INDEX(id); |
| 680 | ||
| 681 | 0 | if (i >= xdr->numclasses) |
| 682 | 0 | return NULL; |
| 683 | 0 | return xdr->registry[i]; |
| 684 | } | |
| 685 |