32. 乾貨系列從零用Rust編寫正反向代理，關於堆和棧以及如何解決stack overflow

来源:https://www.cnblogs.com/wmproxy/archive/2023/12/04/wmproxy32.html
-Advertisement-
Play Games

記憶體是在程式中打交道不可缺少的存在,或者在GC語言中記憶體概念會被刻意的屏蔽掉,但如果是棧遞歸函數調用自身這種情況,stack overflow這種情況是一樣的會碰到的,所以瞭解下堆與棧碰到問題的時候好解決問題 ...


wmproxy

wmproxy已用Rust實現http/https代理, socks5代理, 反向代理, 靜態文件伺服器，四層TCP/UDP轉發，七層負載均衡，內網穿透，後續將實現websocket代理等，會將實現過程分享出來，感興趣的可以一起造個輪子

項目地址

國內: https://gitee.com/tickbh/wmproxy

github: https://github.com/tickbh/wmproxy

關於棧Stack

Stack可以被認為是一堆書。當我們添加更多的書時，我們將它們添加到棧的頂部。當我們需要一本書時，我們從上面拿一本。

  • 添加數據稱為壓入棧
  • 移除數據稱為彈出棧
    這種現象在編程中被稱為後進先出（LIFO）。
    存儲在棧上的數據在編譯時必須具有固定的大小。預設情況下，Rust在棧上為原始類型分配記憶體。所有存儲在堆棧上的數據必須具有已知的固定大小。未知數據編譯時的大小或可能更改的大小必須存儲在堆中而不是棧中。

關於堆Heap

與棧相反，大多數情況下，我們需要將變數（記憶體）傳遞給不同的函數，並使它們保持比單個函數執行更長的時間。這就是我們可以使用heap的時候。

堆的組織性較差：當您將數據放在堆上時，您會請求一個一定的空間。記憶體分配器在堆中找到一個空位這是足夠大的，標志著它正在使用，並返回一個指針，就是那個地方的地址此過程稱為在堆，有時縮寫為分配（將值推到堆棧不被認為是分配的）。因為指向堆的指針是已知的，固定大小的，你可以把指針存儲在堆棧上，但是當你想要的時候，實際數據，您必須遵循指針。想象一下坐在一個餐廳當你進入時，你說明你的小組人數，主人會找到一張適合所有人的空桌子，然後把你帶到那裡。如果如果你的團隊中有人遲到了，他們可以問你坐在哪裡，找到你。

棧與堆對比

  • 分配到棧比在堆上分配更快，因為分配器永遠不必搜索存儲新數據的位置;該位置總是在棧的頂部。相比之下，在堆上分配空間需要更多的工作，因為分配器必須首先找到足夠大的空間，保存數據，然後進行簿記，為下一次配置。
  • 在堆中訪問數據比訪問棧上的數據慢，因為你得跟著指示牌走。因為訪問堆需要得到相應的指示牌，然後再根據相應的指示牌去尋找相應的位置，然後還要確定位置所占的大小。
statck棧 heap堆
在棧中存儲數據的速度更快。 在堆中存儲數據的速度較慢。
管理棧中的記憶體是可預測的，也是微不足道的。 管理堆的記憶體（任意大小）是非常重要的。
Rust堆棧預設分配。 Box用於分配到堆。
函數的基元類型和局部變數在棧上分配。 大小動態的數據類型，如String、Vector、HashMap、Box等，在heap上分配。

棧與堆的分配示例

讓我們通過一個例子來直觀地瞭解記憶體是如何在堆棧上分配和釋放的。

fn foo() {
    let y = 999;
    let z = 333;
}

fn main() {
    let x = 111;
    
    foo();
}

在上面的例子中，我們首先調用函數main()。main()函數有一個變數綁定x。

Address地址 Name名稱 Value值
0 x 111

在表中，“地址”列指的是RAM的記憶體地址。它從0開始，並轉到您的電腦有多少RAM（位元組數）。“名稱”列是指變數，“值”列是指變數的值。

當foo()被調用時，一個新的棧幀被分配。foo()函數有兩個變數綁定，y和z。

Address地址 Name名稱 Value值
2 z 333
1 y 999
0 x 111

數字0、1和2不使用電腦實際使用的地址值。實際上，地址根據值由一定數量的位元組分隔。

