source: webkit/trunk/JavaScriptCore/kjs/protect.h@ 9768

// -*- c-basic-offset: 2 -*-
/*
 *  This file is part of the KDE libraries
 *  Copyright (C) 2004 Apple Computer, Inc.
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Library General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Library General Public License for more details.
 *
 *  You should have received a copy of the GNU Library General Public License
 *  along with this library; see the file COPYING.LIB.  If not, write to
 *  the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 *  Boston, MA 02111-1307, USA.
 *
 */


#ifndef _KJS_PROTECT_H_
#define _KJS_PROTECT_H_

#include "object.h"
#include "reference.h"
#include "value.h"
#include "protected_values.h"

namespace KJS {

    inline void gcProtect(ValueImp *val) 
      { 
        ProtectedValues::increaseProtectCount(val);
      }
    inline void gcUnprotect(ValueImp *val)
      { 
        ProtectedValues::decreaseProtectCount(val);
      }

    inline void gcProtectNullTolerant(ValueImp *val) 
      {
        if (val) gcProtect(val);
      }

    inline void gcUnprotectNullTolerant(ValueImp *val) 
      {
        if (val) gcUnprotect(val);
      }

    
    class ProtectedValue : public Value {
    public:
      ProtectedValue() : Value() {}
      ProtectedValue(const Value&v)  : Value(v) { gcProtectNullTolerant(v.imp()); };
      ProtectedValue(const ProtectedValue&v)  : Value(v) { gcProtectNullTolerant(v.imp()); };
      ~ProtectedValue() { gcUnprotectNullTolerant(imp());}
      ProtectedValue& operator=(const Value &v)
        { 
          ValueImp *old = imp();
          Value::operator=(v); 
          gcProtectNullTolerant(v.imp());
          gcUnprotectNullTolerant(old); 
          return *this;
        }
      ProtectedValue& operator=(const ProtectedValue &v)
        { 
          ValueImp *old = imp();
          Value::operator=(v); 
          gcProtectNullTolerant(v.imp());
          gcUnprotectNullTolerant(old); 
          return *this;
        }
    private:
      explicit ProtectedValue(ValueImp *v);
    };


    class ProtectedObject : public Object {
    public:
      ProtectedObject() : Object() {}
      ProtectedObject(const Object &o)  : Object(o) { gcProtectNullTolerant(o.imp()); };
      ProtectedObject(const ProtectedObject &o)  : Object(o) { gcProtectNullTolerant(o.imp()); };
      ~ProtectedObject() { gcUnprotectNullTolerant(imp());}
      ProtectedObject& operator=(const Object &o)
        { 
          ValueImp *old = imp();
          Object::operator=(o); 
          gcProtectNullTolerant(o.imp());
          gcUnprotectNullTolerant(old); 
          return *this;
        }
      ProtectedObject& operator=(const ProtectedObject &o)
        { 
          ValueImp *old = imp();
          Object::operator=(o); 
          gcProtectNullTolerant(o.imp());
          gcUnprotectNullTolerant(old); 
          return *this;
        }
    private:
      explicit ProtectedObject(ObjectImp *o);
    };


    class ProtectedReference : public Reference {
    public:
      ProtectedReference(const Reference&r)  : Reference(r) { gcProtectNullTolerant(r.base.imp()); };
      ~ProtectedReference() { gcUnprotectNullTolerant(base.imp());}
      ProtectedReference& operator=(const Reference &r)
        { 
          ValueImp *old = base.imp();
          Reference::operator=(r); 
          gcProtectNullTolerant(r.base.imp());
          gcUnprotectNullTolerant(old); 
          return *this;
        }
    private:
      ProtectedReference();
      ProtectedReference(const Object& b, const Identifier& p);
      ProtectedReference(const Object& b, unsigned p);
      ProtectedReference(ObjectImp *b, const Identifier& p);
      ProtectedReference(ObjectImp *b, unsigned p);
      ProtectedReference(const Null& b, const Identifier& p);
      ProtectedReference(const Null& b, unsigned p);
    };

}

#endif