zoukankan      html  css  js  c++  java
  • strlcpy和strlcat

    strncpy 等主要的问题还是虽然不会溢出,但是满了就不给缓冲区添加0结束符了,以前在项目里面自己还写了个 safe_strcpy 现在发现早就有了

    http://blog.csdn.net/linyt/article/details/4383328

    找了一下,代码可以在 libbsd 里面有

    /*
     * Appends src to string dst of size siz (unlike strncat, siz is the
     * full size of dst, not space left).  At most siz-1 characters
     * will be copied.  Always NUL terminates (unless siz <= strlen(dst)).
     * Returns strlen(src) + MIN(siz, strlen(initial dst)).
     * If retval >= siz, truncation occurred.
     */
    size_t
    strlcat(char *dst, const char *src, size_t siz)
    {
            char *d = dst;
            const char *s = src;
            size_t n = siz;
            size_t dlen;
     
            /* Find the end of dst and adjust bytes left but don't go past end */
            while (n-- != 0 && *d != '')
                    d++;
            dlen = d - dst;
            n = siz - dlen;
     
            if (n == 0)
                    return(dlen + strlen(s));
            while (*s != '') {
                    if (n != 1) {
                            *d++ = *s;
                            n--;
                    }
                    s++;
            }
            *d = '';
     
            return(dlen + (s - src));       /* count does not include NUL */
    }
     
    /*
     * Copy src to string dst of size siz.  At most siz-1 characters
     * will be copied.  Always NUL terminates (unless siz == 0).
     * Returns strlen(src); if retval >= siz, truncation occurred.
     */
    size_t
    strlcpy(char *dst, const char *src, size_t siz)
    {
            char *d = dst;
            const char *s = src;
            size_t n = siz;
     
            /* Copy as many bytes as will fit */
            if (n != 0) {
                    while (--n != 0) {
                            if ((*d++ = *s++) == '')
                                    break;
                    }
            }
     
            /* Not enough room in dst, add NUL and traverse rest of src */
            if (n == 0) {
                    if (siz != 0)
                            *d = '';              /* NUL-terminate dst */
                    while (*s++)
                            ;
            }
     
            return(s - src - 1);    /* count does not include NUL */
    }
  • 相关阅读:
    Sum Root to Leaf Numbers深度优先计算路径和
    Path Sum II深度优先找路径
    动态和静态链接库
    C/C++变量
    搜索
    基本格式
    随机数生成函数
    珍惜生命,我用Python 。今天开始学习Python
    在windows里hexo 博客创建步骤
    作为一个程序员,什么是脚本。必须要理解
  • 原文地址:https://www.cnblogs.com/peon/p/4457286.html
Copyright © 2011-2022 走看看