赋值构造函数

#include <iostream>
#include <string>

using namespace std;

class CMyString
{
public:
	CMyString(char *pdata = NULL);
	CMyString(const CMyString& str);
	~CMyString();
	CMyString& operator = (const CMyString& str);

	void Print();

private:
	char* m_pdata;
};

CMyString::CMyString(char* pdata)
{
	if(pdata == NULL)
	{
		m_pdata = new char[1];
		*m_pdata = '\0';
	}
	else
	{
		int length = strlen(pdata);
		m_pdata = new char[length+1];
		strcpy(m_pdata,pdata);
	}
}
CMyString::CMyString(const CMyString& str)
{
	int length = strlen(str.m_pdata);
	m_pdata = new char[length+1];
	strcpy(m_pdata,str.m_pdata);
}
CMyString::~CMyString()
{
	delete[] m_pdata;
}
CMyString& CMyString::operator=(const CMyString& str)
{
	if(&str == this)
		return *this;
	delete []m_pdata;
	m_pdata = NULL;
	int length = strlen(str.m_pdata);
	m_pdata = new char[length+1];
	strcpy(m_pdata, str.m_pdata);
	return *this;
}

// ====================测试代码====================
void CMyString::Print()
{
    printf("%s", m_pdata);
}

void Test1()
{
    printf("Test1 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    CMyString str2;
    str2 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str2.Print();
    printf(".\n");
}

// 赋值给自己
void Test2()
{
    printf("Test2 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    str1 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str1.Print();
    printf(".\n");
}

// 连续赋值
void Test3()
{
    printf("Test3 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    CMyString str2, str3;
    str3 = str2 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str2.Print();
    printf(".\n");

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str3.Print();
    printf(".\n");
}

int main(int argc, char* argv[])
{
    Test1();
    Test2();
    Test3();

    return 0;
}