js/ipc/ContextWrapperChild.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
   2  * vim: set ts=8 sw=4 et tw=80:
   3  *
   4  * ***** BEGIN LICENSE BLOCK *****
   5  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   6  *
   7  * The contents of this file are subject to the Mozilla Public License Version
   8  * 1.1 (the "License"); you may not use this file except in compliance with
   9  * the License. You may obtain a copy of the License at
  10  * http://www.mozilla.org/MPL/
  11  *
  12  * Software distributed under the License is distributed on an "AS IS" basis,
  13  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  14  * for the specific language governing rights and limitations under the
  15  * License.
  16  *
  17  * The Original Code is mozilla.org code.
  18  *
  19  * The Initial Developer of the Original Code is
  20  * The Mozilla Foundation.
  21  * Portions created by the Initial Developer are Copyright (C) 2010
  22  * the Initial Developer. All Rights Reserved.
  23  *
  24  * Contributor(s):
  25  *   Ben Newman <b{enjam,newma}n@mozilla.com> (original author)
  26  *
  27  * Alternatively, the contents of this file may be used under the terms of
  28  * either of the GNU General Public License Version 2 or later (the "GPL"),
  29  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  30  * in which case the provisions of the GPL or the LGPL are applicable instead
  31  * of those above. If you wish to allow use of your version of this file only
  32  * under the terms of either the GPL or the LGPL, and not to allow others to
  33  * use your version of this file under the terms of the MPL, indicate your
  34  * decision by deleting the provisions above and replace them with the notice
  35  * and other provisions required by the GPL or the LGPL. If you do not delete
  36  * the provisions above, a recipient may use your version of this file under
  37  * the terms of any one of the MPL, the GPL or the LGPL.
  38  *
  39  * ***** END LICENSE BLOCK ***** */
  40
  41 #ifndef mozilla_jsipc_ContextWrapperChild_h__
  42 #define mozilla_jsipc_ContextWrapperChild_h__
  43
  44 #include "mozilla/jsipc/PContextWrapperChild.h"
  45 #include "mozilla/jsipc/ObjectWrapperChild.h"
  46
  47 #include "jsapi.h"
  48 #include "nsClassHashtable.h"
  49 #include "nsHashKeys.h"
  50
  51 namespace mozilla {
  52 namespace jsipc {
  53
  54 class ContextWrapperChild
  55     : public PContextWrapperChild
  56 {
  57 public:
  58
  59     ContextWrapperChild(JSContext* cx)
  60         : mContext(cx)
  61     {
  62         mResidentObjectTable.Init();
  63     }
  64
  65     JSContext* GetContext() { return mContext; }
  66
  67     PObjectWrapperChild* GetOrCreateWrapper(JSObject* obj,
  68                                             bool makeGlobal = false)
  69     {
  70         if (!obj) // Don't wrap nothin'!
  71             return NULL;
  72         PObjectWrapperChild* wrapper;
  73         while (!mResidentObjectTable.Get(obj, &wrapper)) {
  74             wrapper = SendPObjectWrapperConstructor(AllocPObjectWrapper(obj),
  75                                                     makeGlobal);
  76             if (wrapper)
  77                 mResidentObjectTable.Put(obj, wrapper);
  78             else
  79                 return NULL;
  80         }
  81         return wrapper;
  82     }
  83
  84 protected:
  85
  86     PObjectWrapperChild* AllocPObjectWrapper(JSObject* obj) {
  87         return new ObjectWrapperChild(mContext, obj);
  88     }
  89
  90     PObjectWrapperChild* AllocPObjectWrapper(const bool&) {
  91         return AllocPObjectWrapper(JS_GetGlobalObject(mContext));
  92     }
  93
  94     bool DeallocPObjectWrapper(PObjectWrapperChild* actor) {
  95         ObjectWrapperChild* owc = static_cast<ObjectWrapperChild*>(actor);
  96         mResidentObjectTable.Remove(owc->GetJSObject());
  97         return true;
  98     }
  99
 100 private:
 101
 102     JSContext* const mContext;
 103
 104     nsClassHashtable<nsPtrHashKey<JSObject>,
 105                      PObjectWrapperChild> mResidentObjectTable;
 106
 107 };
 108
 109 }}
 110
 111 #endif