【原創】（四）Linux進程調度-組調度及帶寬控制

背景

• Read the fucking source code!  --By 魯迅
• A picture is worth a thousand words. --By 高爾基

說明：

1. Kernel版本：4.14
2. ARM64處理器，Contex-A53，雙核
3. 使用工具：Source Insight 3.5， Visio

1. 概述

組調度（task_group）是使用Linux cgroup(control group)的cpu子系統來實現的，能夠將進程進行分組，按組來分配CPU資源等。
好比，看一個實際的例子：
A和B兩個用戶使用同一臺機器，A用戶16個進程，B用戶2個進程，若是按照進程的個數來分配CPU資源，顯然A用戶會佔據大量的CPU時間，這對於B用戶是不公平的。組調度就能夠解決這個問題，分別將A、B用戶進程劃分紅組，並將兩組的權重設置成佔比50%便可。

帶寬（bandwidth）控制，是用於控制用戶組（task_group）的CPU帶寬，經過設置每一個用戶組的限額值，能夠調整CPU的調度分配。在給定週期內，當用戶組消耗CPU的時間超過了限額值，該用戶組內的任務將會受到限制。

因爲組調度和帶寬控制緊密聯繫，所以本文將探討這兩個主題，本文的討論都基於CFS調度器，開始吧。

2. task_group

• 組調度，在內核中是經過struct task_group來組織的，task_group自己支持cfs組調度和rt組調度，本文主要分析cfs組調度。
• CFS調度器管理的是sched_entity調度實體，task_struct（表明進程）和task_group（表明進程組）中分別包含sched_entity，進而來參與調度；

關於組調度的相關數據結構，組織以下：

• 內核維護了一個全局鏈表task_groups，建立的task_group會添加到這個鏈表中；
• 內核定義了root_task_group全局結構，充當task_group的根節點，以它爲根構建樹狀結構；
• struct task_group的子節點，會加入到父節點的siblings鏈表中；
• 每一個struct task_group會分配運行隊列數組和調度實體數組（以CFS爲例，RT調度相似），其中數組的個數爲系統CPU的個數，也就是爲每一個CPU都分配了運行隊列和調度實體；

對應到實際的運行中，以下：

• struct cfs_rq包含了紅黑樹結構，sched_entity調度實體參與調度時，都會掛入到紅黑樹中，task_struct和task_group都屬於被調度對象；
• task_group會爲每一個CPU再維護一個cfs_rq，這個cfs_rq用於組織掛在這個任務組上的任務以及子任務組，參考圖中的Group A；
• 調度器在調度的時候，好比調用pick_next_task_fair時，會從遍歷隊列，選擇sched_entity，若是發現sched_entity對應的是task_group，則會繼續往下選擇；
• 因爲sched_entity結構中存在parent指針，指向它的父結構，所以，系統的運行也能從下而上的進行遍歷操做，一般使用函數walk_tg_tree_from進行遍歷；

2.2 task_group權重

• 進程或進程組都有權重的概念，調度器會根據權重來分配CPU的時間。
• 進程組的權重設置，能夠經過/sys文件系統進行設置，好比操做/sys/fs/cgoup/cpu/A/shares；

調用流程以下圖：

• sched_group_set_shares來完成最終的設置；
• task_group爲每一個CPU都分配了一個sched_entity，針對當前sched_entity設置更新完後，往上對sched_entity->parent設置更新，直到根節點；
• shares的值計算與load相關，所以也須要調用update_load_avg進行更新計算；

看一下實際的效果圖吧：

• 寫節點操做能夠經過echo XXX > /sys/fs/cgroup/cpu/A/B/cpu.shares；
• 橙色的線表明傳入參數指向的對象；
• 紫色的線表明每次更新涉及到的對象，包括三個部分；
• 處理完sched_entity後，繼續按一樣的流程處理sched_entity->parent；

3. cfs_bandwidth

先看一下/sys/fs/cgroup/cpu下的內容吧：

• 有兩個關鍵的字段：cfs_period_us和cfs_quota_us，這兩個與cfs_bandwidth息息相關；
• period表示週期，quota表示限額，也就是在period期間內，用戶組的CPU限額爲quota值，當超過這個值的時候，用戶組將會被限制運行（throttle），等到下一個週期開始被解除限制（unthrottle）；

來一張圖直觀理解一下：

• 在每一個週期內限制在quota的配額下，超過了就throttle，下一個週期從新開始；

3.1 數據結構

內核中使用struct cfs_bandwidth來描述帶寬，該結構包含在struct task_group中。
此外，struct cfs_rq中也有與帶寬控制相關的字段。
仍是來看一下代碼吧：

struct cfs_bandwidth {
#ifdef CONFIG_CFS_BANDWIDTH
	raw_spinlock_t lock;
	ktime_t period;
	u64 quota, runtime;
	s64 hierarchical_quota;
	u64 runtime_expires;

