Understanding & Implementing Smart Pointer in C++

Advertisements

Now we have new C++ standards with us. The new standards have taken into account some of the real good features which were missing in C++. But this article has nothing to do with C++11. perhaps my future articles will talk about them but this article talks about the smart pointers and contains a rudimentary implementation of "auto_ptr".
Background

One of the major benefits of using C++ is its deterministic nature. Using C++, we can deterministically control our memory allocation and deallocation(new and delete). but sometimes this becomes a problem. If a novice programmer forgets to delete any object created by new, then we have a memory leak. In case of long running systems, even small memory leaks will create bigger problems. So how can we ensure that for every new, there is a corresponding delete.

One way to do so is to use auto_ptr available in C++. auto_ptr transfers the ownership on copy operations but if we need shared ownership then perhaps we can use boost libraries' smart pointers. But to really understand how these things must be working internally, we need to first implement them. Implementing smart pointers gives us the full understanding of why we need smart pointers, how they are working underneath and where to look for if we have some bug. This exercise here is an implementation of auto_ptr. The main idea is to understand how auto_ptr takes care of transfer of ownership and using various other operations.
Using the Code

The project contains two main classes. Our implementation of smart pointer in MyAuto_Ptr and other class to test this smart pointer MyTestObject. The test suite is written in the main class.

I have put in comments in all the places where we need to know what that code is doing so without talking any more, I give you the code.
The Test Class
Collapse | Copy Code

class MyTestObject
{
    string name;
    int age;

public:
    MyTestObject(string s, int a)
    {
        name = s;
        age = a;
        cout << "Object created ";
    }

    ~MyTestObject(void)
    {
        cout << "Object destroyed ";
    }

    string GetName()
    {
        return name;
    }

    int GetAge()
    {
        return age;
    }
};

Our Smart Pointer
Collapse | Copy Code

template <class T>
class MyAuto_Ptr
{
private:
    T *ptr;

public:

    //Constructor
    explicit MyAuto_Ptr(T *p = 0)
        :ptr(p)
    {

    }

    //Desctructor
    ~MyAuto_Ptr(void)
    {
        //check if null or not
        if(0 != ptr)
        {
            delete ptr;
        }
    }

    //Copy constructor
    MyAuto_Ptr(MyAuto_Ptr<T> &source)
    {
        ptr = source.ptr;

        //Since the ownership gets transferred lets point the source to 0
        source.ptr = 0;
    }

    MyAuto_Ptr& operator=(MyAuto_Ptr<T> &source)
    {
        //check for self reference
        if(ptr != source.ptr)
        {
            //flush the existing memory
            if(0 != ptr)
            {
                delete ptr;
            }

            // use the new pointer
            ptr = source.ptr;

            //Since the ownership gets transferred lets point the source to 0
            source.ptr = 0;

            return *this;
        }
    }

    //Access the value of pointer
    T& operator *()
    {
        return *ptr;
    }

    //Access the pointer
    T* operator ->()
    {
        return ptr;
    }

    //Get the value of pointer
    T* get()
    {
        return ptr;
    }

    //reset the value of pointer
    void reset(T *newVal)
    {
        if(0 != ptr)
        {
            delete ptr;
            ptr = newVal;
        }
    }

    //release the pointer and returns the ownership
    T* release()
    {
        if(0 != ptr)
        {
            T *t = ptr;
            ptr = 0;
            return t;
        }
    }
};

Testing the Class (Main)
Collapse | Copy Code

int main(int argc, char* argv[])
{
    {//dummy brace to test the destruction
        MyAuto_Ptr<MyTestObject> p(new MyTestObject("Rahul Singh", 29));

        //test the -> operator
        cout << "Name: " << p->GetName() << ", Age: " << p->GetAge() << " ";

        //test the transfer of ownership in copy constructor
        MyAuto_Ptr<MyTestObject> p2(p);

        //test the * operator
        cout << "Name: " << (*p2).GetName() << ", Age: " << (*p2).GetAge() << " ";

        //test the transfer of ownership in assignment operator
        MyAuto_Ptr<MyTestObject> p3;
        p3 = p2;

        //test the get function
        cout << "Name: " << p3.get()->GetName() << ",
            Age: " << p3.get()->GetAge() << " ";

        //lets test the reset function
        p3.reset(new MyTestObject("Megha Singh", 28));

        //test the get function
        cout << "Name: " << p3.get()->GetName() << ",
            Age: " << p3.get()->GetAge() << " ";

        //finally lets test the release function
        MyAuto_Ptr<MyTestObject> p4(p3.release());

        cout << "Name: " << p4.get()->GetName() << ",
            Age: " << p4.get()->GetAge() << " ";
    }
    getchar();
    return 0;
}