天天看點

什麼是狀态機?用C語言實作程序5狀态模型

前言

狀态機在實際工作開發中應用非常廣泛,在剛進入公司的時候,根據公司産品做流程圖的時候,發現自己經常會漏了這樣或那樣的狀态,導緻整體流程會有問題,後來知道了狀态機這樣的東西,發現用這幅圖就可以很清晰的表達整個狀态的流轉。

一口君曾經做過很多網絡協定子產品,很多協定的開發都必須用到狀态機;一個健壯的狀态機可以讓你的程式,不論發生何種突發事件都不會突然進入一個不可預知的程式分支。

本篇通過C語言實作一個簡單的程序5狀态模型的狀态機,讓大家熟悉一下狀态機的魅力。

什麼是狀态機?

定義

狀态機是有限狀态自動機的簡稱,是現實事物運作規則抽象而成的一個數學模型。

先來解釋什麼是“狀态”( State )。現實事物是有不同狀态的,例如一個LED等,就有 亮 和 滅兩種狀态。我們通常所說的狀态機是有限狀态機,也就是被描述的事物的狀态的數量是有限個,例如LED燈的狀态就是兩個 亮和 滅。

狀态機,也就是 State Machine ,不是指一台實際機器,而是指一個數學模型。說白了,一般就是指一張狀态轉換圖。

舉例

以實體課學的燈泡圖為例,就是一個最基本的小型狀态機

可以畫出以下的狀态機圖

這裡就是兩個狀态:①燈泡亮,②燈泡滅

如果打開開關,那麼狀态就會切換為 燈泡亮 。燈泡亮 狀态下如果關閉開關,狀态就會切換為 燈泡滅。

狀态機的全稱是有限狀态自動機,自動兩個字也是包含重要含義的。給定一個狀态機,同時給定它的目前狀态以及輸入,那麼輸出狀态時可以明确的運算出來的。例如對于燈泡,給定初始狀态燈泡滅 ,給定輸入“打開開關”,那麼下一個狀态時可以運算出來的。

四大概念

下面來給出狀态機的四大概念。

  1. State ,狀态。一個狀态機至少要包含兩個狀态。例如上面燈泡的例子,有 燈泡亮和 燈泡滅兩個狀态。
  2. Event ,事件。事件就是執行某個操作的觸發條件或者密碼。對于燈泡,“打開開關”就是一個事件。
  3. Action ,動作。事件發生以後要執行動作。例如事件是“打開開關”,動作是“開燈”。程式設計的時候,一個 Action 一般就對應一個函數。
  4. Transition ,變換。也就是從一個狀态變化為另一個狀态。例如“開燈過程”就是一個變換。

狀态機的應用

狀态機是一個對真實世界的抽象,而且是邏輯嚴謹的數學抽象,是以明顯非常适合用在數字領域。可以應用到各個層面上,例如硬體設計,編譯器設計,以及程式設計實作各種具體業務邏輯的時候。

程序5狀态模型

程序管理是Linux五大子系統之一,非常重要,實際實作起來非常複雜,我們來看下程序是如何切換狀态的。

下圖是程序的5狀态模型:

關于該圖簡單介紹如下:

  1. 可運作态:當程序正在被CPU執行,或已經準備就緒随時可由排程程式執行,則稱該程序為處于運作狀态(running)。程序可以在核心态運作,也可以在使用者态運作。當系統資源已經可用時,程序就被喚醒而進入準備運作狀态,該狀态稱為就緒态。
  2. 淺度睡眠态(可中斷):程序正在睡眠(被阻塞),等待資源到來是喚醒,也可以通過其他程序信号或時鐘中斷喚醒,進入運作隊列。
  3. 深度睡眠态(不可中斷):其和淺度睡眠基本類似,但有一點就是不可由其他程序信号或時鐘中斷喚醒。隻有被使用wake_up()函數明确喚醒時才能轉換到可運作的就緒狀态。
  4. 暫停狀态:當程序收到信号SIGSTOP、SIGTSTP、SIGTTIN或SIGTTOU時就會進入暫停狀态。可向其發送SIGCONT信号讓程序轉換到可運作狀态。
  5. 僵死狀态:當程序已停止運作,但其父程序還沒有詢問其狀态時,未釋放PCB,則稱該程序處于僵死狀态。

