memcpy記憶體複製程式碼的安全版本
阿新 • • 發佈:2019-01-26
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
void *Memcpy(void *dst, const void *src, size_t size);
int main(int argc, char *argv[])
{
char buf[100] = "vincentabcdefghijk";
//memcpy(buf+5, buf, 7);
Memcpy(buf+5, buf, 7);
printf("%s\n", buf);
}
/**
記憶體複製函式,將src記憶體地址開始的size個位元組複製到dst記憶體開始的地方
假如dst地址在src地址到(char *)src + size地址之間,那麼地址有重疊,那麼將覆蓋原來src中dst地址後面
的元素
dst不在src地址到(char *)src + size地址之間,則不會發生覆蓋
*/
void *Memcpy(void *dst, const void *src, size_t size)
{
const char *psrc;
char *pdst;
if(NULL == dst || NULL == src)
{
return NULL;
}
/**
dst 在src到src+count之間的記憶體空間重疊區域
*/
if((src < dst) && ((char *)src + size) > (char *)dst) // 如果有地址重疊時的記憶體拷貝,自後向前拷貝
{
psrc = (char *)src + size - 1;
pdst = (char *)dst + size - 1;
while(size--)
{
*pdst-- = *psrc--;
}
}
else
{
psrc = (char *)src;
pdst = (char *)dst;
while(size--)
{
*pdst++ = *psrc++;
}
}
return pdst;
}