C语言中strcat的实现方法

最近看到一道题目要求， 自己码代码实现strcat的功能， 于是自己实现了一个如下：

/* * 12.编写一个函数JOIN，让它实现字符串连接运算功能。 */

#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <string.h>

// ==============【自己实现的strcat】==============
char * join(char * str1, const char * str2)
{
    assert(str1 != NULL && str2 != NULL);

    char * pstr = str1;
    while (*pstr++);
    --pstr;

    while ((*pstr++ = *str2++) != 0);

    return str1;
}

int main()
{
#define N 20

    char buf[N] = "hello ";
    char * str2 = "world!";
    char buf1[N] = "hello ";

    char * res = join(buf, str2);
    char * res1 = strcat(buf1, str2);

    printf("%s\n", res);
    printf("%s\n", res1);

    return 0;

}

因为， 我们知道这个函数输入的第一个字符串是需要有足够内存空间的， 如果空间不够， 会引起栈崩溃的情况

于是我们试图查看一下标准库中的写法，下面是标准strcat.c的实现代码

/***
*strcat.c - contains strcat() and strcpy()
* * Copyright (c) Microsoft Corporation. All rights reserved.
* *Purpose:
* Strcpy() copies one string onto another.
* * Strcat() concatenates (appends) a copy of the source string to the
* end of the destination string, returning the destination string.
* *******************************************************************************/

#include <cruntime.h>
#include <string.h>

#ifndef _MBSCAT
#pragma function(strcat,strcpy)
#endif /* _MBSCAT */

/***
*char *strcat(dst, src) - concatenate (append) one string to another
* *Purpose:
* Concatenates src onto the end of dest.  Assumes enough
* space in dest.
* *Entry:
* char *dst - string to which "src" is to be appended
* const char *src - string to be appended to the end of "dst"
* *Exit:
* The address of "dst"
* *Exceptions:
* *******************************************************************************/

char * __cdecl strcat (
 char * dst,
 const char * src
 )
{
 char * cp = dst;

 while( *cp )
 cp++; /* find end of dst */

 while( *cp++ = *src++ ) ; /* Copy src to end of dst */

 return( dst ); /* return dst */

}


/***
*char *strcpy(dst, src) - copy one string over another
* *Purpose:
* Copies the string src into the spot specified by
* dest; assumes enough room.
* *Entry:
* char * dst - string over which "src" is to be copied
* const char * src - string to be copied over "dst"
* *Exit:
* The address of "dst"
* *Exceptions:
*******************************************************************************/

char * __cdecl strcpy(char * dst, const char * src)
{
 char * cp = dst;

 while( *cp++ = *src++ )
 ; /* Copy src over dst */

 return( dst );
}

很明显， 原始的strcat函数是没有对 可能出现的各种问题做检查的。也就是说，如果我们输入一个空指针， 程序是会直接崩溃掉的<-_->!!
难怪现在推出了一个strcat_s版本。

/***
*strcat_s.c - contains strcat_s()
* * Copyright (c) Microsoft Corporation. All rights reserved.
* *Purpose:
* strcat_s() concatenates (appends) a copy of the source string to the
* end of the destination string.
* *******************************************************************************/

#include <string.h>
#include <internal_securecrt.h>

#define _FUNC_PROLOGUE
#define _FUNC_NAME strcat_s
#define _CHAR char
#define _DEST _Dst
#define _SIZE _SizeInBytes
#define _SRC _Src

#include <tcscat_s.inl>

其中tcscat_s.inl内容如下

/*** *tcscat_s.inl - general implementation of _tcscpy_s * * Copyright (c) Microsoft Corporation. All rights reserved. * *Purpose: * This file contains the general algorithm for strcat_s and its variants. * ****/

_FUNC_PROLOGUE
errno_t __cdecl _FUNC_NAME(_CHAR *_DEST, size_t _SIZE, const _CHAR *_SRC)
{
    _CHAR *p;
    size_t available;

    /* validation section */
    _VALIDATE_STRING(_DEST, _SIZE);
    _VALIDATE_POINTER_RESET_STRING(_SRC, _DEST, _SIZE);

    p = _DEST;
    available = _SIZE;
    while (available > 0 && *p != 0)
    {
        p++;
        available--;
    }

    if (available == 0)
    {
        _RESET_STRING(_DEST, _SIZE);
        _RETURN_DEST_NOT_NULL_TERMINATED(_DEST, _SIZE);
    }

    while ((*p++ = *_SRC++) != 0 && --available > 0)
    {
    }

    if (available == 0)
    {
        _RESET_STRING(_DEST, _SIZE);
        _RETURN_BUFFER_TOO_SMALL(_DEST, _SIZE);
    }
    _FILL_STRING(_DEST, _SIZE, _SIZE - available + 1);
    _RETURN_NO_ERROR;
}

很明显可以看到， strcat_s 除了添加了一个参数之外， 还加入了根据这个参数进行检测的功能， 检测数据的有效性， 以及回滚操作。