程序的狀态就是按照這個狀态圖進行切換的。

該狀态流程有點複雜,因為我們目标隻是實作一個簡單的狀态機,是以我們簡化一下該狀态機如下:

要想實作狀态機,首先将該狀态機轉換成下面的狀态遷移表。

什麼是狀态機?用C語言實作程式5狀态模型

簡要說明如下:

假設目前程序處于running狀态下,那麼隻有schedule事件發生之後,該程序才會産生狀态的遷移,遷移到owencpu狀态下,如果在此狀态下發生了其他的事件,比如wake、wait_event都不會導緻狀态的遷移。

如上圖所示:

  1. 每一清單示一個狀态,每一行對應一個事件。
  2. 該表是實作狀态機的最核心的一個圖,請讀者詳細對比該表和狀态遷移圖的的關系。
  3. 實際場景中,程序的切換會遠比這個圖複雜,好在衆多大神都幫我們解決了這些複雜的問題,我們隻需要站在巨人的肩膀上就可以了。

實作

根據狀态遷移表,定義該狀态機的狀态如下:

typedef enum {
  sta_origin=0,
  sta_running,
  sta_owencpu,
  sta_sleep_int,
  sta_sleep_unint
}State;
           

發生的事件如下:

typedef enum{
  evt_fork=0,
  evt_sched,
  evt_wait,
  evt_wait_unint,
  evt_wake_up,
  evt_wake, 
}EventID;
           

不論是狀态還是事件都可以根據實際情況增加調整。

定義一個結構體用來表示目前狀态轉換資訊:

typedef struct {
  State curState;//目前狀态
  EventID eventId;//事件ID
  State nextState;//下個狀态
  CallBack action;//回調函數,事件發生後,調用對應的回調函數
}StateTransform ; 
           

事件回調函數:

實際應用中不同的事件發生需要執行不同的action,就需要定義不同的函數,

為友善起見,本例所有的事件都統一使用同一個回調函數。

功能:

列印事件發生後程序的前後狀态,如果狀态發生了變化,就調用對應的回調函數。

void action_callback(void *arg)
{
	StateTransform *statTran = (StateTransform *)arg;
	
	if(statename[statTran->curState] == statename[statTran->nextState])
	{
		printf("invalid event,state not change\n");
	}else{
		printf("call back state from %s --> %s\n",
			statename[statTran->curState],
			statename[statTran->nextState]);
	}
}
           

為各個狀态定義遷移表數組:

/*origin*/
StateTransform stateTran_0[]={
	{sta_origin,evt_fork,        sta_running,action_callback},
	{sta_origin,evt_sched,       sta_origin,NULL},
	{sta_origin,evt_wait,        sta_origin,NULL},
	{sta_origin,evt_wait_unint,  sta_origin,NULL},
	{sta_origin,evt_wake_up,     sta_origin,NULL},
	{sta_origin,evt_wake,        sta_origin,NULL},
}; 

/*running*/
StateTransform stateTran_1[]={
	{sta_running,evt_fork,        sta_running,NULL},
	{sta_running,evt_sched,       sta_owencpu,action_callback},
	{sta_running,evt_wait,        sta_running,NULL},
	{sta_running,evt_wait_unint,  sta_running,NULL},
	{sta_running,evt_wake_up,     sta_running,NULL},
	{sta_running,evt_wake,        sta_running,NULL},
}; 
/*owencpu*/
StateTransform stateTran_2[]={
	{sta_owencpu,evt_fork,        sta_owencpu,NULL},
	{sta_owencpu,evt_sched,       sta_owencpu,NULL},
	{sta_owencpu,evt_wait,        sta_sleep_int,action_callback},
	{sta_owencpu,evt_wait_unint,  sta_sleep_unint,action_callback},
	{sta_owencpu,evt_wake_up,     sta_owencpu,NULL},
	{sta_owencpu,evt_wake,        sta_owencpu,NULL},
}; 