foo()完成後，其棧幀被釋放。

Address地址 Name名稱 Value值
0 x 111

main()完成後，其棧幀被釋放。Rust自動在堆棧中分配和釋放記憶體。

與堆棧相反，大多數情況下，我們需要將變數（記憶體）傳遞給不同的函數，並使它們保持比單個函數執行更長的時間。這就是我們可以使用heap的時候。

我們可以使用Box<T>類型在堆上分配記憶體。比如說，

fn main() {
    let x = Box::new(100);
    let y = 222;
    
    println!("x = {}, y = {}", x, y);
}

讓我們可視化在上面的例子中調用main()時的記憶體。

Address地址 Name名稱 Value值
0 x ??? addr
1 y 222

和前面一樣，我們在堆棧上分配兩個變數x和y。
然而，當調用x時，Box::new()的值被分配在堆上。因此，x的實際值是指向堆的指針。

Address地址 Name名稱 Value值
578 100
... ... ...
0 x -> 578
1 y 222

這裡，變數x保存指向地址→578，這是用於演示的任意地址。堆可以以任何順序分配和釋放。因此，它可能會以不同的地址結束，併在地址之間產生漏洞。

因此，當x消失時，它首先釋放堆上分配的記憶體。

Address地址 Name名稱 Value值
... ... ...
1 y 222

一旦main()完成，我們釋放堆棧幀，所有東西都消失了，釋放了所有記憶體。

如何排查問題

堆記憶體的排查

關於堆記憶體的排查，堆記憶體的記憶體量比較大，因此數值相對會大很多，堆記憶體的大小通常小到幾M，大到幾個G，所以在堆記憶體排查的時候可以用巨集觀的記憶體管理器，有以下幾種方法

  • 如TOP查看記憶體，也可以通過調用系統的api，
  • 如memory-stats實時查看進程當前占用記憶體數：
use memory_stats::memory_stats;

fn main() {
    if let Some(usage) = memory_stats() {
        println!("Current physical memory usage: {}", usage.physical_mem);
        println!("Current virtual memory usage: {}", usage.virtual_mem);
    } else {
        println!("Couldn't get the current memory usage :(");
    }
}
  • 可以自定義Alloc，因為Rust提供的全局global_alloc，我們可以通過自定義Alloc計算當前申請的記憶體數，以及可以用這種方式檢查記憶體泄漏，典型的jemalloc就是通過這種方式來的，我們用這種方式實現簡單的記憶體統計，我們定義了一個Trallocator：

use std::alloc::{GlobalAlloc, Layout};
use std::sync::atomic::{AtomicU64, Ordering};

pub struct Trallocator<A: GlobalAlloc>(pub A, AtomicU64);

unsafe impl<A: GlobalAlloc> GlobalAlloc for Trallocator<A> {
    unsafe fn alloc(&self, l: Layout) -> *mut u8 {
        self.1.fetch_add(l.size() as u64, Ordering::SeqCst);
        self.0.alloc(l)
    }
    unsafe fn dealloc(&self, ptr: *mut u8, l: Layout) {
        self.0.dealloc(ptr, l);
        self.1.fetch_sub(l.size() as u64, Ordering::SeqCst);
    }
}

impl<A: GlobalAlloc> Trallocator<A> {
    pub const fn new(a: A) -> Self {
        Trallocator(a, AtomicU64::new(0))
    }

    pub fn reset(&self) {
        self.1.store(0, Ordering::SeqCst);
    }
    pub fn get(&self) -> u64 {
        self.1.load(Ordering::SeqCst)
    }
}

我們通過調用該類，實現

use std::alloc::System;

// 這句使全局的的分配器變成我們自己的分配器
#[global_allocator]
static GLOBAL: Trallocator<System> = Trallocator::new(System);

fn main() {
    GLOBAL.reset();
    println!("memory used: {} bytes", GLOBAL.get());
    GLOBAL.reset();
    {
        let mut vec = vec![1, 2, 3, 4];
        for i in 5..20 {
            vec.push(i);
            println!("memory used: {} bytes", GLOBAL.get());
        }
        println!("{:?}", v);
    }
    
    println!("memory used: {} bytes", GLOBAL.get());
}

我們可以得到以下輸出：

