diff -r 000000000000 -r 6474c204b198 toolkit/devtools/server/protocol.js --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/toolkit/devtools/server/protocol.js Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,1338 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +"use strict"; + +let {Cu} = require("chrome"); +let Services = require("Services"); +let promise = require("devtools/toolkit/deprecated-sync-thenables"); +let {Class} = require("sdk/core/heritage"); +let {EventTarget} = require("sdk/event/target"); +let events = require("sdk/event/core"); +let object = require("sdk/util/object"); + +// Waiting for promise.done() to be added, see bug 851321 +function promiseDone(err) { + console.error(err); + return promise.reject(err); +} + +/** + * Types: named marshallers/demarshallers. + * + * Types provide a 'write' function that takes a js representation and + * returns a protocol representation, and a "read" function that + * takes a protocol representation and returns a js representation. + * + * The read and write methods are also passed a context object that + * represent the actor or front requesting the translation. + * + * Types are referred to with a typestring. Basic types are + * registered by name using addType, and more complex types can + * be generated by adding detail to the type name. + */ + +let types = Object.create(null); +exports.types = types; + +let registeredTypes = new Map(); +let registeredLifetimes = new Map(); + +/** + * Return the type object associated with a given typestring. + * If passed a type object, it will be returned unchanged. + * + * Types can be registered with addType, or can be created on + * the fly with typestrings. Examples: + * + * boolean + * threadActor + * threadActor#detail + * array:threadActor + * array:array:threadActor#detail + * + * @param [typestring|type] type + * Either a typestring naming a type or a type object. + * + * @returns a type object. + */ +types.getType = function(type) { + if (!type) { + return types.Primitive; + } + + if (typeof(type) !== "string") { + return type; + } + + // If already registered, we're done here. + let reg = registeredTypes.get(type); + if (reg) return reg; + + // New type, see if it's a collection/lifetime type: + let sep = type.indexOf(":"); + if (sep >= 0) { + let collection = type.substring(0, sep); + let subtype = types.getType(type.substring(sep + 1)); + + if (collection === "array") { + return types.addArrayType(subtype); + } else if (collection === "nullable") { + return types.addNullableType(subtype); + } + + if (registeredLifetimes.has(collection)) { + return types.addLifetimeType(collection, subtype); + } + + throw Error("Unknown collection type: " + collection); + } + + // Not a collection, might be actor detail + let pieces = type.split("#", 2); + if (pieces.length > 1) { + return types.addActorDetail(type, pieces[0], pieces[1]); + } + + // Might be a lazily-loaded type + if (type === "longstring") { + require("devtools/server/actors/string"); + return registeredTypes.get("longstring"); + } + + throw Error("Unknown type: " + type); +} + +/** + * Don't allow undefined when writing primitive types to packets. If + * you want to allow undefined, use a nullable type. + */ +function identityWrite(v) { + if (v === undefined) { + throw Error("undefined passed where a value is required"); + } + return v; +} + +/** + * Add a type to the type system. + * + * When registering a type, you can provide `read` and `write` methods. + * + * The `read` method will be passed a JS object value from the JSON + * packet and must return a native representation. The `write` method will + * be passed a native representation and should provide a JSONable value. + * + * These methods will both be passed a context. The context is the object + * performing or servicing the request - on the server side it will be + * an Actor, on the client side it will be a Front. + * + * @param typestring name + * Name to register + * @param object typeObject + * An object whose properties will be stored in the type, including + * the `read` and `write` methods. + * @param object options + * Can specify `thawed` to prevent the type from being frozen. + * + * @returns a type object that can be used in protocol definitions. + */ +types.addType = function(name, typeObject={}, options={}) { + if (registeredTypes.has(name)) { + throw Error("Type '" + name + "' already exists."); + } + + let type = object.merge({ + name: name, + primitive: !(typeObject.read || typeObject.write), + read: identityWrite, + write: identityWrite + }, typeObject); + + registeredTypes.set(name, type); + + if (!options.thawed) { + Object.freeze(type); + } + + return type; +}; + +/** + * Add an array type to the type system. + * + * getType() will call this function if provided an "array:" + * typestring. + * + * @param type subtype + * The subtype to be held by the array. + */ +types.addArrayType = function(subtype) { + subtype = types.getType(subtype); + + let name = "array:" + subtype.name; + + // Arrays of primitive types are primitive types themselves. + if (subtype.primitive) { + return types.addType(name); + } + return types.addType(name, { + category: "array", + read: (v, ctx) => [subtype.read(i, ctx) for (i of v)], + write: (v, ctx) => [subtype.write(i, ctx) for (i of v)] + }); +}; + +/** + * Add a dict type to the type system. This allows you to serialize + * a JS object that contains non-primitive subtypes. + * + * Properties of the value that aren't included in the specializations + * will be serialized as primitive values. + * + * @param object specializations + * A dict of property names => type + */ +types.addDictType = function(name, specializations) { + return types.addType(name, { + category: "dict", + specializations: specializations, + read: (v, ctx) => { + let ret = {}; + for (let prop in v) { + if (prop in specializations) { + ret[prop] = types.getType(specializations[prop]).read(v[prop], ctx); + } else { + ret[prop] = v[prop]; + } + } + return ret; + }, + + write: (v, ctx) => { + let ret = {}; + for (let prop in v) { + if (prop in specializations) { + ret[prop] = types.getType(specializations[prop]).write(v[prop], ctx); + } else { + ret[prop] = v[prop]; + } + } + return ret; + } + }) +} + +/** + * Register an actor type with the type system. + * + * Types are marshalled differently when communicating server->client + * than they are when communicating client->server. The server needs + * to provide useful information to the client, so uses the actor's + * `form` method to get a json representation of the actor. When + * making a request from the client we only need the actor ID string. + * + * This function can be called before the associated actor has been + * constructed, but the read and write methods won't work until + * the associated addActorImpl or addActorFront methods have been + * called during actor/front construction. + * + * @param string name + * The typestring to register. + */ +types.addActorType = function(name) { + let type = types.addType(name, { + _actor: true, + category: "actor", + read: (v, ctx, detail) => { + // If we're reading a request on the server side, just + // find the actor registered with this actorID. + if (ctx instanceof Actor) { + return ctx.conn.getActor(v); + } + + // Reading a response on the client side, check for an + // existing front on the connection, and create the front + // if it isn't found. + let actorID = typeof(v) === "string" ? v : v.actor; + let front = ctx.conn.getActor(actorID); + if (front) { + front.form(v, detail, ctx); + } else { + front = new type.frontClass(ctx.conn, v, detail, ctx) + front.actorID = actorID; + ctx.marshallPool().manage(front); + } + return front; + }, + write: (v, ctx, detail) => { + // If returning a response from the server side, make sure + // the actor is added to a parent object and return its form. + if (v instanceof Actor) { + if (!v.actorID) { + ctx.marshallPool().manage(v); + } + return v.form(detail); + } + + // Writing a request from the client side, just send the actor id. + return v.actorID; + }, + }, { + // We usually freeze types, but actor types are updated when clients are + // created, so don't freeze yet. + thawed: true + }); + return type; +} + +types.addNullableType = function(subtype) { + subtype = types.getType(subtype); + return types.addType("nullable:" + subtype.name, { + category: "nullable", + read: (value, ctx) => { + if (value == null) { + return value; + } + return subtype.read(value, ctx); + }, + write: (value, ctx) => { + if (value == null) { + return value; + } + return subtype.write(value, ctx); + } + }); +} + +/** + * Register an actor detail type. This is just like an actor type, but + * will pass a detail hint to the actor's form method during serialization/ + * deserialization. + * + * This is called by getType() when passed an 'actorType#detail' string. + * + * @param string name + * The typestring to register this type as. + * @param type actorType + * The actor type you'll be detailing. + * @param string detail + * The detail to pass. + */ +types.addActorDetail = function(name, actorType, detail) { + actorType = types.getType(actorType); + if (!actorType._actor) { + throw Error("Details only apply to actor types, tried to add detail '" + detail + "'' to " + actorType.name + "\n"); + } + return types.addType(name, { + _actor: true, + category: "detail", + read: (v, ctx) => actorType.read(v, ctx, detail), + write: (v, ctx) => actorType.write(v, ctx, detail) + }); +} + +/** + * Register an actor lifetime. This lets the type system find a parent + * actor that differs from the actor fulfilling the request. + * + * @param string name + * The lifetime name to use in typestrings. + * @param string prop + * The property of the actor that holds the parent that should be used. + */ +types.addLifetime = function(name, prop) { + if (registeredLifetimes.has(name)) { + throw Error("Lifetime '" + name + "' already registered."); + } + registeredLifetimes.set(name, prop); +} + +/** + * Register a lifetime type. This creates an actor type tied to the given + * lifetime. + * + * This is called by getType() when passed a ':' + * typestring. + * + * @param string lifetime + * A lifetime string previously regisered with addLifetime() + * @param type subtype + * An actor type + */ +types.addLifetimeType = function(lifetime, subtype) { + subtype = types.getType(subtype); + if (!subtype._actor) { + throw Error("Lifetimes only apply to actor types, tried to apply lifetime '" + lifetime + "'' to " + subtype.name); + } + let prop = registeredLifetimes.get(lifetime); + return types.addType(lifetime + ":" + subtype.name, { + category: "lifetime", + read: (value, ctx) => subtype.read(value, ctx[prop]), + write: (value, ctx) => subtype.write(value, ctx[prop]) + }) +} + +// Add a few named primitive types. +types.Primitive = types.addType("primitive"); +types.String = types.addType("string"); +types.Number = types.addType("number"); +types.Boolean = types.addType("boolean"); +types.JSON = types.addType("json"); + +/** + * Request/Response templates and generation + * + * Request packets are specified as json templates with + * Arg and Option placeholders where arguments should be + * placed. + * + * Reponse packets are also specified as json templates, + * with a RetVal placeholder where the return value should be + * placed. + */ + +/** + * Placeholder for simple arguments. + * + * @param number index + * The argument index to place at this position. + * @param type type + * The argument should be marshalled as this type. + * @constructor + */ +let Arg = Class({ + initialize: function(index, type) { + this.index = index; + this.type = types.getType(type); + }, + + write: function(arg, ctx) { + return this.type.write(arg, ctx); + }, + + read: function(v, ctx, outArgs) { + outArgs[this.index] = this.type.read(v, ctx); + }, + + describe: function() { + return { + _arg: this.index, + type: this.type.name, + } + } +}); +exports.Arg = Arg; + +/** + * Placeholder for an options argument value that should be hoisted + * into the packet. + * + * If provided in a method specification: + * + * { optionArg: Option(1)} + * + * Then arguments[1].optionArg will be placed in the packet in this + * value's place. + * + * @param number index + * The argument index of the options value. + * @param type type + * The argument should be marshalled as this type. + * @constructor + */ +let Option = Class({ + extends: Arg, + initialize: function(index, type) { + Arg.prototype.initialize.call(this, index, type) + }, + + write: function(arg, ctx, name) { + if (!arg) { + return undefined; + } + let v = arg[name] || undefined; + if (v === undefined) { + return undefined; + } + return this.type.write(v, ctx); + }, + read: function(v, ctx, outArgs, name) { + if (outArgs[this.index] === undefined) { + outArgs[this.index] = {}; + } + if (v === undefined) { + return; + } + outArgs[this.index][name] = this.type.read(v, ctx); + }, + + describe: function() { + return { + _option: this.index, + type: this.type.name, + } + } +}); + +exports.Option = Option; + +/** + * Placeholder for return values in a response template. + * + * @param type type + * The return value should be marshalled as this type. + */ +let RetVal = Class({ + initialize: function(type) { + this.type = types.getType(type); + }, + + write: function(v, ctx) { + return this.type.write(v, ctx); + }, + + read: function(v, ctx) { + return this.type.read(v, ctx); + }, + + describe: function() { + return { + _retval: this.type.name + } + } +}); + +exports.RetVal = RetVal; + +/* Template handling functions */ + +/** + * Get the value at a given path, or undefined if not found. + */ +function getPath(obj, path) { + for (let name of path) { + if (!(name in obj)) { + return undefined; + } + obj = obj[name]; + } + return obj; +} + +/** + * Find Placeholders in the template and save them along with their + * paths. + */ +function findPlaceholders(template, constructor, path=[], placeholders=[]) { + if (!template || typeof(template) != "object") { + return placeholders; + } + + if (template instanceof constructor) { + placeholders.push({ placeholder: template, path: [p for (p of path)] }); + return placeholders; + } + + for (let name in template) { + path.push(name); + findPlaceholders(template[name], constructor, path, placeholders); + path.pop(); + } + + return placeholders; +} + + +function describeTemplate(template) { + return JSON.parse(JSON.stringify(template, (key, value) => { + if (value.describe) { + return value.describe(); + } + return value; + })); +} + +/** + * Manages a request template. + * + * @param object template + * The request template. + * @construcor + */ +let Request = Class({ + initialize: function(template={}) { + this.type = template.type; + this.template = template; + this.args = findPlaceholders(template, Arg); + }, + + /** + * Write a request. + * + * @param array fnArgs + * The function arguments to place in the request. + * @param object ctx + * The object making the request. + * @returns a request packet. + */ + write: function(fnArgs, ctx) { + let str = JSON.stringify(this.template, (key, value) => { + if (value instanceof Arg) { + return value.write(fnArgs[value.index], ctx, key); + } + return value; + }); + return JSON.parse(str); + }, + + /** + * Read a request. + * + * @param object packet + * The request packet. + * @param object ctx + * The object making the request. + * @returns an arguments array + */ + read: function(packet, ctx) { + let fnArgs = []; + for (let templateArg of this.args) { + let arg = templateArg.placeholder; + let path = templateArg.path; + let name = path[path.length - 1]; + arg.read(getPath(packet, path), ctx, fnArgs, name); + } + return fnArgs; + }, + + describe: function() { return describeTemplate(this.template); } +}); + +/** + * Manages a response template. + * + * @param object template + * The response template. + * @construcor + */ +let Response = Class({ + initialize: function(template={}) { + this.template = template; + let placeholders = findPlaceholders(template, RetVal); + if (placeholders.length > 1) { + throw Error("More than one RetVal specified in response"); + } + let placeholder = placeholders.shift(); + if (placeholder) { + this.retVal = placeholder.placeholder; + this.path = placeholder.path; + } + }, + + /** + * Write a response for the given return value. + * + * @param val ret + * The return value. + * @param object ctx + * The object writing the response. + */ + write: function(ret, ctx) { + return JSON.parse(JSON.stringify(this.template, function(key, value) { + if (value instanceof RetVal) { + return value.write(ret, ctx); + } + return value; + })); + }, + + /** + * Read a return value from the given response. + * + * @param object packet + * The response packet. + * @param object ctx + * The object reading the response. + */ + read: function(packet, ctx) { + if (!this.retVal) { + return undefined; + } + let v = getPath(packet, this.path); + return this.retVal.read(v, ctx); + }, + + describe: function() { return describeTemplate(this.template); } +}); + +/** + * Actor and Front implementations + */ + +/** + * A protocol object that can manage the lifetime of other protocol + * objects. + */ +let Pool = Class({ + extends: EventTarget, + + /** + * Pools are used on both sides of the connection to help coordinate + * lifetimes. + * + * @param optional conn + * Either a DebuggerServerConnection or a DebuggerClient. Must have + * addActorPool, removeActorPool, and poolFor. + * conn can be null if the subclass provides a conn property. + * @constructor + */ + initialize: function(conn) { + if (conn) { + this.conn = conn; + } + }, + + /** + * Return the parent pool for this client. + */ + parent: function() { return this.conn.poolFor(this.actorID) }, + + /** + * Override this if you want actors returned by this actor + * to belong to a different actor by default. + */ + marshallPool: function() { return this; }, + + /** + * Pool is the base class for all actors, even leaf nodes. + * If the child map is actually referenced, go ahead and create + * the stuff needed by the pool. + */ + __poolMap: null, + get _poolMap() { + if (this.__poolMap) return this.__poolMap; + this.__poolMap = new Map(); + this.conn.addActorPool(this); + return this.__poolMap; + }, + + /** + * Add an actor as a child of this pool. + */ + manage: function(actor) { + if (!actor.actorID) { + actor.actorID = this.conn.allocID(actor.actorPrefix || actor.typeName); + } + + this._poolMap.set(actor.actorID, actor); + return actor; + }, + + /** + * Remove an actor as a child of this pool. + */ + unmanage: function(actor) { + this.__poolMap.delete(actor.actorID); + }, + + // true if the given actor ID exists in the pool. + has: function(actorID) this.__poolMap && this._poolMap.has(actorID), + + // The actor for a given actor id stored in this pool + actor: function(actorID) this.__poolMap ? this._poolMap.get(actorID) : null, + + // Same as actor, should update debugger connection to use 'actor' + // and then remove this. + get: function(actorID) this.__poolMap ? this._poolMap.get(actorID) : null, + + // True if this pool has no children. + isEmpty: function() !this.__poolMap || this._poolMap.size == 0, + + /** + * Destroy this item, removing it from a parent if it has one, + * and destroying all children if necessary. + */ + destroy: function() { + let parent = this.parent(); + if (parent) { + parent.unmanage(this); + } + if (!this.__poolMap) { + return; + } + for (let actor of this.__poolMap.values()) { + // Self-owned actors are ok, but don't need destroying twice. + if (actor === this) { + continue; + } + let destroy = actor.destroy; + if (destroy) { + // Disconnect destroy while we're destroying in case of (misbehaving) + // circular ownership. + actor.destroy = null; + destroy.call(actor); + actor.destroy = destroy; + } + }; + this.conn.removeActorPool(this, true); + this.__poolMap.clear(); + this.__poolMap = null; + }, + + /** + * For getting along with the debugger server pools, should be removable + * eventually. + */ + cleanup: function() { + this.destroy(); + } +}); +exports.Pool = Pool; + +/** + * An actor in the actor tree. + */ +let Actor = Class({ + extends: Pool, + + // Will contain the actor's ID + actorID: null, + + /** + * Initialize an actor. + * + * @param optional conn + * Either a DebuggerServerConnection or a DebuggerClient. Must have + * addActorPool, removeActorPool, and poolFor. + * conn can be null if the subclass provides a conn property. + * @constructor + */ + initialize: function(conn) { + Pool.prototype.initialize.call(this, conn); + + // Forward events to the connection. + if (this._actorSpec && this._actorSpec.events) { + for (let key of this._actorSpec.events.keys()) { + let name = key; + let sendEvent = this._sendEvent.bind(this, name) + this.on(name, (...args) => { + sendEvent.apply(null, args); + }); + } + } + }, + + _sendEvent: function(name, ...args) { + if (!this._actorSpec.events.has(name)) { + // It's ok to emit events that don't go over the wire. + return; + } + let request = this._actorSpec.events.get(name); + let packet = request.write(args, this); + packet.from = packet.from || this.actorID; + this.conn.send(packet); + }, + + destroy: function() { + Pool.prototype.destroy.call(this); + this.actorID = null; + }, + + /** + * Override this method in subclasses to serialize the actor. + * @param [optional] string hint + * Optional string to customize the form. + * @returns A jsonable object. + */ + form: function(hint) { + return { actor: this.actorID } + }, + + writeError: function(err) { + console.error(err); + if (err.stack) { + dump(err.stack); + } + this.conn.send({ + from: this.actorID, + error: "unknownError", + message: err.toString() + }); + }, + + _queueResponse: function(create) { + let pending = this._pendingResponse || promise.resolve(null); + let response = create(pending); + this._pendingResponse = response; + } +}); +exports.Actor = Actor; + +/** + * Tags a prtotype method as an actor method implementation. + * + * @param function fn + * The implementation function, will be returned. + * @param spec + * The method specification, with the following (optional) properties: + * request (object): a request template. + * response (object): a response template. + * oneway (bool): 'true' if no response should be sent. + * telemetry (string): Telemetry probe ID for measuring completion time. + */ +exports.method = function(fn, spec={}) { + fn._methodSpec = Object.freeze(spec); + if (spec.request) Object.freeze(spec.request); + if (spec.response) Object.freeze(spec.response); + return fn; +} + +/** + * Process an actor definition from its prototype and generate + * request handlers. + */ +let actorProto = function(actorProto) { + if (actorProto._actorSpec) { + throw new Error("actorProto called twice on the same actor prototype!"); + } + + let protoSpec = { + methods: [], + }; + + // Find method specifications attached to prototype properties. + for (let name of Object.getOwnPropertyNames(actorProto)) { + let desc = Object.getOwnPropertyDescriptor(actorProto, name); + if (!desc.value) { + continue; + } + + if (desc.value._methodSpec) { + let frozenSpec = desc.value._methodSpec; + let spec = {}; + spec.name = frozenSpec.name || name; + spec.request = Request(object.merge({type: spec.name}, frozenSpec.request || undefined)); + spec.response = Response(frozenSpec.response || undefined); + spec.telemetry = frozenSpec.telemetry; + spec.release = frozenSpec.release; + spec.oneway = frozenSpec.oneway; + + protoSpec.methods.push(spec); + } + } + + // Find event specifications + if (actorProto.events) { + protoSpec.events = new Map(); + for (let name in actorProto.events) { + let eventRequest = actorProto.events[name]; + Object.freeze(eventRequest); + protoSpec.events.set(name, Request(object.merge({type: name}, eventRequest))); + } + } + + // Generate request handlers for each method definition + actorProto.requestTypes = Object.create(null); + protoSpec.methods.forEach(spec => { + let handler = function(packet, conn) { + try { + let args = spec.request.read(packet, this); + + let ret = this[spec.name].apply(this, args); + + if (spec.oneway) { + // No need to send a response. + return; + } + + let sendReturn = (ret) => { + let response = spec.response.write(ret, this); + response.from = this.actorID; + // If spec.release has been specified, destroy the object. + if (spec.release) { + try { + this.destroy(); + } catch(e) { + this.writeError(e); + return; + } + } + + conn.send(response); + }; + + this._queueResponse(p => { + return p + .then(() => ret) + .then(sendReturn) + .then(null, this.writeError.bind(this)); + }) + } catch(e) { + this._queueResponse(p => { + return p.then(() => this.writeError(e)); + }); + } + }; + + actorProto.requestTypes[spec.request.type] = handler; + }); + + actorProto._actorSpec = protoSpec; + return actorProto; +} + +/** + * Create an actor class for the given actor prototype. + * + * @param object proto + * The object prototype. Must have a 'typeName' property, + * should have method definitions, can have event definitions. + */ +exports.ActorClass = function(proto) { + if (!proto.typeName) { + throw Error("Actor prototype must have a typeName member."); + } + proto.extends = Actor; + if (!registeredTypes.has(proto.typeName)) { + types.addActorType(proto.typeName); + } + let cls = Class(actorProto(proto)); + + registeredTypes.get(proto.typeName).actorSpec = proto._actorSpec; + return cls; +}; + +/** + * Base class for client-side actor fronts. + */ +let Front = Class({ + extends: Pool, + + actorID: null, + + /** + * The base class for client-side actor fronts. + * + * @param optional conn + * Either a DebuggerServerConnection or a DebuggerClient. Must have + * addActorPool, removeActorPool, and poolFor. + * conn can be null if the subclass provides a conn property. + * @param optional form + * The json form provided by the server. + * @constructor + */ + initialize: function(conn=null, form=null, detail=null, context=null) { + Pool.prototype.initialize.call(this, conn); + this._requests = []; + if (form) { + this.actorID = form.actor; + this.form(form, detail, context); + } + }, + + destroy: function() { + // Reject all outstanding requests, they won't make sense after + // the front is destroyed. + while (this._requests && this._requests.length > 0) { + let deferred = this._requests.shift(); + deferred.reject(new Error("Connection closed")); + } + Pool.prototype.destroy.call(this); + this.actorID = null; + }, + + /** + * @returns a promise that will resolve to the actorID this front + * represents. + */ + actor: function() { return promise.resolve(this.actorID) }, + + toString: function() { return "[Front for " + this.typeName + "/" + this.actorID + "]" }, + + /** + * Update the actor from its representation. + * Subclasses should override this. + */ + form: function(form) {}, + + /** + * Send a packet on the connection. + */ + send: function(packet) { + if (packet.to) { + this.conn._transport.send(packet); + } else { + this.actor().then(actorID => { + packet.to = actorID; + this.conn._transport.send(packet); + }); + } + }, + + /** + * Send a two-way request on the connection. + */ + request: function(packet) { + let deferred = promise.defer(); + this._requests.push(deferred); + this.send(packet); + return deferred.promise; + }, + + /** + * Handler for incoming packets from the client's actor. + */ + onPacket: function(packet) { + // Pick off event packets + if (this._clientSpec.events && this._clientSpec.events.has(packet.type)) { + let event = this._clientSpec.events.get(packet.type); + let args = event.request.read(packet, this); + if (event.pre) { + event.pre.forEach((pre) => pre.apply(this, args)); + } + events.emit.apply(null, [this, event.name].concat(args)); + return; + } + + // Remaining packets must be responses. + if (this._requests.length === 0) { + let msg = "Unexpected packet " + this.actorID + ", " + JSON.stringify(packet); + let err = Error(msg); + console.error(err); + throw err; + } + + let deferred = this._requests.shift(); + if (packet.error) { + deferred.reject(packet.error); + } else { + deferred.resolve(packet); + } + } +}); +exports.Front = Front; + +/** + * A method tagged with preEvent will be called after recieving a packet + * for that event, and before the front emits the event. + */ +exports.preEvent = function(eventName, fn) { + fn._preEvent = eventName; + return fn; +} + +/** + * Mark a method as a custom front implementation, replacing the generated + * front method. + * + * @param function fn + * The front implementation, will be returned. + * @param object options + * Options object: + * impl (string): If provided, the generated front method will be + * stored as this property on the prototype. + */ +exports.custom = function(fn, options={}) { + fn._customFront = options; + return fn; +} + +function prototypeOf(obj) { + return typeof(obj) === "function" ? obj.prototype : obj; +} + +/** + * Process a front definition from its prototype and generate + * request methods. + */ +let frontProto = function(proto) { + let actorType = prototypeOf(proto.actorType); + if (proto._actorSpec) { + throw new Error("frontProto called twice on the same front prototype!"); + } + proto._actorSpec = actorType._actorSpec; + proto.typeName = actorType.typeName; + + // Generate request methods. + let methods = proto._actorSpec.methods; + methods.forEach(spec => { + let name = spec.name; + + // If there's already a property by this name in the front, it must + // be a custom front method. + if (name in proto) { + let custom = proto[spec.name]._customFront; + if (custom === undefined) { + throw Error("Existing method for " + spec.name + " not marked customFront while processing " + actorType.typeName + "."); + } + // If the user doesn't need the impl don't generate it. + if (!custom.impl) { + return; + } + name = custom.impl; + } + + proto[name] = function(...args) { + let histogram, startTime; + if (spec.telemetry) { + if (spec.oneway) { + // That just doesn't make sense. + throw Error("Telemetry specified for a oneway request"); + } + let transportType = this.conn.localTransport + ? "LOCAL_" + : "REMOTE_"; + let histogramId = "DEVTOOLS_DEBUGGER_RDP_" + + transportType + spec.telemetry + "_MS"; + try { + histogram = Services.telemetry.getHistogramById(histogramId); + startTime = new Date(); + } catch(ex) { + // XXX: Is this expected in xpcshell tests? + console.error(ex); + spec.telemetry = false; + } + } + + let packet = spec.request.write(args, this); + if (spec.oneway) { + // Fire-and-forget oneway packets. + this.send(packet); + return undefined; + } + + return this.request(packet).then(response => { + let ret = spec.response.read(response, this); + + if (histogram) { + histogram.add(+new Date - startTime); + } + + return ret; + }).then(null, promiseDone); + } + + // Release methods should call the destroy function on return. + if (spec.release) { + let fn = proto[name]; + proto[name] = function(...args) { + return fn.apply(this, args).then(result => { + this.destroy(); + return result; + }) + } + } + }); + + + // Process event specifications + proto._clientSpec = {}; + + let events = proto._actorSpec.events; + if (events) { + // This actor has events, scan the prototype for preEvent handlers... + let preHandlers = new Map(); + for (let name of Object.getOwnPropertyNames(proto)) { + let desc = Object.getOwnPropertyDescriptor(proto, name); + if (!desc.value) { + continue; + } + if (desc.value._preEvent) { + let preEvent = desc.value._preEvent; + if (!events.has(preEvent)) { + throw Error("preEvent for event that doesn't exist: " + preEvent); + } + let handlers = preHandlers.get(preEvent); + if (!handlers) { + handlers = []; + preHandlers.set(preEvent, handlers); + } + handlers.push(desc.value); + } + } + + proto._clientSpec.events = new Map(); + + for (let [name, request] of events) { + proto._clientSpec.events.set(request.type, { + name: name, + request: request, + pre: preHandlers.get(name) + }); + } + } + return proto; +} + +/** + * Create a front class for the given actor class, with the given prototype. + * + * @param ActorClass actorType + * The actor class you're creating a front for. + * @param object proto + * The object prototype. Must have a 'typeName' property, + * should have method definitions, can have event definitions. + */ +exports.FrontClass = function(actorType, proto) { + proto.actorType = actorType; + proto.extends = Front; + let cls = Class(frontProto(proto)); + registeredTypes.get(cls.prototype.typeName).frontClass = cls; + return cls; +} + + +exports.dumpActorSpec = function(type) { + let actorSpec = type.actorSpec; + let ret = { + category: "actor", + typeName: type.name, + methods: [], + events: {} + }; + + for (let method of actorSpec.methods) { + ret.methods.push({ + name: method.name, + release: method.release || undefined, + oneway: method.oneway || undefined, + request: method.request.describe(), + response: method.response.describe() + }); + } + + if (actorSpec.events) { + for (let [name, request] of actorSpec.events) { + ret.events[name] = request.describe(); + } + } + + + JSON.stringify(ret); + + return ret; +} + +exports.dumpProtocolSpec = function() { + let ret = { + types: {}, + }; + + for (let [name, type] of registeredTypes) { + // Force lazy instantiation if needed. + type = types.getType(name); + if (type.category === "dict") { + ret.types[name] = { + category: "dict", + typeName: name, + specializations: type.specializations + } + } else if (type.category === "actor") { + ret.types[name] = exports.dumpActorSpec(type); + } + } + + return ret; +}