Mem, MyString

最新推荐文章于 2025-07-21 15:09:40 发布

转载最新推荐文章于 2025-07-21 15:09:40 发布 · 64 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/kateblog/p/5317512.html

本文介绍了一个使用C++实现的自定义字符串类MyString的设计与实现过程，包括其内部使用的内存管理类MemB。同时，文章还讨论了如何通过这种方式避免strcpy和strcat等函数可能引发的安全问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

MemB.h的代码：

#pragma once

#ifndef MEMB
#define MEMB

typedef unsigned char byte;

class MemB
{
private:
    byte * mem;
    int size;
    void ensureMinSize(int minSize);
public:
    MemB();
    MemB(int sz);
    int msize();
    byte * pointer();
    byte * pointer(int minSize);
    ~MemB(void);
};
#endif

MemB.cpp的代码：

#include "stdafx.h"
#include "MemB.h"
#include <cstring>
using namespace std;

MemB::MemB(int sz){
    mem=0;
    size=0;
    ensureMinSize(sz);
}


MemB::MemB()
{
    mem=0;
    size=0;
}

int MemB::msize(){
    return size;
}

void MemB::ensureMinSize(int minSize){
    if(size<minSize){
        byte * newmem= new byte[minSize];
        memset(newmem+size,0,minSize-size);
        memcpy(newmem,mem,size);
        delete []mem;
        mem=newmem;
        size=minSize;
    }
}

byte *MemB::pointer(){
    return mem;
}

byte *MemB::pointer(int minSize){
    ensureMinSize(minSize);        
    return mem;
}


MemB::~MemB(void)
{
    delete []mem;
}

MyString的代码和main函数：

#include "stdafx.h"
#include <iostream>
#include <cstring>
#include "MemB.h"
using namespace std;

class MyString{
    MemB *buf;
public:
    MyString();
    MyString(char * str);
    ~MyString();
    void concat(char * str);
    void print(ostream &os);
};

MyString::MyString(){
    buf = 0;
}

MyString::MyString(char * str){
    buf = new MemB(strlen(str)+1);
    strcpy((char *)buf->pointer(),str);
}

void MyString::concat(char * str){
    if(!buf)buf=new MemB;
    strcat((char*)buf->pointer(buf->msize()+strlen(str)+1),str);
}

void MyString::print(ostream & os){
    if(!buf)return;
    os<<buf->pointer()<<endl;
}

MyString::~MyString(){
    delete buf;
}

int _tmain(int argc, _TCHAR* argv[])
{
    MyString s("hello,Kate.");
    s.print(cout);
    s.concat("You will be good at it");
    s.print(cout);
    MyString s2;
    s2.concat("the second test");
    s2.print(cout);
    return 0;
}