memory used: 0 bytes
memory used: 32 bytes
memory used: 32 bytes
memory used: 32 bytes
memory used: 32 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 64 bytes
memory used: 128 bytes
memory used: 128 bytes
memory used: 128 bytes
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
memory used: 0 bytes

可以看到分配完之後已經及時釋放

棧記憶體的排查

因為系統提供的棧記憶體通常只有8m左右，且Rust中的線程的預設棧記憶體只有2M，如果分配過大的棧記憶體將會導致棧溢出，比如

fn main() {
    let bad = [0;10240000];
}

就會出現如下提示

thread 'main' has overflowed its stack
fatal runtime error: stack overflow

在現在的方法中，我並未找到有合適的檢查當前進程占用的棧記憶體數。

  • 測試用alloc看是否能測出棧記憶體：
use std::alloc::System;
#[global_allocator]
static GLOBAL: Trallocator<System> = Trallocator::new(System);

fn main() {
    GLOBAL.reset();
    println!("memory used: {} bytes", GLOBAL.get());
    GLOBAL.reset();
    let x = 0;
    let bad = [0;10240];
    println!("memory used: {} bytes", GLOBAL.get());
}

運行上述程式，如下輸出：

memory used: 0 bytes
memory used: 0 bytes

程式無法感知到棧記憶體的變化。

  • 測試用memory-stats實時查看記憶體
use memory_stats::memory_stats;

fn main() {

    if let Some(usage) = memory_stats() {
        println!("初始記憶體     usage: {}", usage.physical_mem);
    } else {
        println!("Couldn't get the current memory usage :(");
    }
    
    let value1 = vec![10;102400];
    
    std::thread::sleep(std::time::Duration::from_secs(1));

    if let Some(usage) = memory_stats() {
        println!("申請堆記憶體後 usage: {}", usage.physical_mem);
    } else {
        println!("Couldn't get the current memory usage :(");
    }
    
    let value = [10;102400];
    
    std::thread::sleep(std::time::Duration::from_secs(1));
    if let Some(usage) = memory_stats() {
        println!("申請棧記憶體後 usage: {}", usage.physical_mem);
    } else {
        println!("Couldn't get the current memory usage :(");
    }
    
}

以上程式會輸出：

初始記憶體     usage: 1024000
申請堆記憶體後 usage: 1478656
申請棧記憶體後 usage: 1478656

我們可以感知到堆記憶體的變化，無法感知到棧記憶體的變化。

  • 目前找到的可以測量類對象的棧記憶體值。可以用std::mem::size_of_val來測量類對象占用的棧記憶體大小，我們可以通過該方法進行棧大小的排查，看是否存在超級大的占用棧的對象，如果存在，需將其移動到堆，也就是用Box進行包裹。
fn main() {
    let x = 0u32;
    assert_eq!(4, std::mem::size_of_val(&x));
    let val = vec![0u64;9999];
    assert_eq!(24, std::mem::size_of_val(&val));
    
    let mut hash = HashMap::new();
    hash.insert(1, 2);
    assert_eq!(48, std::mem::size_of_val(&hash));
    hash.insert(2, 4);
    assert_eq!(48, std::mem::size_of_val(&hash));
}

我們來分析下Vec的記憶體，為什麼其占用大小為24個位元組（64位的機器）

pub struct Vec<T, A: Allocator = Global> {
    buf: RawVec<T, A>, /// 需要再進行類的分析
    len: usize, /// 占用64位，也就是8個位元組
}

pub(crate) struct RawVec<T, A: Allocator = Global> {
    ptr: Unique<T>, /// 指針大小，占用64位，8位元組
    cap: usize, /// 容量大小，占用64位，8位元組
    alloc: A, /// 分配器，不占用棧記憶體
}

綜上分析，每個Vec的棧大小占用記憶體均為24位元組。程式測試一致。同樣HashMap占用的棧大小均為48個位元組，不受其Map大小的影響。

註意：如果用非同步的Future的包圍，如果返回的對象也就是Furture<Output=xxx>的棧大小過大，很容易在遞進處理非同步的情況下直接棧溢出，而此時完全還未執行到該函數，造成一種很難排查的景象
註意！！！非同步的返回值千萬棧大小不要過大！不要過大！不要過大！

  • 另外還有一種是遞歸的函數調用，也會造成棧溢出，這類問題相對好定位：
