沉默杀手HomeArchivesTagsCategories
memset和memcpy函数
|C++C++ 基础|
Word count: 443|Reading time: 1 min

memset函数

原型：void *memset(void *s, int ch, size_t n);
作用：将s所指向的内存中的前n个字节的内容全部设置为ch指定的ASCII值，这个函数通常为新申请的内存做初始化工作。一般用于结构体和数组的初始化。

  1. memset中的第三个参数一定要使用sizeof操作符，因为每个系统下对类型长度的定义可能不一样。
  2. memset中的第一个参数一定要是一个已知的、已经被分配内存的地址，否则会出错。
  3. 对于单字节数据类型（char）可以初始化为任意支持的值，都没有问题，但是对于非多字节数据类型只能初始化为0，而不能初始化成别的初值，否则容易出错。

memset的效率很高，比手动赋值要高的多，比bzero也要高，尤其大数组的情况。

我是这样实现的:

1
2
3
4
5
6
7
8
9
10
11
void* _memset(void* dst,int val, size_t count)
{
	assert(dst!=NULL);
    char* tmpdst = (char*)dst;
    while(count--)
    {
        *tmpdst = (char)val;
        tmpdst++;
    }
    return dst;
}

memcpy函数

memcpy函数的使用场合是不需要考虑内存重叠问题的，因为涉及到内存重叠时我们应该调用的是memmove函数。我是这样实现的:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
void* _memcpy(void* dst, const void* src, size_t n)
{
    assert(dst!=NULL && src!=NULL);
    assert(n>=0);
    char* temp = (char*)dst;
    const char* p = (char*)src;
    size_t m=0; //void指针不能自增
    while(n--)
    {
        *temp = *p;
        temp++;
        p++;
    }
    return dst;
}

memmove进行了改进，考虑了内存重叠的情况:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
void* my_memmove(void* dst, const void* src, size_t n)
{
    char* s_dst;
    char* s_src;
    s_dst = (char*)dst;
    s_src = (char*)src;
    if(s_dst>s_src && (s_src+n>s_dst)) {
        s_dst = s_dst+n-1;
        s_src = s_src+n-1;
        while(n--) {
            *s_dst-- = *s_src--;
        }
    }else {
        while(n--) {
            *s_dst++ = *s_src++;
        }
    }
    return dst;
}