/*sleep_int*/
StateTransform stateTran_3[]={
	{sta_sleep_int,evt_fork,        sta_sleep_int,NULL},
	{sta_sleep_int,evt_sched,       sta_sleep_int,NULL},
	{sta_sleep_int,evt_wait,        sta_sleep_int,NULL},
	{sta_sleep_int,evt_wait_unint,  sta_sleep_int,NULL},
	{sta_sleep_int,evt_wake_up,     sta_sleep_int,NULL},
	{sta_sleep_int,evt_wake,        sta_running,action_callback},
}; 
/*sleep_unint*/
StateTransform stateTran_4[]={
	{sta_sleep_unint,evt_fork,        sta_sleep_unint,NULL},
	{sta_sleep_unint,evt_sched,       sta_sleep_unint,NULL},
	{sta_sleep_unint,evt_wait,        sta_sleep_unint,NULL},
	{sta_sleep_unint,evt_wait_unint,  sta_sleep_unint,NULL},
	{sta_sleep_unint,evt_wake_up,     sta_running,action_callback},
	{sta_sleep_unint,evt_wake,        sta_sleep_unint,NULL},
}; 
           

實作event發生函數:

void event_happen(unsigned int event)
功能:
	根據發生的event以及目前的程序state,找到對應的StateTransform 結構體,并調用do_action()
           
void do_action(StateTransform *statTran)
功能:
	根據結構體變量StateTransform,實作狀态遷移,并調用對應的回調函數。
           
#define STATETRANS(n)  (stateTran_##n)
/*change state & call callback()*/
void do_action(StateTransform *statTran)
{
	if(NULL == statTran)
	{
		perror("statTran is NULL\n");
		return;
	}
	//狀态遷移
	globalState = statTran->nextState;
	if(statTran->action != NULL)
	{//調用回調函數
		statTran->action((void*)statTran);
	}else{
		printf("invalid event,state not change\n");
	}
}
void event_happen(unsigned int event)
{
	switch(globalState)
	{
		case sta_origin:
			do_action(&STATETRANS(0)[event]);
			break;
		case sta_running:
			do_action(&STATETRANS(1)[event]);
			break;
		case sta_owencpu:
			do_action(&STATETRANS(2)[event]);	
			break;
		case sta_sleep_int:
			do_action(&STATETRANS(3)[event]);	
			break;
		case sta_sleep_unint:
			do_action(&STATETRANS(4)[event]);	
			break;
		default:
			printf("state is invalid\n");
			break;
	}
}
           

測試程式:

  1. 初始化狀态機的初始狀态為sta_origin;
  2. 建立子線程,每隔一秒鐘顯示目前程序狀态;
  3. 事件發生順序為:evt_fork-->evt_sched-->evt_sched-->evt_wait-->evt_wake。

讀者可以跟自己的需要,修改事件發生順序,觀察狀态的變化。

main.c

/*顯示目前狀态*/
void *show_stat(void *arg)
{
	int len;
	char buf[64]={0};
	
	while(1)
	{
		sleep(1);
		printf("cur stat:%s\n",statename[globalState]);
	}	
}
void main(void)
{
	init_machine();
	//建立子線程,子線程主要用于顯示目前狀态
	pthread_create(&pid, NULL,show_stat, NULL);

	sleep(5);
	event_happen(evt_fork);

	sleep(5);
	event_happen(evt_sched);
	sleep(5);
	event_happen(evt_sched);
	sleep(5);
	event_happen(evt_wait);
	sleep(5);
	event_happen(evt_wake);
	
}

           

運作結果:

由結果可知:

evt_fork-->evt_sched-->evt_sched-->evt_wait-->evt_wake
           

該事件發生序列對應的狀态遷移順序為:

origen-->running-->owencpu-->owencpu-->sleep_int-->running
           

完整代碼請關注公衆号:一口Linux,回複statmachine

歡迎關注公衆号:一口Linux

繼續閱讀