fn f(x: i32) {
    f(1);
}
fn main() {
    f(2);
}

直接會顯示

thread 'main' has overflowed its stack
fatal runtime error: stack overflow

小結

所以在排查記憶體泄漏還是排查棧大小時都需要對當前的數據進行分析，需要處理的東西較多，需要有比較好的耐心去處理，一步步的去排查推進。記得非同步返回的Output如果過大，會導致代碼還未執行，但已經棧溢出的情況。

點擊 [關註]，[在看]，[點贊] 是對作者最大的支持


您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • python跳出多層for迴圈的方法
    在業務邏輯中有時候會遇到兩層for迴圈的情況，觸發某些條件時，需要直接退出兩層for迴圈 而python官方是沒有 goto 語句的那麼我們可以這樣實現 第一種定義變數flag，根據flag的值做退出 flag=True for i in range(10): for j in range(10): ...
  • 如何查看python中某些包的方法介紹？
    當你使用一個不存在的鍵（key）去訪問一個Python字典（dict）時，會觸發一個KeyError異常。這是Python提供的一種機制，用於指示你正在嘗試訪問一個字典中不存在的鍵。 以下是一個簡單的示例，演示了當使用一個不存在的鍵去訪問字典時會發生的情況： my_dict = {"apple": ...
  • Java開發者的Python快速實戰指南：探索向量資料庫之圖像相似搜索-文字版
    在開始講解之前，我想給大家介紹一個很有用的第三方包，它就是gradio。如果你想與他人共用你的機器學習模型、API或數據科學工作流的最佳方式之一，可以創建一個互動式應用，讓用戶或同事可以在瀏覽器中試用你的演示。而gradio正是可以幫助你在Python中構建這樣的演示，並且只需要幾行代碼即可完成！ ...
  • Python 實現Word轉HTML
    將Word轉換為HTML能將文檔內容發佈在網頁上，這樣，用戶就可以通過瀏覽器直接查看或閱讀文檔而無需安裝特定的軟體。Word轉HTML對於線上發佈信息、創建線上文檔庫以及構建互動式網頁應用程式都非常有用。以下是使用Python將Word轉換為HTML網頁的攻略，包含兩個示例。 使用Python 將W ...
  • 關於函數巨集offset_of 和 container_of的學習
    機器學習的第一步是準備數據，好的數據能幫助我們加深對機器學習演算法的理解。 不管是在學習還是實際工作中，準備數據永遠是一個枯燥乏味的步驟。scikit-learn庫顯然看到了這個痛點，才在它的數據載入子模塊中為我們準備了直接可用的數據集。 在它的數據載入子模塊中，提供了6種直接可用來學習演算法的經典數據 ...
  • 這下對阿裡java這幾條規範有更深理解了
    背景 阿裡java開發規範是阿裡巴巴總結多年來的最佳編程實踐，其中每一條規範都經過仔細打磨或踩坑而來，目的是為社區提供一份最佳編程規範，提升代碼質量，減少bug。 這基本也是java業界都認可的開發規範，我們團隊也是以此規範為基礎，在結合實際情況，補充完善。最近在團隊遇到的幾個問題，加深了我對這份開 ...
  • c語言實現this指針效果
    概要 由於目前在做一個比較複雜的嵌入式項目，想要藉此提升一下代碼的結構設計能力，所以想要以面向對象的思想來完成這個項目，即把每個板載外設資源視為一個對象，採用msp+bsp的模式，對每個bsp外設實現對象化處理，現有方案需要手動傳入對象引用，調用方法時比較麻煩，所以考慮簡化調用方式。 面向對象實現思 ...
  • java線上運行異常：Error parsing HTTP request header
    1.部署異常如下： 2.出現原因 這個問題的原因是高版本的tomcat中的新特性：就是嚴格按照 RFC 3986規範進行訪問解析，而 RFC 3986規範定義了Url中只允許包含英文字母（a-zA-Z）、數字（0-9）、-_.~4個特殊字元以及所有保留字元(RFC3986中指定了以下字元為保留字元： ...
一周排行
    -Advertisement-
    Play Games
  • 移動開發（一）：使用.NET MAUI開發第一個安卓APP
    移動開發（一）：使用.NET MAUI開發第一個安卓APP 對於工作多年的C#程式員來說，近來想嘗試開發一款安卓APP，考慮了很久最終選擇使用.NET MAUI這個微軟官方的框架來嘗試體驗開發安卓APP，畢竟是使用Visual Studio開發工具，使用起來也比較的順手，結合微軟官方的教程進行了安卓 ...
  • wpf ToggleButton選中效果和一個登錄界面
    前言 QuestPDF 是一個開源 .NET 庫，用於生成 PDF 文檔。使用了C# Fluent API方式可簡化開發、減少錯誤並提高工作效率。利用它可以輕鬆生成 PDF 報告、發票、導出文件等。 項目介紹 QuestPDF 是一個革命性的開源 .NET 庫，它徹底改變了我們生成 PDF 文檔的方 ...
  • 彈幕樹洞項目功能新增篇
    項目地址 項目後端地址： https://github.com/ZyPLJ/ZYTteeHole 項目前端頁面地址： ZyPLJ/TreeHoleVue (github.com) https://github.com/ZyPLJ/TreeHoleVue 目前項目測試訪問地址： http://tree ...
  • 第27篇 sqlserver2022詳細安裝步驟
    話不多說，直接開乾 一.下載 1.官方鏈接下載： https://www.microsoft.com/zh-cn/sql-server/sql-server-downloads 2.在下載目錄中找到下麵這個小的安裝包 SQL2022-SSEI-Dev.exe，運行開始下載SQL server； 二. ...
  • .NET 開源高性能 MQTT 類庫
    前言 隨著物聯網（IoT）技術的迅猛發展，MQTT（消息隊列遙測傳輸）協議憑藉其輕量級和高效性，已成為眾多物聯網應用的首選通信標準。 MQTTnet 作為一個高性能的 .NET 開源庫，為 .NET 平臺上的 MQTT 客戶端與伺服器開發提供了強大的支持。 本文將全面介紹 MQTTnet 的核心功能 ...
  • Serilog文檔翻譯系列（六） - 可用的接收器、增強器、格式化輸出
    Serilog支持多種接收器用於日誌存儲，增強器用於添加屬性，LogContext管理動態屬性，支持多種輸出格式包括純文本、JSON及ExpressionTemplate。還提供了自定義格式化選項，適用於不同需求。 ...
  • 警惕 Visual Studio 屬性求值副作用導致邏輯不符合預期
    目錄簡介獲取 HTML 文檔解析 HTML 文檔測試參考文章 簡介 動態內容網站使用 JavaScript 腳本動態檢索和渲染數據，爬取信息時需要模擬瀏覽器行為，否則獲取到的源碼基本是空的。 本文使用的爬取步驟如下： 使用 Selenium 獲取渲染後的 HTML 文檔 使用 HtmlAgility ...
  • [使用目前最新版]HybridCLR6.9.0+YooAsset2.2.4實現純C# Unity熱更新方案 (一)
    1.前言 什麼是熱更新 游戲或者軟體更新時，無需重新下載客戶端進行安裝，而是在應用程式啟動的情況下，在內部進行資源或者代碼更新 Unity目前常用熱更新解決方案 HybridCLR，Xlua，ILRuntime等 Unity目前常用資源管理解決方案 AssetBundles，Addressable， ...
  • 在 ASP.NET Core Web API 中使用操作篩選器統一處理通用操作
    本文章主要是在C# ASP.NET Core Web API框架實現向手機發送驗證碼簡訊功能。這裡我選擇是一個互億無線簡訊驗證碼平臺，其實像阿裡雲，騰訊雲上面也可以。 首先我們先去 互億無線 https://www.ihuyi.com/api/sms.html 去註冊一個賬號 註冊完成賬號後，它會送 ...
  • 第28篇 如何.net中實現高效可靠數據同步api
    通過以下方式可以高效，並保證數據同步的可靠性 1.API設計 使用RESTful設計，確保API端點明確，並使用適當的HTTP方法（如POST用於創建，PUT用於更新）。 設計清晰的請求和響應模型，以確保客戶端能夠理解預期格式。 2.數據驗證 在伺服器端進行嚴格的數據驗證，確保接收到的數據符合預期格 ...
所有分類