C語言實(shí)現(xiàn)四窗口聊天
更新時(shí)間：2021年06月14日 10:09:34 作者：帶著你的名字
這篇文章主要為大家詳細(xì)介紹了C語言實(shí)現(xiàn)四窗口聊天，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
C語言實(shí)現(xiàn)四窗口聊天，供大家參考，具體內(nèi)容如下
為了練習(xí)前段時(shí)間學(xué)習(xí)的共享內(nèi)存、管道、消息隊(duì)列等進(jìn)程同步機(jī)制，做了一個(gè)聊天小項(xiàng)目。
項(xiàng)目描述：
有4個(gè)進(jìn)程，A進(jìn)程和B進(jìn)程負(fù)責(zé)通信，從標(biāo)準(zhǔn)輸入讀到的字符串通過管道發(fā)給對(duì)方，A1和B1進(jìn)程負(fù)責(zé)顯示，其中：
- A進(jìn)程和B進(jìn)程通過管道通信，A進(jìn)程和A1進(jìn)程通過共享內(nèi)存通信，B進(jìn)程和B1進(jìn)程通過消息隊(duì)列通信；
- A進(jìn)程從標(biāo)準(zhǔn)輸入讀到的字符串后，放到管道和共享內(nèi)存里，從管道中讀到的字符串放到共享內(nèi)存里，B進(jìn)程從管道中拿到A放的字符串，A1進(jìn)程到共享內(nèi)存中拿到字符串，打印到屏幕上；
- B進(jìn)程從標(biāo)準(zhǔn)輸入讀到的字符串發(fā)給A進(jìn)程，同時(shí)通過消息隊(duì)列發(fā)給B1進(jìn)程，B1進(jìn)程從消息隊(duì)列中讀出消息，打印到屏幕上；
- 退出時(shí)，在A和B任意一個(gè)進(jìn)程中輸入 Crtl+c 或者 Ctrl+\ ，四個(gè)進(jìn)程會(huì)刪除所有管道、共享內(nèi)存、消息隊(duì)列等資源，然后有序退出。
操作系統(tǒng)：Ubuntu20.4
語言：c
編譯器：gcc
func.h
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <unistd.h>
#include <time.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <sys/select.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <sys/msg.h>
#include <signal.h>
#define ARGS_CHECK(argc, num){if(argc!=num){fprintf(stderr,"args error!\n"); return -1;}}
#define ERROR_CHECK(ret,num,msg){if(ret == num){perror(msg); return -1;}}
a.c
//========= A窗口 ===========
//1.從標(biāo)準(zhǔn)輸入讀取數(shù)據(jù)，通過有名管道發(fā)送給B窗口
//2.接收從B窗口發(fā)送過來的數(shù)據(jù)
//3.通過共享內(nèi)存和信號(hào)量，將從B來的數(shù)據(jù)發(fā)送給A1
//===========================
#include <func.h>
int chatA(int shmid, int semid, char *p);
int sndToA1(int semid, int shmid, char *p, char *msg);//把要打印的消息發(fā)送給A1
void closeAll();//把關(guān)閉消息發(fā)送出去，關(guān)閉共享內(nèi)存和信號(hào)量集
void sigFunc(int signum);//新的2號(hào)和3號(hào)信號(hào)處理函數(shù)，如果在A窗口發(fā)生2號(hào)和3號(hào)信號(hào)就調(diào)用close函數(shù)
//全局變量，后面捕獲到退出信號(hào)回收資源時(shí)使用
int semid;
int shmid;//共享內(nèi)存
char *p;
int fdWrite;
int fdRead;
int main()
{
//1.創(chuàng)建信號(hào)量集，如果有新消息就往共享內(nèi)存中寫，類似生產(chǎn)者
semid = semget(2000, 1, IPC_CREAT|0666);
ERROR_CHECK(semid, -1, "A semget");
shmid = shmget(1000, 4096, IPC_CREAT|0666);//創(chuàng)建一個(gè)共享內(nèi)存
ERROR_CHECK(shmid, -1, "shmget");
p = (char *)shmat(shmid, NULL, 0);
signal(SIGINT, sigFunc);
signal(SIGQUIT, sigFunc);
int ret = chatA(shmid, semid, p);
ERROR_CHECK(ret, -1, "run A");//檢查是否成功打開通信窗口
return 0;
}
int chatA(int shmid, int semid, char *p){
//成功運(yùn)行返回1，否則返回-1
fdRead = open("1.pipe", O_RDONLY);//以只讀模式打開管道1
ERROR_CHECK(fdRead, -1, "open fdRead");//檢查是否成功打開
fdWrite = open("2.pipe", O_WRONLY);//以只寫模式打開管道2
ERROR_CHECK(fdWrite, -1, "open fdWrite");//檢查
setbuf(stdin, NULL);
puts("=========== A ===========");
char buf[512] = {0};
fd_set rdset;//設(shè)置一個(gè)信箱，用來監(jiān)控有沒有讀取到信息
while(1){
struct timeval timeout;//設(shè)置超時(shí)
timeout.tv_sec = 5;
timeout.tv_usec = 15000000;//超過5秒沒有接收到信息就是超時(shí)
FD_ZERO(&rdset);//初始化集合，清空信箱
FD_SET(fdRead, &rdset);//將要監(jiān)聽的管道1注冊(cè)到集合中
FD_SET(STDIN_FILENO, &rdset);//將要監(jiān)聽的標(biāo)準(zhǔn)輸入注冊(cè)到集合中
int tret = select(fdRead + 1,&rdset,NULL,NULL,&timeout);//調(diào)用select進(jìn)行監(jiān)聽
if(tret == 0){
puts("time out!");
}
//select阻塞進(jìn)程，任意一個(gè)FD就緒，解除阻塞
//解除阻塞，檢查是誰就緒
if(FD_ISSET(fdRead, &rdset)){
//如果是管道就緒，讀取管道中的內(nèi)容，發(fā)送給A1
memset(buf, 0, sizeof(buf));//清空buf中的內(nèi)容，用來接收管道中的信息
int ret = read(fdRead, buf, 1024);//將管道中的信息讀取出來
if(ret == 0){
//如果另一端對(duì)管道的寫先關(guān)閉了，退出聊天
sigFunc(2);
break;
}
//獲取從B來的消息的類型
int type = 0;
sscanf(buf, "%*d %d", &type);//讀取消息的類別，1類為正常，2類為關(guān)閉所有窗口
int snd_ret = 0;
switch (type){
case 1:
//如果是1號(hào)信息，通過共享內(nèi)存直接把消息發(fā)送給A1
snd_ret = sndToA1(shmid, semid, p, buf);
ERROR_CHECK(snd_ret, -1, "sndToA1");
break;
case 2:
//=====如果是從B發(fā)過來的2號(hào)信息，關(guān)閉所有窗口=====
//向A1發(fā)送一個(gè)空的2號(hào)信號(hào)，讓A1自己退出，然后自己再退出
sigFunc(2);
exit(0);
}
}
if(FD_ISSET(STDIN_FILENO, &rdset)){
//如果標(biāo)準(zhǔn)輸入準(zhǔn)備就緒，讀取標(biāo)準(zhǔn)輸入?yún)^(qū)的數(shù)據(jù)，標(biāo)記為3號(hào)信號(hào)，發(fā)送給A1和B
time_t localtm;
time(&localtm);//獲取當(dāng)前時(shí)間
localtm += 8*3600;
memset(buf, 0, sizeof(buf));//清空buf
int ret = read(STDIN_FILENO, buf, 1024);//讀取數(shù)據(jù)
if(ret == 0){
//如果在標(biāo)準(zhǔn)輸入中讀到了終止符，退出聊天窗口
puts("I quite.");
break;
}
char sstoA1[1024] = {0};//用來拼接數(shù)據(jù),發(fā)送給A1的數(shù)據(jù)
char sstoB[1024] = {0};//用來拼接數(shù)據(jù)，發(fā)送給B的數(shù)據(jù)
sprintf(sstoA1, "%ld %d %s", localtm, 3, buf); //標(biāo)注為三號(hào)信號(hào)發(fā)送給A1
sprintf(sstoB, "%ld %d %s", localtm, 1, buf); //標(biāo)注為1號(hào)信號(hào)發(fā)送給B
sndToA1(shmid, semid, p, sstoA1);//發(fā)送給A1
write(fdWrite, sstoB, sizeof(sstoB));//通過管道發(fā)送給B
}
}
close(fdRead);
close(fdWrite);
return 1;//程序成功運(yùn)行結(jié)束，返回1
}
int sndToA1(int shmid, int semid, char *p, char *msg){
//使用共享內(nèi)存和信號(hào)量給A1傳遞信息
//信號(hào)量集的操作，如果有新消息就往共享內(nèi)存中寫，類似生產(chǎn)者
struct sembuf V;
V.sem_num = 0;
V.sem_op = +1;
V.sem_flg = SEM_UNDO;
semop(semid, &V, 1);
/* int shmid = shmget(1000, 4096, IPC_CREAT|0666);//創(chuàng)建一個(gè)共享內(nèi)存 */
ERROR_CHECK(shmid, -1, "shmget");
/* char *p = (char *)shmat(shmid, NULL, 0); */
memcpy(p, msg, strlen(msg));//向共享內(nèi)存中寫信息
return 1;
}
void closeAll(){
//根據(jù)共享內(nèi)存和信號(hào)量級(jí)的標(biāo)識(shí)符，關(guān)閉并刪除它們
write(fdWrite, "0 2 0", 5);//通過管道發(fā)送給B
shmdt(p);
shmctl(shmid, IPC_RMID, NULL);
semctl(semid, IPC_RMID, 0);
close(fdWrite);
close(fdRead);
exit(0);
}
void sigFunc(int signum){
printf("Bye Bye.\n");
//捕捉2號(hào)和3號(hào)信號(hào)，發(fā)送關(guān)閉信息給A1，然后調(diào)用closeAll
sndToA1(shmid, semid, p, "0 2 0");//發(fā)送給A1
usleep(500);
closeAll();
}
b.c
//========= B窗口 ===========
//1.從標(biāo)準(zhǔn)輸入讀取數(shù)據(jù)，通過有名管道發(fā)送給A窗口
//2.接收從A窗口發(fā)送過來的數(shù)據(jù)
//3.通過共享內(nèi)存和信號(hào)量，將從A來的數(shù)據(jù)發(fā)送給B1
//===========================
#include <func.h>
//自定義一個(gè)消息結(jié)構(gòu)體，用來和B1傳遞數(shù)據(jù)
typedef struct myMsg{
long mtype;
char mtext[512];
}myMsg_t;
int chatB(char *pipe1, char *pipe2);
int sndToB1(int msqid, char *msg);//把從A來的消息發(fā)送給B1
void closeAll();
void sigFunc(int signum);
//全局變量，后面回收資源時(shí)要用到
int msqid;
int fdWrite;
int fdRead;
int main()
{
msqid = msgget(3000, IPC_CREAT|0666);
ERROR_CHECK(msqid, -1, "B msgget");
//注冊(cè)新的信號(hào)處理函數(shù)
signal(SIGINT, sigFunc);
signal(SIGQUIT, sigFunc);
int ret = chatB("./1.pipe", "./2.pipe");
ERROR_CHECK(ret, -1, "run B");
return 0;
}
int chatB(char *pipe1, char *pipe2){
//通信窗口2，讀管道2中的信息，向管道1寫信息
fdWrite = open(pipe1, O_WRONLY);
ERROR_CHECK(fdWrite, -1, "open pipe1");
fdRead = open(pipe2, O_RDONLY);
ERROR_CHECK(fdRead, -1, "open pipe2");
setbuf(stdin, NULL);
puts("============ B ============");
char buf[512] = {0};
fd_set rdset;//設(shè)置集合，用來監(jiān)聽
while(1){
//利用集合設(shè)置阻塞
struct timeval timeout;
timeout.tv_sec = 5;
timeout.tv_usec = 15000000;
FD_ZERO(&rdset);//初始化集合
FD_SET(fdRead, &rdset);
FD_SET(STDIN_FILENO, &rdset);
int tret = select(fdRead+1,&rdset,NULL,NULL,&timeout);
if(tret == 0){
puts("time out!");
}
//集合中有就緒的，檢查是誰就緒，并進(jìn)行相應(yīng)的操作
if(FD_ISSET(fdRead, &rdset)){
//如果是管道就緒，讀取數(shù)據(jù)并發(fā)送給B1
memset(buf, 0, sizeof(buf));
int ret = read(fdRead, buf, 1024);
if(ret == 0){
sigFunc(2);
break;
}
int type = 0;//用來存儲(chǔ)消息的類型
sscanf(buf, "%*d %d", &type);//從消息中獲取類型信息
//如果是2號(hào)信息，關(guān)閉所有窗口
//向B1發(fā)送關(guān)閉信號(hào)，然后回收消息隊(duì)列，再自己結(jié)束
if(type == 2){
sigFunc(2);
exit(0);
}
//如果是其他有效信息，發(fā)送給B1
int snd_ret = sndToB1(msqid, buf);
ERROR_CHECK(snd_ret, -1, "B sndToB1");
}
if(FD_ISSET(STDIN_FILENO, &rdset)){
//如果是標(biāo)準(zhǔn)輸入?yún)^(qū)就緒，讀取數(shù)據(jù),分別發(fā)給A和B1
time_t localtm;
time(&localtm);//獲取當(dāng)前時(shí)間
localtm += 8*3600;
memset(buf, 0, sizeof(buf));
int ret = read(STDIN_FILENO, buf, 1024);
if(ret == 0){
puts("I quite.");
break;
}
//按照協(xié)議拼接數(shù)據(jù)并發(fā)送出去
char sstoA[1024] = {0};//發(fā)送給A的數(shù)據(jù)
sprintf(sstoA, "%ld %d %s", localtm, 1, buf);
write(fdWrite, sstoA, sizeof(sstoA));
char sstoB1[1024] = {0};//發(fā)送給B1的數(shù)據(jù)標(biāo)注為3號(hào)
sprintf(sstoB1, "%ld %d %s", localtm, 3, buf);
sndToB1(msqid, sstoB1);
}
}
close(fdRead);
close(fdWrite);
return 1;//程序成功運(yùn)行結(jié)束，返回1
}
int sndToB1(int msqid, char *msg){
//通過消息隊(duì)列，把數(shù)據(jù)發(fā)送給B1
myMsg_t msgtoB1;//創(chuàng)建一個(gè)消息結(jié)構(gòu)體
msgtoB1.mtype = 1;
memset(msgtoB1.mtext, 0, sizeof(msgtoB1.mtext));
memcpy(msgtoB1.mtext, msg, strlen(msg));
msgsnd(msqid, &msgtoB1, strlen(msg), 0);
return 1;
}
void closeAll(){
msgctl(msqid, IPC_RMID, 0);//刪除消息隊(duì)列
close(fdWrite);//關(guān)閉管道
close(fdRead);
exit(0);
}
void sigFunc(int signum){
printf("Bye Bye.\n");
//通過消息隊(duì)列，把關(guān)閉信息發(fā)送給B1，然后刪除消息隊(duì)列，然后自己退出
sndToB1(msqid, "0 2 0");//發(fā)送給B1關(guān)閉信號(hào)
write(fdWrite, "0 2 0", 5);//發(fā)送給A關(guān)閉信號(hào)
usleep(500);//睡一下，等B1先關(guān)閉
//捕獲2號(hào)和3號(hào)信號(hào)，調(diào)用closeAll函數(shù)
closeAll();
}
a1.c
//========== A1 ==========
//1.從共享內(nèi)存中讀取消息
//2.打印
int display();
#include <func.h>
int main()
{
int ret = display();
ERROR_CHECK(ret, -1, "A1 display");
return 0;
}
int display(){
//1.從共享內(nèi)存中讀取數(shù)據(jù)
//沒有消息就等待，有消息就讀取，使用信號(hào)量集，類似消費(fèi)者
//1.1 創(chuàng)建一個(gè)信號(hào)量集，如果共享內(nèi)存中有數(shù)據(jù)就讀取，如果共享內(nèi)存中沒有數(shù)據(jù)就阻塞
int semid = semget(2000, 1, IPC_CREAT|0666);
ERROR_CHECK(semid, -1, "A1 semget");
semctl(semid, 0, SETVAL, 0);//信號(hào)量初始值設(shè)為0
//設(shè)置信號(hào)量，測(cè)試并取資源，類似消費(fèi)者操作
struct sembuf P;
P.sem_num = 0;
P.sem_op = -1;
P.sem_flg = SEM_UNDO;
printf("=========== A1 ===========\n");
while(1){
semop(semid, &P, 1);//P操作，測(cè)試并取資源
int shmid = shmget(1000, 4096, IPC_CREAT|0666);
ERROR_CHECK(shmid, -1, "A1 shmget");
char *p = (char *)shmat(shmid, NULL, 0);//連接共享內(nèi)存
int type = 0;
sscanf(p, "%*d %d", &type);//獲取消息的屬性，然后根據(jù)協(xié)議執(zhí)行相應(yīng)的操作
switch (type){
case 1:
//從B來的消息
printf("<<<<<<<<< receive <<<<<<<<<<\n");
struct tm *ptm = NULL;
time_t tmp = 0;
char ss[512] = {0};
sscanf(p, "%ld", &tmp);//讀取消息中的時(shí)間信息
sscanf(p, "%*d %*d %[^\n]", ss);
ptm = gmtime(&tmp);
printf("%4d-%02d-%02d %02d:%02d:%02d\n",
ptm->tm_year+1900,ptm->tm_mon+1,ptm->tm_mday,
ptm->tm_hour, ptm->tm_min, ptm->tm_sec);
puts(ss);
printf("\n");
//清空共享內(nèi)存中的數(shù)據(jù)
memset(p, 0, 4096);
break;
case 2:
printf("Bye Bye.\n");
shmdt(p);
shmctl(shmid, IPC_RMID, NULL);
exit(0);
break;
case 3:
printf(">>>>>>>>> send >>>>>>>>>>>\n");
struct tm *ptm3 = NULL;
time_t tmp3 = 0;
char ss3[512] = {0};
sscanf(p, "%ld", &tmp3);//讀取消息中的時(shí)間信息
sscanf(p, "%*d %*d %[^\n]", ss3);
ptm3 = gmtime(&tmp3);
printf("%4d-%02d-%02d %02d:%02d:%02d\n",
ptm3->tm_year+1900,ptm3->tm_mon+1,ptm3->tm_mday,
ptm3->tm_hour, ptm3->tm_min, ptm3->tm_sec);
puts(ss3);
printf("\n");
//清空共享內(nèi)存中的數(shù)據(jù)
memset(p, 0, 4096);
break;
default:
printf("sonething wrong!\n");
}
}
}
b1.c
//========== B1 ===========
//接收來自B的消息，并打印
#include <func.h>
typedef struct myMsg{
long mtype;
char mtext[512];
}myMsg_t;
int display();
int main()
{
int ret = display();
ERROR_CHECK(ret, -1, "B1 display");
return 0;
}
int display(){
printf("=========== B1 ===========\n");
while(1){
//接收來自B的消息
int msqid = msgget(3000, IPC_CREAT|0666);
ERROR_CHECK(msqid, -1, "B1 msgget");
myMsg_t msgfromB;
memset(&msgfromB, 0, sizeof(msgfromB));
msgrcv(msqid, &msgfromB, sizeof(msgfromB.mtext), 1, 0);
//1.如果是2類信號(hào)，退出
int type = 0;
sscanf(msgfromB.mtext, "%*d %d", &type);//讀取消息的屬性，根據(jù)不同屬性，執(zhí)行相應(yīng)的操作
switch (type){
case 1:
//從B來的消息
printf("<<<<<<<<< receive <<<<<<<<<<\n");
struct tm *ptm = NULL;
time_t tmp = 0;
char ss[512] = {0};
sscanf(msgfromB.mtext, "%ld", &tmp);//讀取消息中的時(shí)間信息
sscanf(msgfromB.mtext, "%*d %*d %[^\n]", ss);
ptm = gmtime(&tmp);
printf("%4d-%02d-%02d %02d:%02d:%02d\n",
ptm->tm_year+1900,ptm->tm_mon+1,ptm->tm_mday,
ptm->tm_hour, ptm->tm_min, ptm->tm_sec);
puts(ss);
printf("\n");
//清空共享內(nèi)存中的數(shù)據(jù)
break;
case 2:
//刪除消息隊(duì)列并退出
printf("Bye Bye.\n");
msgctl(msqid, IPC_RMID, NULL);
exit(0);
case 3:
printf(">>>>>>>>> send >>>>>>>>>>>\n");
struct tm *ptm3 = NULL;
time_t tmp3 = 0;
char ss3[512] = {0};
sscanf(msgfromB.mtext, "%ld", &tmp3);//讀取消息中的時(shí)間信息
sscanf(msgfromB.mtext, "%*d %*d %[^\n]", ss3);
ptm3 = gmtime(&tmp3);
printf("%4d-%02d-%02d %02d:%02d:%02d\n",
ptm3->tm_year+1900,ptm3->tm_mon+1,ptm3->tm_mday,
ptm3->tm_hour, ptm3->tm_min, ptm3->tm_sec);
puts(ss3);
printf("\n");
break;
default:
printf("Something wrong!\n");
}
}
}
運(yùn)行如下：
以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
C++程序把輸入和輸出看作字節(jié)流,輸入時(shí)程序從輸入流中抽取字節(jié),輸出時(shí)程序?qū)⒆止?jié)插入到輸出流中,下面這篇文章主要給大家介紹了關(guān)于C++中簡單的文本文件輸入/輸出的相關(guān)資料,需要的朋友可以參考下2021-12-12
這篇文章主要為大家詳細(xì)介紹了C++如何利用第三方庫實(shí)現(xiàn)播放音頻的功能,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-01-01
for循環(huán)中刪除map中的元素valgrind檢測(cè)提示error：Invalid read of size 8
這篇文章主要介紹了for循環(huán)中刪除map中的元素valgrind檢測(cè)提示error：Invalid read of size 8 的相關(guān)資料,需要的朋友可以參考下2016-07-07
這篇文章主要介紹了詳解C++編程中的私有繼承和公有繼承,是C++入門學(xué)習(xí)中的基礎(chǔ)知識(shí),需要的朋友可以參考下2015-09-09
這篇文章主要介紹了C語言遞歸實(shí)現(xiàn)字符串逆序的方式詳解,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-10-10
C語言實(shí)現(xiàn)宿舍管理課程設(shè)計(jì)
這篇文章主要為大家詳細(xì)介紹了C語言實(shí)現(xiàn)宿舍管理課程設(shè)計(jì)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-03-03
類中盡量采用指針或引用方式調(diào)用其它類，這樣就可以只聲明class xxx了。并且這也符合資源最優(yōu)利用，更利于使用多態(tài)2013-09-09
C語言中關(guān)于scanf函數(shù)的一些問題詳解
這篇文章主要為大家介紹了C語言中關(guān)于scanf函數(shù)的一些問題，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
C++實(shí)現(xiàn)LeetCode(113.二叉樹路徑之和之二)
這篇文章主要介紹了C++實(shí)現(xiàn)LeetCode(113.二叉樹路徑之和之二),本篇文章通過簡要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-07-07
使用C/C++讀取matlab中.mat格式數(shù)據(jù)的操作
這篇文章給大家介紹了使用C/C++讀取matlab中.mat格式數(shù)據(jù)的操作,文中通過圖文結(jié)合的方式介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下2023-12-12