C/C++ 代码安全（一）

男娘i 2022-07-17 02:38 147阅读 0赞

规则1：
    
    不通过链接创建通用字符串。
    
    C11标准：如果匹配通用字符名称的字符序列由符号链接生成，将产生未定义行为。

#define assign(uc1,uc2,val) uc1##uc2 = val void func(){ int u0401; assign(u04,01,4); printf("u0401 = %d\n",u0401); } int main(){ func(); }

此结果是未定义的，不要使用链接诶字符串来赋值。 规则2： 避免使用不安全的宏参数

#define ABS(x)  (((x) < 0) ? -(x) : (x))
    
    void fun(int n){
        /*
         *   n++;
         *   ABS(n);  right
         */
        ABS(n++);/*erro*/
    
    }

注意n的加减。 解决方案一： 我们可以写一个内联函数或者写一个静态函数来代替此宏。 解决方案二： 使用范型编程\_Generic 参数,范型选择不求值，所以我们的值只会求一次。

#include<stdio.h>
    #include<complex.h>
    #include<math.h>
    
    static inline long labs(long v){
        return v < 0 ?  -v : v;
    }
    
    static inline int ilabs(int v){
        return v < 0 ? -v : v;
    }
    
    static inline short slabs(int v){
        return v < 0 ? -v : v;
    }
    
    
    #define ABS(v) _Generic(v,long : labs ,\
                              int  : ilabs ,\
                              short: slabs)(v)
    
    void fun(int n){
        int m = ABS(++n);
    }
    
    int main(){
    
        fun(10);
    }

解决方案三： 使用GCC 扩展，但是会带来平台问题。

#define ABS(x) __extension__ ({__typeof(x) tmp = x ;\
                                   tmp < 0 ? -tmp : tmp;})

规则： 不要在类函数的宏调用中使用预处理器指令

void func(const char * src){
    
        char *dest;
        memcpy(dest,src,
            #ifdef A
              12
            #else 
              13
            #endif);
    
    }

其实能写出这种代码的人脑洞也是大的没谁了，我感觉这都不关乎安全问题了。 解决方法很简单，写完整就好了。