如何從 C 函數安全返回數組並避免記憶體洩漏？

在 C 中傳回本地數組：避免記憶體洩漏

在 C 中，傳回本地數組可能會導致記憶體洩漏。考慮以下範例：

char *recvmsg() {
    char buffer[1024];
    return buffer;
}

int main() {
    char *reply = recvmsg();
    ...
}

這裡，recvmsg 函數傳回一個指向本機陣列緩衝區的指標。但是，當函數傳回時，該陣列將被銷毀，留下一個懸空指標。稍後訪問該指標將導致未定義的行為，可能導致記憶體洩漏。

回傳 std::vector

要解決此問題，可以傳回一個std::vector;而不是本機陣列。在std::vector中，資料是由容器本身動態分配和管理的。因此，返回 std::vector 將防止懸空指標問題。

std::vector<char> recvmsg() {
    std::vector<char> buffer(1024);
    ...
    return buffer;
}

int main() {
    std::vector<char> reply = recvmsg();
}

將char* 與std::vector 一起使用

如果需要使用char* 在程式碼的其他地方，您可以使用&reply[0] 來取得指向std::vector 第一個元素的指標。例如：

void f(const char* data, size_t size) {}

f(&reply[0], reply.size());

避免使用new

最後，建議盡量避免使用new。如果未正確釋放分配的內存，則使用 new 手動管理內存可能會導致內存洩漏。相反，依賴像 std::vector 這樣自動管理記憶體的容器。

以上是如何從 C 函數安全返回數組並避免記憶體洩漏？的詳細內容。更多資訊請關注PHP中文網其他相關文章！