strcpy的使用--安全编程策略

#include 
#include 
#include 
char* myStrcpy(char* dst,const char* src)
{
    char* ret = dst;
    assert(dst&&src);//安全编程策略,一定要写这一句代码,确保dst和src都不为空

    while((*dst++ = *src++)!='\0');//遇到\0就返回

    return ret;
}
int main(void)
{
    char* s1 = (char*)malloc(10*sizeof(char));
    char* s2 = "hello";

    char* s3 = (char*)malloc(2*sizeof(char));
    char* s4 = "1234";

    printf("%s\n",myStrcpy(s1,s2));

    printf("%s\n",myStrcpy(s3,s4));  //不安全,s3申请的空间大小小于s4的空间大小,复制的时候容易发生溢出

    printf("-----end------\n");
    return 0;
}

安全的字符串编程:

#include 
#include 
#include 
#include 

size_t myStrlen(const char* s)
{
    size_t length = 0;
    assert(s);//安全编程策略,一定要写这一句代码,确保s不为空
    while(*s++)   //遇到\0就返回
    {
        length++;
    }

    return length;
}



char* myStrcpy(char* dst,const char* src)
{
    char* ret = dst;
    assert(dst&&src);//安全编程策略,一定要写这一句代码,确保dst和src都不为空

    while((*dst++ = *src++)!='\0');//遇到\0就返回

    return ret;
}


void test_1()
{
    char* s1 = "hello";
    char* s2 = "hellow";
    //不要使用 myStrlen(s1) - myStrlen(s2) > 0,因为size_t是unsigned int,相减之后不可能产生负值
    if(myStrlen(s1) > myStrlen(s2))
    {
        printf("the length of s1 is long\n");
    }
    else if(myStrlen(s1) < myStrlen(s2))
    {
        printf("the length of s2 is long\n");
    }
    else
    {
        printf("s1 = s2\n");
    }

}

int main(void)
{
    char* s1 = (char*)malloc(10*sizeof(char));
    char* s2 = "hello";

    char* s3 = (char*)malloc(2*sizeof(char));
    char* s4 = "1234";

    printf("%s\n",myStrcpy(s1,s2));

    printf("%s\n",myStrcpy(s3,s4));  //不安全,s3申请的空间大小小于s4的空间大小,复制的时候容易发生溢出

    //安全的字符串复制方法:char* strncpy(char* dst,const char* src,size_t len)
    int length = strlen(s1)>strlen(s2)?strlen(s2):strlen(s1);
    strncpy(s1,s2,length); //(strlen(s1)>strlen(s2):strlen(s2)?strlen(s1))
    printf("%s\n",s1);

    //安全的字符串连接方法:char* strncat(char* dst,const char* src,size_t len)
    //安全的字符串比较方法:int strncmp(char* dst,const char* src,size_t len)

    printf("-----end------\n");
    return 0;
}

你可能感兴趣的:(C语言)