《编写高质量代码:改善c程序代码的125个建议》——建议2-3:使用rsize_t或size_t类型来表示一个对象所占用空间的整数值单位...

本节书摘来自华章计算机《编写高质量代码:改善c程序代码的125个建议》一书中的第1章,建议2-3,作者:马 伟 更多章节内容可以访问云栖社区“华章计算机”公众号查看。

建议2-3:使用rsize_t或size_t类型来表示一个对象所占用空间的整数值单位

C语言标准规定size_t是一种无符号整数类型,编译器可以根据操作系统的不同而用typedef来定义不同的size_t类型,即在不同的操作系统上所定义的size_t可能不一样。例如在32位操作系统上可以将size_t定义为unsigned int类型,而在64位操作系统上则可以定义为unsigned long int类型,甚至还可以将size_t定义为unsigned long long int类型等,如下面的示例所示。
在GCC的stddef.h文件中将size_t定义为:

#ifndef __SIZE_TYPE__
#define __SIZE_TYPE__ long unsigned int
#endif
#if !(defined (__GNUG__) && defined (size_t))
typedef __SIZE_TYPE__ size_t;
#ifdef __BEOS__
typedef long ssize_t;
#endif /* __BEOS__ */
而在VC++ 2010的crtdefs.h文件中将size_t定义为:
#ifndef _SIZE_T_DEFINED
#ifdef  _WIN64
typedef unsigned __int64    size_t;
#else
typedef _W64 unsigned int   size_t;
#endif
#define _SIZE_T_DEFINED
#endif

从上面的定义可以看出,size_t类型的引入增强了程序在不同平台上的可移植性,而它也正是为了方便系统之间的移植而定义的。size_t类型的变量大小足以保证存储内存中对象的大小,任何表示对象长度的变量,包括作为大小、索引、循环计数和长度的整数值,都可以声明为size_t类型。比如我们常用的sizeof操作符的结果返回的就是size_t类型,该类型保证能容纳实现所建立的最大对象的字节大小。size_t类型的限制是由SIZE_MAX宏指定的。
接下来看看size_t类型的使用示例,如代码清单1-3所示。

代码清单1-3 size_t类型的使用示例
char *copy(size_t n, const char *str) 
{ 
    int i; 
    char *p; 
    if (n == 0) 
    { 
            /* 处理n==0的情况 */
    } 
    p = (char *)malloc(n); 
    if (p == NULL) 
    { 
            /*处理p==NULL的情况 */
    } 
    for ( i = 0; i < n; ++i ) 
    { 
            p[i] = *str++; 
    } 
    return p; 
}

不难发现,代码清单1-3中存在着一个严重的问题:当p所引用的动态分配的缓冲区在n > INT_MAX时将会发生溢出。我们知道,int 类型的限制是由INT_MAX宏指定的,而size_t类型代表的是一个无符号整数类型,它可能包含一个大于INT_MAX的值。因此,当n的值为0 < n <= INT_MAX时,执行循环n次,代码如预期一样正常运行;但当n的值为INT_MAX < n <= SIZE_MAX,且整型变量i的增值超过INT_MAX时,i的值将是从INT_MIN开始的负值。这时,p[i]所引用的内存位置是在p所引用的内存之前,这就会导致写入发生在数组边界之外。
因此,为了避免发生这种潜在性的错误,应该将变量i也声明成size_t类型,如代码清单1-4所示。

代码清单1-4 代码清单1-3的解决方法
char *copy(size_t n, const char *str) 
{ 
    size_t i; 
    char *p; 
    if (n == 0||n>SIZE_MAX) 
    { 
            /* 处理n==0的情况 */
    } 
    p = (char *)malloc(n); 
    if (p == NULL) 
    { 
            /*处理p==NULL的情况 */
    } 
    for ( i = 0; i < n; ++i ) 
    { 
        p[i] = *str++; 
    } 
    return p; 
}

除了size_t类型之外,ISO/IEC TR 24731-1:2007中引入了一种新类型rsize_t,虽然它被定义为size_t类型,但它明确地表示是用于保存单个对象的长度的。
在VC++ 2010的crtdefs.h文件中将rsize_t定义为:

#if __STDC_WANT_SECURE_LIB__
#ifndef _RSIZE_T_DEFINED
typedef size_t rsize_t;
#define _RSIZE_T_DEFINED
#endif
#endif

在支持rsize_t类型的代码中,你可以检查对象的长度,验证它不大于RSIZE_MAX(一个正常单个对象的最大长度),库函数也可以使用rsize_t进行输入校验。
在VC++ 2010的limits.h文件中将RSIZE_MAX定义为:

#if __STDC_WANT_SECURE_LIB__
#ifndef RSIZE_MAX
#define RSIZE_MAX SIZE_MAX
#endif
#endif

这样就消除了示例整数溢出的可能性,现在我们可以将代码清单1-3中的变量i声明成rsize _t类型,同时也可将参数n修改成rsize _t类型,并与RSIZE_MAX进行比较以验证数据的合法范围,如代码清单1-5所示。

代码清单1-5 代码清单1-3的rsize_t解决方法
char *copy(rsize_t n, const char *str) 
{ 
    rsize_t i; 
    char *p; 
    if (n == 0 || n > RSIZE_MAX) 
    { 
            /* 处理n==0|| n > RSIZE_MAX的情况 */
    } 
    p = (char *)malloc(n); 
    if (p == NULL) 
    { 
            /*处理p==NULL的情况 */
    } 
    for (i = 0; i < n; ++i) 
    { 
            p[i] = *str++; 
    } 
    return p; 
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值