	int idle, period_active;
	struct hrtimer period_timer, slack_timer;
	struct list_head throttled_cfs_rq;

	/* statistics */
	int nr_periods, nr_throttled;
	u64 throttled_time;
#endif
};

• period：週期值；
• quota：限額值；
• runtime：記錄限額剩餘時間，會使用quota值來週期性賦值；
• hierarchical_quota：層級管理任務組的限額比率；
• runtime_expires：每一個週期的到期時間；
• idle：空閒狀態，不須要運行時分配；
• period_active：週期性計時已經啓動；
• period_timer：高精度週期性定時器，用於從新填充運行時間消耗；
• slack_timer：延遲定時器，在任務出列時，將剩餘的運行時間返回到全局池裏；
• throttled_cfs_rq：限流運行隊列列表；
• nr_periods/nr_throttled/throttled_time：統計值；

struct cfs_rq結構中相關字段以下：

struct cfs_rq {
...
#ifdef CONFIG_CFS_BANDWIDTH
	int runtime_enabled;
	u64 runtime_expires;
	s64 runtime_remaining;

	u64 throttled_clock, throttled_clock_task;
	u64 throttled_clock_task_time;
	int throttled, throttle_count;
	struct list_head throttled_list;
#endif /* CONFIG_CFS_BANDWIDTH */
...
}

• runtime_enabled：週期計時器使能；
• runtime_expires：週期計時器到期時間；
• runtime_remaining：剩餘的運行時間；

3.2 流程分析

3.2.1 初始化流程

先看一下初始化的操做，初始化函數init_cfs_bandwidth自己比較簡單，完成的工做就是將struct cfs_bandwidth結構體進程初始化。

• 註冊兩個高精度定時器：period_timer和slack_timer；
• period_timer定時器，用於在時間到期時從新填充關聯的任務組的限額，並在適當的時候unthrottlecfs運行隊列；
• slack_timer定時器，slack_period週期默認爲5ms，在該定時器函數中也會調用distribute_cfs_runtime從全局運行時間中分配runtime；
• start_cfs_bandwidth和start_cfs_slack_bandwidth分別用於啓動定時器運行，其中能夠看出在dequeue_entity的時候會去利用slack_timer，將運行隊列的剩餘時間返回給tg->cfs_b這個runtime pool；
• unthrottle_cfs_rq函數，會將throttled_list中的對應cfs_rq刪除，而且從下往上遍歷任務組，針對每一個任務組調用tg_unthrottle_up處理，最後也會根據cfs_rq對應的sched_entity從下往上遍歷處理，若是sched_entity不在運行隊列上，那就從新enqueue_entity以便參與調度運行，這個也就完成了解除限制的操做；

do_sched_cfs_period_timer函數與do_sched_cfs_slack_timer()函數都調用了distrbute_cfs_runtime()，該函數用於分發tg->cfs_b的全局運行時間runtime，用於在該task_group中平衡各個CPU上的cfs_rq的運行時間runtime，來一張示意圖：

• 系統中兩個CPU，所以task_group針對每一個cpu都維護了一個cfs_rq，這些cfs_rq來共享該task_group的限額運行時間；
• CPU0上的運行時間，淺黃色模塊表示超額了，那麼在下一個週期的定時器點上會進行彌補處理；

3.2.2 用戶設置流程

用戶能夠經過操做/sys中的節點來進行設置：

• 操做/sys/fs/cgroup/cpu/下的cfs_quota_us/cfs_period_us節點，最終會調用到tg_set_cfs_bandwidth函數；
• tg_set_cfs_bandwidth會從root_task_group根節點開始，遍歷組調度樹，並逐個設置限額比率 ；
• 更新cfs_bandwidth的runtime信息；
• 若是使能了cfs_bandwidth功能，則啓動帶寬定時器；
• 遍歷task_group中的每一個cfs_rq隊列，設置runtime_remaining值，若是cfs_rq隊列限流了，則須要進行解除限流操做；

3.2.3 throttle限流操做

cfs_rq運行隊列被限制，是在throttle_cfs_rq函數中實現的，其中調用關係以下圖：

• 調度實體sched_entity入列時，進行檢測是否運行時間已經達到限額，達到則進行限制處理；
• pick_next_task_fair/put_prev_task_fair在選擇任務調度時，也須要進行檢測判斷；

3.2.4 總結

整體來講，帶寬控制的原理就是經過task_group中的cfs_bandwidth來管理一個全局的時間池，分配給屬於這個任務組的運行隊列，當超過限額的時候則限制隊列的調度。同時，cfs_bandwidth維護兩個定時器，一個用於週期性的填充限額並進行時間分發處理，一個用於將未用完的時間再返回到時間池中，大抵如此。

組調度和帶寬控制就先分析到此，下篇文章將分析CFS調度器了，敬請期待。