一、 文件锁用于多个用户共同使用或操作同一个文件。有读锁的时候可以再加读锁，不能再加写锁。有写锁的时候，不能加任何锁，加读锁时，该描述符必须是读打开，加写锁时，该描述符必须是写打开， 且只在进程间有用。
使用flock（锁定文件或解除锁定）函数
头文件 #include<sys/file.h> 定义函数 int flock(int fd,int operation); 函数说明 flock()会依参数operation所指定的方式对参数fd所指的文件做各种锁定或解除锁定的动作。此函数只能锁定整个文件，无法锁定文件的某一区域。 参数 operation有下列四种情况: LOCK_SH 建立共享锁定。多个进程可同时对同一个文件作共享锁定。 LOCK_EX 建立互斥锁定。一个文件同时只有一个互斥锁定。 LOCK_UN 解除文件锁定状态。 LOCK_NB 无法建立锁定时，此操作可不被阻断，马上返回进程。通常与LOCK_SH或LOCK_EX 做OR(|)组合。 单一文件无法同时建立共享锁定和互斥锁定，而当使用dup()或fork()时文件描述词不会继承此种锁定。 返回值 返回0表示成功，若有错误则返回-1，错误代码存于errno。
上锁方法：
#include int lock_set(int fd, int type) { struct flock lock; lock.l_type = type; lock.l_start = 0; lock.l_whence = SEEK_SET; lock.l_len = 0; lock.l_pid = -1; fcntl(fd, F_GETLK, &lock); if(lock.l_type != F_UNLCK) { if(lock.l_type == F_RDLCK) { printf("Read lock already set by %d\n", lock.l_pid); } else if(lock.l_type == F_WRLCK) { printf("Write lock already set by %d\n", lock.l_pid); } } lock.l_type = type; if(fcntl(fd, F_SETLKW, &lock) < 0) { printf("lock failed :type = %d\n", lock.l_type); } switch(lock.l_type) { case F_RDLCK: printf("read lock set by %d\n", getpid()); break; case F_WRLCK: printf("write lock set by %d\n", getpid()); break; case F_UNLCK: printf("UN lock set by %d\n", getpid()); break; default: break; } }
上写锁：
#include #include #include #include #include "set_lock.h" int main() { int fd; fd = open("readme", O_RDWR | O_CREAT, 0666); if(fd < 0) { printf("Open file error\n"); exit(1); } lock_set(fd, F_WRLCK); getchar(); lock_set(fd, F_UNLCK); getchar(); return 0; } 上读锁： #include #include #include #include #include "set_lock.h" int main() { int fd; fd = open("readme", O_RDWR | O_CREAT, 0666); if(fd < 0) { printf("Open file error\n"); exit(1); } lock_set(fd, F_RDLCK); getchar(); lock_set(fd, F_UNLCK); getchar(); return 0; } 在两个终端上进行测试。 二、多路复用： select 、 poll select: #include #include #include #include #define MAX(a, b) ((a) > (b) ? (a) : (b)) int main() { int fd[3]; char buf[1024]; int res; int max_fd; int i; int num; fd_set insert; fd_set temp_insert; struct timeval tv; fd[0] = 0; if(fd[1] = open("in1", O_RDONLY | O_NONBLOCK) < 0) { printf("open in1 error!\n"); return 1; } if(fd[2] = open("in2", O_RDONLY | O_NONBLOCK) < 0) { printf("open in2 error!\n"); return 1; } max_fd = MAX(MAX(fd[0], fd[1]), fd[2]); FD_ZERO(&insert); for(i = 0; i < 3; i++) { FD_SET(fd[i], &insert); } tv.tv_sec = 60; tv.tv_usec = 0; while(FD_ISSET(fd[0], &insert) || FD_ISSET(fd[1], &insert) || FD_ISSET(fd[2], &insert)) { temp_insert = insert; res = select(max_fd + 1, &temp_insert, NULL, NULL, &tv); switch(res) { case -1: printf("select error\n"); return 1; break; case 0: printf("time out\n"); return 1; break; default: for(i = 0; i < 3; i++) { if(FD_ISSET(fd[i], &temp_insert)) { memset(buf, 0, 1024); num = read(fd[i], buf, 1024); if(num < 0) { return 1; } else if(num == 0) { close(fd[i]); FD_CLR(fd[i], &insert); } else { if(i == 0) { if(buf[0] == 'q' || buf[0] == 'Q') { return 0; } } write(STDOUT_FILENO, buf, num); } } } } } } poll: #include #include #include #include #include #define BUF_SZ 1024 #define IO_IN_FILE 3 #define TIMEDELAY 60000 int main() { struct pollfd fd[IO_IN_FILE]; char buf[BUF_SZ]; int res; int real_read; int i; fd[0].fd = 0; if(fd[1].fd = open("in1", O_RDONLY | O_NONBLOCK) < 0) { printf("open in1 error!\n"); return 1; } if(fd[2].fd = open("in2", O_RDONLY | O_NONBLOCK) < 0) { printf("open in2 error!\n"); return 1; } for(i = 0; i < IO_IN_FILE; i++) { fd[i].events = POLLIN; } while(fd[0].events || fd[1].events || fd[2].events) { res = poll(fd, IO_IN_FILE, TIMEDELAY); switch(res) { case -1: printf("poll error\n"); return 1; break; case 0: printf("time out\n"); return 1; break; default: for(i = 0; i < IO_IN_FILE; i++) { if(fd[i].events) { memset(buf, 0, BUF_SZ); real_read = read(fd[i].fd, buf, BUF_SZ); if(real_read < 0) { printf("read error\n"); return 1; } else if(real_read == 0) { close(fd[i].fd); fd[i].events = 0; } else { if(i == 0) { if(buf[0] == 'q' || buf[0] == 'Q') { return 0; } } write(STDOUT_FILENO, buf, real_read); } } } } } return 0; }