theArcticOcean

博观而约取,厚积而薄发

【github myLocker】select监听多个fd,互斥锁保护终端交互

select监听多个fd

select用于多个fd的管理,检测哪个fd可读、可写,如果可读则接收消息(读),如果可写则输出消息(写)。
相关的函数:

int select(int nfds, fd_set *readfds, fd_set *writefds,
          fd_set *exceptfds, struct timeval *timeout);

void FD_CLR(int fd, fd_set *set);        /* 将某个fd从集合set中删除 */
int  FD_ISSET(int fd, fd_set *set);        /* 检测fd是否还在集合set中 */
void FD_SET(int fd, fd_set *set);        /* 将fd加入集合set中 */
void FD_ZERO(fd_set *set);                /* 将集合set清空 */

其中,select函数的选项timeout很有用,当其是NULL时,函数无限期阻塞;当其是{0,0}时,立即返回结果;当其是某一个大于0的时间值时,表示一定时间后return。
关于timeout:

struct timeval {
   long    tv_sec;         /* seconds */
   long    tv_usec;        /* microseconds */
};

应用:

    fd_set readfds;
    struct timeval tv = {0,0};

    while(1){
        memset(buff,0,BUFF_LEN);
        FD_ZERO(&readfds);
        FD_SET(main_send_fd,&readfds);
        FD_SET(main_lock_fd,&readfds);

        if(select(FD_SETSIZE,&readfds,NULL,NULL,&tv) <= 0){
            continue;
        }

        if(FD_ISSET(main_send_fd,&readfds)){    
            ret = recvfrom(main_send_fd,buff,BUFF_LEN,0,(struct sockaddr *)&client_addr,&client_addr_len);
            if(ret > 0){
                //...
            }
        }
        else if(FD_ISSET(main_lock_fd,&readfds)){
            ret = recvfrom(main_lock_fd,buff,BUFF_LEN,0,(struct sockaddr *)&locker_addr,&locker_addr_len);
            if(ret > 0){
                //...
            }
        }
    }

互斥锁保护终端交互

程序运行,未使用读写锁的情况下,输出信息是杂乱的,提示信息的输出和用户的输入被打断分开了。
如下:

[textRWThread] ret: 13, recv buff: write to .key
[lockerThread] ret: 7, recv buff: encrypt
[lockerThread] start encrypt text file...
[lockerThread] please enter filename to enrypt which file is put in folder:
[main] main recv "write ok" from pthTextRW.
[textCodeThread] pthTextCode -> main "prime numbers created" ok.
txt
[lockerThread] the file need to encrypt is folder/txt
[lockerThread] encrypt work finished.
[main] main recv "encrypt ok" from pthLocker.
[main] encrypt and prime number write work all finished!
[main] retval from textCode thread: 1
[main] retval from textRW thread: 1
[main] retval from Locker thread: 1

注意4-7行,当lockerThread线程输出please enter filename to enrypt which file is put in folder.要求用户输入文件名时,另外两个线程main, textCodeThread在屏幕上输出了信息,然后用户的输入文件名明显和LockerThread的提示信息分割开来。于是我们有这样的需求:怎样让提示信息的输出和文件名的输入成为一个原子操作(不可分割)。从本质上讲,线程都是在向屏幕写东西,这是典型的多人使用打印机问题。读写锁适用于读的次数多于写的次数,互斥锁则适用于多人使用同一资源的问题。
互斥锁的相关函数:

int pthread_mutex_destroy(pthread_mutex_t *mutex);
int pthread_mutex_init(pthread_mutex_t *restrict mutex, const pthread_mutexattr_t *restrict attr);
int pthread_mutex_lock(pthread_mutex_t *mutex);
int pthread_mutex_trylock(pthread_mutex_t *mutex);
int pthread_mutex_unlock(pthread_mutex_t *mutex);

互斥锁在使用完毕后,必须要对互斥锁进行销毁,以释放资源。即调用pthread_mutex_destroy()函数。
因为工程中的情况复杂,在屏幕上输入和输出,我们不能直接定义一个函数在函数里面上锁、解锁,再用线程调用此函数来实现不同线程独享stdout, stderr的资源,但是可以用全局变量mutex来实现上锁和阻塞,当一个线程已经对它上锁了,另一个线程试图再次上锁就会block。
code:
public.h:

pthread_mutex_t mutex_output;

main.c:

pthread_mutex_init(&mutex_output,NULL);  //初始化
/* 给输出动作上锁,然后解锁 */
pthread_mutex_lock(&mutex_output);
pthPrint();
printf("recv msg \"prime numbers created!\"\n");
pthread_mutex_unlock(&mutex_output);
...
pthread_mutex_destroy(&mutex_output);   /* 销毁互斥锁 */

pthTextCode.c输出的上锁情况和main类似。
pthLocker.c:

/* 给一个连续的输出和输入动作上锁,然后解锁 */    
pthread_mutex_lock(&mutex_output);
pthPrint();
printf("please enter filename to enrypt which file is put in folder: \n");
scanf("%s",filename);
pthread_mutex_unlock(&mutex_output);

之后的效果:

工程地址:https://github.com/theArcticOcean/myLocker

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/theArcticOcean/article/details/77608543
文章标签: github select 终端
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