在写windows C++代码的时候，从代码安全角度考虑，我们应该注意什么？

Thomas_Lbw2023-11-19 11:12

在写windows C++代码的时候，从代码安全角度考虑，我们应该注意什么？分别是：输入验证、内存管理、错误处理、并发和线程安全、使用安全的API、避免使用不安全的函数、最小权限原则。

一、输入验证

1. 用户输入验证

cpp 复制代码 
#include <iostream>
#include <string>
#include <limits>

int main() {
    int age;
    while (true) {
        std::cout << "请输入您的年龄: ";
        if (std::cin >> age && age > 0 && age < 130) {
            break;
        } else {
            std::cout << "无效输入，请重新输入。\n";
            std::cin.clear(); // 清除失败的输入
            std::cin.ignore(std::numeric_limits<std::streamsize>::max(), '\n'); // 忽略错误输入之后的字符
        }
    }

    std::cout << "输入的年龄是: " << age << std::endl;
    return 0;
}

2. 文件读取验证

当从文件中读取数据时，您需要验证文件是否存在，是否可以被打开，以及读取的数据是否符合预期格式。例如，读取一个存储数字的文件：

cpp 复制代码 
#include <fstream>
#include <iostream>
#include <vector>

int main() {
    std::ifstream file("numbers.txt");
    std::vector<int> numbers;
    int number;

    if (!file) {
        std::cerr << "无法打开文件" << std::endl;
        return 1;
    }

    while (file >> number) {
        numbers.push_back(number);
    }

    if (!file.eof()) {
        std::cerr << "文件读取过程中发生错误" << std::endl;
        return 1;
    }

    std::cout << "读取到的数字有: ";
    for (int num : numbers) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

3. 网络数据接收验证

在处理网络通信时，接收到的数据应该经过严格验证。例如，如果您的程序是一个服务器，它可能需要验证从客户端接收到的消息格式是否正确：

cpp 复制代码 
// 假设这是一个简单的TCP服务器接收数据的代码片段
char buffer[1024];
int receivedBytes = recv(clientSocket, buffer, 1024, 0);
if (receivedBytes > 0) {
    // 验证接收到的数据
    if (isValidMessage(buffer, receivedBytes)) {
        // 处理消息
    } else {
        std::cerr << "接收到无效的消息格式" << std::endl;
    }
}

二、内存管理

1. 使用裸指针管理动态内存（传统方法）

cpp 复制代码 
int* ptr = new int(10); // 分配内存
// 使用 ptr...
delete ptr; // 释放内存
ptr = nullptr; // 防止悬挂指针

在这个例子中，必须确保在 ptr 不再需要时释放分配的内存，并将指针设为 nullptr 以避免悬挂指针。这种方法容易出错，因为需要手动管理内存。

2. 使用智能指针管理动态内存（现代方法）

C++11 引入的智能指针（如 std::unique_ptr 和 std::shared_ptr）自动管理内存，可以减少内存泄漏的风险。

cpp 复制代码 
#include <memory>

std::unique_ptr<int> ptr(new int(10));
// 使用 ptr...
// 不需要手动释放内存，当 ptr 离开作用域时，内存会自动被释放

3. 防止数组越界

4. 避免内存泄漏

cpp 复制代码 
void function() {
    int* ptr = new int(10); // 分配内存
    // 函数的其他操作...
    delete ptr; // 释放内存
}

如果函数中有多个返回路径或可能抛出异常，使用智能指针来保证内存在任何情况下都能被正确释放。

三、错误处理机制

1. 使用异常处理机制

C++ 提供了异常处理机制，允许在检测到错误时抛出异常，并在上层代码中捕获和处理这些异常。

cpp 复制代码 
#include <iostream>
#include <stdexcept>

double divide(double a, double b) {
    if (b == 0) {
        throw std::invalid_argument("除数不能为0");
    }
    return a / b;
}

int main() {
    try {
        double result = divide(10.0, 0.0);
        std::cout << "结果: " << result << std::endl;
    } catch (const std::invalid_argument& e) {
        std::cerr << "捕获到错误: " << e.what() << std::endl;
    }
    return 0;
}

2. 使用返回值进行错误指示

在某些情况下，使用返回值来指示错误是更加合适的。这种方法常用于 C 风格的代码或者需要保持与旧代码的兼容性。

cpp 复制代码 
#include <iostream>

bool safeDivide(double a, double b, double& result) {
    if (b == 0) {
        return false; // 错误指示
    }
    result = a / b;
    return true; // 操作成功
}

int main() {
    double result;
    if (!safeDivide(10.0, 0.0, result)) {
        std::cerr << "错误：除数不能为0" << std::endl;
    } else {
        std::cout << "结果: " << result << std::endl;
    }
    return 0;
}

3. 使用错误码

四、最小权限原则

最小权限原则（Principle of Least Privilege, PoLP）是一种安全设计原则，意在确保程序、进程或用户仅具有完成其任务所必需的最小权限集。在 Windows C++ 编程中，这通常涉及到操作系统资源的访问和管理。以下是一些应用最小权限原则的例子：

1. 运行权限限制

当开发一个应用程序时，确保它以最低必要的权限运行。例如，如果您的程序仅需读取文件，不应请求或具有写入文件的权限。

cpp 复制代码 
// 以只读方式打开文件
std::ifstream file("example.txt");
if (!file) {
    std::cerr << "无法打开文件，权限不足或文件不存在" << std::endl;
    // 错误处理
}

2. 数据库访问

当您的程序需要与数据库交互时，为数据库用户分配只能执行其需求的操作的最小权限。例如，如果程序只需要从数据库读取数据，那么数据库用户不应该具有写入、修改或删除数据的权限。

cpp 复制代码 
// 假设这是一个数据库连接代码片段
// 这个数据库用户只有读取数据的权限
const char* connectionString = "User=ReadOnlyUser;Password=example;...";

3. 网络服务权限

如果您的程序是一个网络服务，确保它在一个受限的环境中运行，例如在低权限的用户账户下运行，避免在需要管理员权限的账户下运行。

4. 文件和目录权限

当您的程序需要创建文件或目录时，设置合理的访问控制列表（ACL），以确保只有必要的用户或程序有权访问这些资源。

五、并发和线程安全

在并发编程中，线程安全是一个核心考虑因素。线程安全的代码可以在多线程环境中安全地被多个线程同时执行，而不会导致数据损坏或不一致的状态。以下是一些并发和线程安全的例子：

1. 使用互斥锁

互斥锁（mutex）是保护共享资源免受多个线程同时访问的一种方法。

cpp 复制代码 
#include <iostream>
#include <thread>
#include <mutex>

std::mutex mtx; // 全局互斥锁
int sharedData = 0; // 共享数据

void increment() {
    mtx.lock(); // 获取锁
    ++sharedData; // 修改共享数据
    mtx.unlock(); // 释放锁
}

int main() {
    std::thread t1(increment);
    std::thread t2(increment);

    t1.join();
    t2.join();

    std::cout << "共享数据的值: " << sharedData << std::endl;
    return 0;
}

2. 使用条件变量

条件变量是一种允许线程等待特定条件的同步机制。

cpp 复制代码 
#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>

std::mutex mtx;
std::condition_variable cv;
bool ready = false;

void printNumber(int num) {
    std::unique_lock<std::mutex> lck(mtx);
    while (!ready) {
        cv.wait(lck);
    }
    std::cout << "Number: " << num << std::endl;
}

void setReady() {
    {
        std::lock_guard<std::mutex> lck(mtx);
        ready = true;
    }
    cv.notify_all();
}

int main() {
    std::thread t1(printNumber, 1);
    std::thread t2(printNumber, 2);

    setReady();

    t1.join();
    t2.join();

    return 0;
}

在这个例子中，两个线程等待 ready 变量变为 true。一旦 setReady 函数被调用，条件变量通知所有等待的线程继续执行。

3. 使用原子操作

原子操作是不可分割的操作，可以在多线程环境中安全地执行，不需要额外的同步机制。

cpp 复制代码 
#include <iostream>
#include <thread>
#include <atomic>

std::atomic<int> count(0); // 原子变量

void increment() {
    for (int i = 0; i < 10000; ++i) {
        count++; // 原子操作
    }
}

int main() {
    std::thread t1(increment);
    std::thread t2(increment);

    t1.join();
    t2.join();

    std::cout << "计数: " << count << std::endl;
    return 0;
}

在这个例子中，两个线程同时增加一个计数器。由于 count 是一个原子类型，每次增加操作都是线程安全的。

六、使用安全的函数（Windows api）

这里只需注意需要使用安全的。

1. 字符串复制

cpp 复制代码 
char source[] = "Hello World";
char dest[11];
strcpy(dest, source); // 不安全，没有边界检查


char source[] = "Hello World";
char dest[11];
strncpy(dest, source, sizeof(dest) - 1); // 安全，指定最大复制长度
dest[sizeof(dest) - 1] = '\0'; // 确保字符串以空字符结尾

strcpy 的问题

strcpy 函数用于将一个字符串复制到另一个字符串。它继续复制字符直到遇到源字符串的空终止字符（'\0'）。这个函数的问题在于它不考虑目标字符串的大小。如果源字符串长度超过了目标字符串的容量，strcpy 会继续写入，导致超出目标数组的边界，造成缓冲区溢出。这种溢出可能覆盖其他重要的内存数据，引起程序崩溃或安全漏洞。

char source[] = "这个字符串很长，超过目标缓冲区的大小"; char dest[10]; // 只有10个字符的空间 strcpy(dest, source); // 危险：源字符串长度超过了dest的容量

在这个例子中，dest 只能容纳9个字符和一个空终止符，但 source 的长度远远超过这个限制，导致 dest 的边界被溢出。

strncpy 的相对安全性

strncpy 函数也是用于复制字符串，但它接受一个额外的参数来指定最大复制的字符数。这个函数在到达最大字符数或遇到源字符串的空终止符时停止复制。这有助于防止超出目标数组的边界，如果正确使用的话。

cpp 复制代码 
char source[] = "这个字符串很长，可能超过目标缓冲区的大小";
char dest[10];
strncpy(dest, source, sizeof(dest) - 1); // 复制最多9个字符
dest[sizeof(dest) - 1] = '\0'; // 确保有空终止符

strcpy 的问题

strcpy 函数用于将一个字符串复制到另一个字符串。它继续复制字符直到遇到源字符串的空终止字符（'\0'）。这个函数的问题在于它不考虑目标字符串的大小。如果源字符串长度超过了目标字符串的容量，strcpy 会继续写入，导致超出目标数组的边界，造成缓冲区溢出。这种溢出可能覆盖其他重要的内存数据，引起程序崩溃或安全漏洞。

strncpy 的相对安全性

strncpy 函数也是用于复制字符串，但它接受一个额外的参数来指定最大复制的字符数。这个函数在到达最大字符数或遇到源字符串的空终止符时停止复制。这有助于防止超出目标数组的边界，如果正确使用的话。

例子：使用 strncpy 防止缓冲区溢出

2. 字符串连接

cpp 复制代码 
char str[10] = "Hello";
strcat(str, " World"); // 不安全，可能导致缓冲区溢出


char str[15] = "Hello";
strncat(str, " World", sizeof(str) - strlen(str) - 1); // 安全，限制最大添加长度

3. 格式化字符串

cpp 复制代码 
char buffer[50];
sprintf(buffer, "Value: %d", 123); // 不安全，没有边界检查

char buffer[50];
snprintf(buffer, sizeof(buffer), "Value: %d", 123); // 安全，限制最大写入长度

虽然 strncpy 比 strcpy 安全，但它仍然需要小心使用。如果最大复制长度小于源字符串长度，strncpy 不会自动添加空终止符，可能导致目标字符串没有正确终止。因此，使用 strncpy 时，程序员需要确保目标字符串有足够的空间，并在需要时手动添加空终止符。

相关推荐
汉字萌萌哒4 分钟前
【2022 CCF 非专业级别软件能力认证第一轮（CSP-J1）入门级 C++语言试题及解析】
数据结构·c++·算法
th新港5 分钟前
CCF201909_1
数据结构·c++·算法·ccf
意如流水任东西5 分钟前
[C++]类和对象（上）
开发语言·c++
孤寂大仙v11 分钟前
【C++】STL----stack和queue常见用法
开发语言·c++
一丝晨光30 分钟前
逻辑运算符
java·c++·python·kotlin·c#·c·逻辑运算符
Code哈哈笑2 小时前
【C++ 学习】多态的基础和原理（10）
java·c++·学习
安全方案2 小时前
政务安全运营核心能力模块
安全
Aurora_th2 小时前
树与图的深度优先遍历（dfs的图论中的应用）
c++·算法·深度优先·图论·dfs·树的直径
sysin.org2 小时前
VMware ESXi 7.0U3q macOS Unlocker 集成驱动版更新 OEM BIOS 2.7 支持 Windows Server 2025
windows·macos·esxi·bios·oem·网卡驱动·nvme驱动
shuxianshrng3 小时前
大气网格化精细化监管监测系统
大数据·服务器·windows·经验分享
热门推荐
01RAG 实践- Ollama+RagFlow 部署本地知识库02组基轨迹建模 GBTM的介绍与实现（Stata 或 R）032024年高教社杯数学建模国赛C题超详细解题思路分析04苍穹外卖面试总结05yolov8实战第五天——yolov8+ffmpg实时视频流检测并进行实时推流——（推流，保姆教学）06【2024数模国赛赛题思路公开】国赛B题思路丨附可运行代码丨无偿自提07CCF-CSP认证考试 202406-3 文本分词 100分题解08Coze扣子平台完整体验和实践（附国内和国际版对比）09【2024高教社杯全国大学生数学建模竞赛】B题 生产过程中的决策问题——解题思路 代码 论文10软件工程从理论到实践客观题汇总(头歌第一章至第八章)