河北科技大學操作系統課程設計——讀者寫者問題

#include <windows.h>
#include <conio.h>
#include <stdlib.h>
#include <fstream>
#include <io.h>
#include <string.h>
#include <stdio.h>
using namespace std;
#define READER 'R'
#define WRITER 'W'
#define MAX_THREAD_NUM 32 //最大線程數目
CRITICAL_SECTION RP_Write; //臨界區
CRITICAL_SECTION pright; //權限 
CRITICAL_SECTION w_Mutex; //臨界區
CRITICAL_SECTION cs;//對輸出格式加鎖控制 
#define INTE_PER_SEC 100  //每秒時鐘中斷數目int flag;
int readcount=0;          //讀者數目
int writecount=0;         //寫者數目
struct ThreadInfo
{	int serial;           //線程序號char entity;          //線程類別double delay;         //線程開始時間double persist;       //線程讀寫持續時間
};
ThreadInfo  thread_info[MAX_THREAD_NUM]; 
//讀者優先處理函數
void ReaderPriority(int num);   
void RP_ReaderThread(void *p);
void RP_WriterThread(void *p);
//寫者優先處理函數
void WriterPriority(int num);  
void WP_ReaderThread(void *p);
void WP_WriterThread(void *p);//顯示函數
void showInput(void);
void showWhich(void); 
int main()
{InitializeCriticalSection(&cs);int num;int choice; while(1){system("cls");showInput();scanf("%d",&choice);switch(choice){case 1:system("cls");showWhich();break;case 2:exit(1); break;default :printf("輸入有誤，請重新輸入！\n");Sleep(1000);break;}}system("pause");      return 0;
}
void showInput(void)
{printf("            *********請選擇功能*********\n");printf("            *                          *\n"); printf("            *    1:模擬讀者寫者問題    *\n");printf("            *                          *\n");  printf("            *    2:退出系統            *\n"); printf("            *                          *\n"); printf("            ****************************\n"); 
} 
void show()
{printf("            *********請選擇功能*********\n");printf("            *                          *\n"); printf("            *        1:讀者優先        *\n");printf("            *        2:寫者優先        *\n");  printf("            *        3:返回上一級      *\n"); printf("            *        4:清屏并繼續      *\n"); printf("            *        5:退出系統        *\n"); printf("            *                          *\n");printf("            ****************************\n");
}
void showWhich(void)
{int num;int choice;printf("請輸入總的讀者寫者線程數:(<=32)\n");scanf("%d",&num);for(int i=0;i<num;i++){printf("請輸入*線程號*線程類別(R/W)*線程開始時間*線程持續時間:\n");scanf("%d %c %lf %lf",&thread_info[i].serial,&thread_info[i].entity,&thread_info[i].delay,&thread_info[i].persist);	}system("cls");show();while(1){ scanf("%d",&choice); switch(choice){case 1:printf("序號  讀者或寫者   開始時間   持續時間\n");for(int i=0;i<num;i++){	printf(" %d        %c        %f   %f  \n",thread_info[i].serial,thread_info[i].entity,thread_info[i].delay,thread_info[i].persist);}ReaderPriority(num);//讀者優先continue;case 2:printf("序號  讀者或寫者   開始時間   持續時間\n");for(int i=0;i<num;i++){	printf(" %d        %c        %f   %f  \n",thread_info[i].serial,thread_info[i].entity,thread_info[i].delay,thread_info[i].persist);}WriterPriority(num);//寫者優先continue;case 3:flag=1;break;case 4:system("cls");show();continue; case 5:flag=0;break; default:printf("輸入有誤,請重新輸入！\n");Sleep(1000);system("cls");show();continue; };if(flag==1) break;if(flag==0) exit(1);}
}
//讀者優先處理函數
void ReaderPriority(int num)
{DWORD n_thread=num;        //線程數目DWORD thread_ID;         //線程IDDWORD wait_for_all;      //等待所有線程結束HANDLE rmutex;	//使讀者互斥訪問readcountrmutex=CreateMutex(NULL,FALSE,"mutex_for_readcount");// HANDLE h_Thread[MAX_THREAD_NUM];readcount=0;     //初始化readcountInitializeCriticalSection(&RP_Write);  //初始化臨界區printf("讀者優先:（縱坐標為執行順序）\n");for (int i=0;i<(int)(n_thread);i++){if (thread_info[i].entity==READER || thread_info[i].entity=='r')//創建讀者線程h_Thread[i]=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)(RP_ReaderThread),&thread_info[i],0,&thread_ID);else//創建寫者線程h_Thread[i]=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)(RP_WriterThread),&thread_info[i],0,&thread_ID);}//等待所有線程結束wait_for_all=WaitForMultipleObjects(n_thread,h_Thread,TRUE,-1);printf("所有操作完成\n\n");
}
//讀者優先-----讀者線程
void RP_ReaderThread(void *p)
{	int m_ID;           //線程序號DWORD m_delay;      //延遲時間DWORD m_persist;    //讀文件持續時間m_ID=((ThreadInfo *)(p))->serial;m_delay=(DWORD)(((ThreadInfo *)(p))->delay*INTE_PER_SEC);m_persist=(DWORD)(((ThreadInfo *)(p))->persist*INTE_PER_SEC);HANDLE rmutex;//使讀者互斥訪問readcountrmutex=OpenMutex(MUTEX_ALL_ACCESS,FALSE,"mutex_for_readcount");DWORD wait_for_mutex;   //等待互斥變量所有權Sleep(m_delay); //延遲等待EnterCriticalSection(&cs);for(int i=1;i<m_ID;i++)printf("           ");printf("讀者%d讀請求\n",m_ID);LeaveCriticalSection(&cs);	wait_for_mutex=WaitForSingleObject(rmutex,-1); //P(rmutex)readcount++;   //讀者數目加1if (readcount==1)EnterCriticalSection(&RP_Write);  //如果這是第一個讀者，進入臨界區，開始執行 P(RP_Write);ReleaseMutex(rmutex);//V（rmutex）EnterCriticalSection(&cs);	for(int j=1;j<m_ID;j++)//讀文件printf("           ");printf("讀者%d開始讀\n",m_ID);LeaveCriticalSection(&cs);Sleep(m_persist);EnterCriticalSection(&cs);for(int k=1;k<m_ID;k++)//退出線程printf("           ");printf("讀者%d完成讀\n",m_ID);LeaveCriticalSection(&cs);wait_for_mutex=WaitForSingleObject(rmutex,-1);  //P（rmutex）readcount--;          //讀者數目減1if (readcount==0)LeaveCriticalSection(&RP_Write); //如果這是最后一個讀者，則其釋放臨界區V(RP_Write);ReleaseMutex(rmutex); //V（rmutex）
}
//讀者優先-----寫者線程
void RP_WriterThread(void *p)
{DWORD m_delay;      //延遲時間DWORD m_persist;    //讀文件持續時間int m_ID;           //線程序號m_ID=((ThreadInfo *)(p))->serial;m_delay=(DWORD)(((ThreadInfo *)(p))->delay*INTE_PER_SEC);m_persist=(DWORD)(((ThreadInfo *)(p))->persist*INTE_PER_SEC);Sleep(m_delay);//延遲等待EnterCriticalSection(&cs);for(int i=1;i<m_ID;i++)printf("           ");printf("寫者%d寫請求\n",m_ID);LeaveCriticalSection(&cs);EnterCriticalSection(&RP_Write);//進入臨界區  P(RP_Write);EnterCriticalSection(&cs);for(int l=1;l<m_ID;l++)	//寫文件printf("           ");printf("寫者%d開始寫\n",m_ID);LeaveCriticalSection(&cs);Sleep(m_persist);EnterCriticalSection(&cs);for(int q=1;q<m_ID;q++)	//退出線程printf("           ");printf("寫者%d完成寫\n",m_ID);LeaveCriticalSection(&cs);LeaveCriticalSection(&RP_Write);  //退出后，離開臨界區 V(RP_Write);
}
//寫者優先處理函數
void WriterPriority(int num){DWORD n_thread=num;     //線程數目DWORD thread_ID;      //線程IDDWORD wait_for_all;   //等待所有線程結束HANDLE r_Mutex;       //使讀者互斥訪問readcountHANDLE writerm;       //寫互斥 r_Mutex=CreateMutex(NULL,FALSE,"r_Mutex");writerm=CreateMutex(NULL,FALSE,"writerm");InitializeCriticalSection(&pright);InitializeCriticalSection(&w_Mutex);HANDLE h_Thread[MAX_THREAD_NUM];readcount=0;     //初始化readcountwritecount=0;     //初始化writecountprintf("寫者優先:（縱坐標為執行順序）\n");for (int i=0;i<(int)(n_thread);i++){if (thread_info[i].entity==READER || thread_info[i].entity=='r')//創建讀者線程h_Thread[i]=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)(WP_ReaderThread),&thread_info[i],0,&thread_ID);else//創建寫者線程h_Thread[i]=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)(WP_WriterThread),&thread_info[i],0,&thread_ID);}//等待所有線程結束wait_for_all=WaitForMultipleObjects(n_thread,h_Thread,TRUE,-1);printf("所有操作完成\n");
}//寫者優先-----讀者線程
void WP_ReaderThread(void *p){	DWORD m_delay;        //延遲時間DWORD m_persist;     //讀文件持續時間int m_ID;           //線程序號m_ID=((ThreadInfo *)(p))->serial;m_delay=(DWORD)(((ThreadInfo *)(p))->delay*INTE_PER_SEC);m_persist=(DWORD)(((ThreadInfo *)(p))->persist*INTE_PER_SEC);HANDLE r_Mutex;       //使讀者互斥訪問readcount    r_Mutex=OpenMutex(MUTEX_ALL_ACCESS,FALSE,"r_Mutex");DWORD wait_for_rmutex;   //等待互斥變量所有權Sleep(m_delay);       //延遲等待EnterCriticalSection(&cs);for(int i=1;i<m_ID;i++)printf("           ");printf("讀者%d讀請求\n",m_ID);LeaveCriticalSection(&cs);EnterCriticalSection(&pright); //拿到權限 wait_for_rmutex=WaitForSingleObject(r_Mutex,INFINITE);  //控制臨界資源readcount if (readcount==0)EnterCriticalSection(&w_Mutex);     //如果這是第一個讀者，寫者優先進入臨界區 P(w_Mutex);readcount++;  //讀者數目加1ReleaseMutex(r_Mutex);//V(r_Mutex)LeaveCriticalSection(&pright);//V(pright)EnterCriticalSection(&cs);for(int s=1;s<m_ID;s++)	//讀文件printf("           ");printf("讀者%d開始讀\n",m_ID);LeaveCriticalSection(&cs);Sleep(m_persist);EnterCriticalSection(&cs);for(int r=1;r<m_ID;r++)//退出線程printf("           ");printf("讀者%d完成讀\n",m_ID);LeaveCriticalSection(&cs);wait_for_rmutex=WaitForSingleObject(r_Mutex,INFINITE);  //P(r_Mutex)readcount--;    //讀者數目減1if (readcount==0)LeaveCriticalSection(&w_Mutex);    //如果讀者都退出，寫者進入臨界區V(w_Mutex);ReleaseMutex(r_Mutex);//V(r_Mutex)
}
//寫者優先-----寫者線程
void WP_WriterThread(void *p){DWORD m_delay;        //延遲時間DWORD m_persist;     //讀文件持續時間int m_ID;           //線程序號m_ID=((ThreadInfo *)(p))->serial;m_delay=(DWORD)(((ThreadInfo *)(p))->delay*INTE_PER_SEC);m_persist=(DWORD)(((ThreadInfo *)(p))->persist*INTE_PER_SEC);HANDLE writerm;     writerm=OpenMutex(MUTEX_ALL_ACCESS,FALSE,"writerm");DWORD wait_for_writerm; //等待互斥變量所有權Sleep(m_delay);     //延遲等待EnterCriticalSection(&cs);for(int i=1;i<m_ID;i++)printf("           ");printf("寫者%d寫請求\n",m_ID);LeaveCriticalSection(&cs);wait_for_writerm=WaitForSingleObject(writerm,INFINITE);//P(writerm)writecount++;   //寫者數目加1if (writecount==1)EnterCriticalSection(&pright);//P(pright)ReleaseMutex(writerm);//V(writerm)EnterCriticalSection(&w_Mutex);//P(w_mutex)EnterCriticalSection(&cs);for(int h=1;h<m_ID;h++)	//寫文件printf("           ");printf("寫者%d開始寫\n",m_ID);LeaveCriticalSection(&cs);Sleep(m_persist);EnterCriticalSection(&cs);for(int t=1;t<m_ID;t++)	//退出線程printf("           ");printf("寫者%d完成寫\n",m_ID);LeaveCriticalSection(&cs);LeaveCriticalSection(&w_Mutex);wait_for_writerm=WaitForSingleObject(writerm,INFINITE);//P(writerm)writecount--;   //寫者數目減1if (writecount==0)LeaveCriticalSection(&pright);ReleaseMutex(writerm);//V(writerm)
}