CString实现原理

 

MFC中CString类的实现在strcore.cpp中，Cstring封装了一个用来存放字符串的缓冲区和对施加于这个字符串的操作。也就是说 ，CString里需要有一个用来存放字符串的缓冲区，并且有一个指针指向该缓冲区，该指针就是LPTSTR m_pchData。但是有些字符串操作会增建或减少字符串的长度，因此为了减少频繁的申请内存或者释放内存，Cstring会先申请一个大的内存块用来存放字符串。这样，以后当字符串长度增长时，如果增加的总长度不超过预先申请的内存块的长度，就不用再申请内存。当增加后的字符串长度超过预先申请的内存时，Cstring 先释放原先的内存，然后再重新申请一个更大的内存块。同样的，当字符串长度减少时，也不释放多出来的内存空间。而是等到积累到一定程度时，才一次性将多余的内存释放。

还有，当使用一个Cstring对象a来初始化另一个Cstring对象b时，为了节省空间，新对象b并不分配空间，它所要做的只是将自己的指针指向对象a的那块内存空间，只有当需要修改对象a或者b中的字符串时，才会为新对象b申请内存空间，这叫做写入复制技术(CopyBe foreWrite)。

这样，仅仅通过一个指针就不能完整的描述这块内存的具体情况，需要更多的信息来描述。
首先，需要有一个变量来描述当前内存块的总的大小。
其次，需要一个变量来描述当前内存块已经使用的情况。也就是当前字符串的长度
另外，还需要一个变量来描述该内存块被其他Cstring引用的情况。有一个对象引用该内存块，就将该数值加一。

Cstring中专门定义了一个结构体来描述这些信息:

C++代码  

1. struct CStringData    
2. {    
3.    long nRefs;             // reference count   
4.    int nDataLength;        // length of data (including terminator)   
5.    int nAllocLength;       // length of allocation   
6.    // TCHAR data[nAllocLength]    
7.   
8.    TCHAR* data()           // TCHAR* to managed data   
9.    { return (TCHAR*)(this+1); }   
10. };  

实际使用时，该结构体的所占用的内存块大小是不固定的，在Cstring内部的内存块头部，放置的是该结构体。从该内存块头部开始的sizeof(CstringData)个BYTE后才是真正的用于存放字符串的内存空间。这种结构的数据结构的申请方法是这样实现的:
pData = (CStringData*) new BYTE[sizeof(CStringData) + (nLen+1)*sizeof(TCHAR)];
pData->nAllocLength = nLen;
其中nLen是用于说明需要一次性申请的内存空间的大小的。

从代码中可以很容易的看出，如果想申请一个256个TCHAR的内存块用于存放字符串，实际申请的大小是：
sizeof(CstringData)个BYTE ＋ (nLen+1)个TCHAR
其中前面sizeof(CstringData)个BYTE是用来存放CstringData信息的。后面的nLen＋1个TCHAR才是真正用来存放字符串的，多出来的一个用来存放'\0'。

CString中所有的operations的都是针对这个缓冲区的。比如LPTSTR CString::GetBuffer(int nMinBufLength)，它的实现方法是:
首先通过Cstring::GetData()取得CstringData对象的指针。该指针是通过存放字符串的指针m_pchData先后偏移sizeof(CstringData)，从而得到了CstringData的地址。
然后根据参数nMinBufLength给定的值重新实例化一个CstringData对象，使得新的对象里的字符串缓冲长度能够满足nMinBufLength。
然后在重新设置一下新的CstringData中的一些描述值。
最后将新CstringData对象里的字符串缓冲直接返回给调用者。

这些过程用C++代码描述就是：

C++代码  

1. if (GetData()->nRefs > 1 || nMinBufLength > GetData()->nAllocLength)    
2. {    
3.     // we have to grow the buffer    
4.     CStringData* pOldData = GetData();    
5.     int nOldLen = GetData()->nDataLength;   // AllocBuffer will tromp it   
6.     if (nMinBufLength < nOldLen)    
7.         nMinBufLength = nOldLen;    
8.     AllocBuffer(nMinBufLength);    
9.     memcpy(m_pchData, pOldData->data(), (nOldLen+1)*sizeof(TCHAR));    
10.     GetData()->nDataLength = nOldLen;    
11.     CString::Release(pOldData);    
12. }    
13.   
14. ASSERT(GetData()->nRefs <= 1);    
15. // return a pointer to the character storage for this string   
16. ASSERT(m_pchData != NULL);    
17. return m_pchData;  

很多时候，我们经常的对大批量的字符串进行互相拷贝修改等，Cstring 使用了CopyBeforeWrite技术。使用这种方法，当利用一个Cstring对象a实例化另一个对象b的时候，其实两个对象的数值是完全相同的，但是如果简单的给两个对象都申请内存的话，对于只有几个、几十个字节的字符串还没有什么，如果是一个几K甚至几M的数据量来说，是一个很大的浪费。因此 Cstring 在这个时候只是简单的将新对象b的字符串地址m_pchData直接指向另一个对象a的字符串地址m_pchData。所做的额外工作是将对象a的内存引用CstringData:: nRefs加一。

C++代码  

1. CString::CString(const CString& stringSrc)    
2. {    
3.     m_pchData = stringSrc.m_pchData;    
4.     InterlockedIncrement(&GetData()->nRefs);    
5. }    

这样当修改对象a或对象b的字符串内容时，首先检查CStringData::nRefs的值，如果大于一(等于一，说明只有自己一个应用该内存空间)，说明该对象引用了别的对象内存或者自己的内存被别人应用，该对象首先将该引用值减一，然后将该内存交给其他的对象管理，自己重新申请一块内存，并将原来内存的内容拷贝过来。

其实现的简单代码是：

C++代码  

1. void CString::CopyBeforeWrite()    
2. {    
3.     if (GetData()->nRefs > 1)    
4.     {    
5.         CStringData* pData = GetData();    
6.         Release();    
7.         AllocBuffer(pData->nDataLength);    
8.         memcpy(m_pchData, pData->data(), (pData- >nDataLength+1)*sizeof(TCHAR));    
9.     }    
10. }   

其中Release就是用来判断该内存的被引用情况的。

C++代码  

1. void CString::Release()    
2. {    
3.     if (GetData() != _afxDataNil)    
4.     {    
5.         if (InterlockedDecrement(&GetData()->nRefs) <= 0)    
6.             FreeData(GetData());    
7.    }    
8. }    

当多个对象共享同一块内存时，这块内存就属于多个对象，而不在属于原来的申请这块内存的那个对象了。但是，每个对象在其生命结束时，都首先将这块内存的引用减一，然后再判断这个引用值，如果小于等于零时，就将其释放，否则，将之交给另外的正在引用这块内存的对象控制。

Cstring使用这种数据结构，对于大数据量的字符串操作，可以节省很多频繁申请释放内存的时间，有助于提升系统性能。

CString使用中最典型的一个问题就是用来描述内存块属性的属性值和实际的值不一致。出现这个问题的原因就是CString为了方便某些应用，提供了一些operations，这些operation可以直接返回内存块中的字符串的地址值，用户可以通过对这个地址值指向的地址进行修改，但是，修改后又没有调用相应的operations1使CStringData中的值来保持一致。比如，用户可以首先通过operations得到字符串地址，然后将一些新的字符增加到这个字符串中，使得字符串的长度增加，但是，由于是直接通过指针修改的，所以描述该字符串长度的CstringData中的nDataLength却还是原来的长度，因此当通过GetLength获取字符串长度时，返回的必然是不正确的。

 

本文转自：http://tangfeng.javaeye.com/blog/491985