linux文件读写 文件锁、select、poll【转】
本文转载自:http://blog.csdn.net/fansongy/article/details/6853395
一、文件锁
文件锁用于多个用户共同使用或操作同一个文件。有读锁的时候可以再加读锁,不能再加写锁。有写锁的时候,不能加任何锁,加读锁时,该描述符必须是读打开,加写锁时,该描述符必须是写打开,且只在进程间有用。
重写的上锁方法
[cpp] view plain copy print?
//lock.c
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;
//此处的F_SETLKW为F_SETLK的阻塞版本,当无法获取锁时进入睡眠等待状态
if(fcntl(fd,F_SETLKW,&lock)<0){
printf("Lock failed:type=%d!\n",lock.l_type);
exit(1);
}
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;
}
}
上写锁
[cpp] view plain copy print?
#include
#include
#include
#include"lock.c"
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;
}
上读锁
[cpp] view plain copy print?
#include
#include
#include
#include"lock.c"
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
[cpp] view plain copy print?
#include
#include
#include
#include
#define MAX(a,b) (a>b?a:b)
int main(){
int fd[3];
char buff[1024];
int res,max_fd,i,num;
fd_set insert,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;
}
//选出最大的fd,select()函数用的
max_fd=MAX(MAX(fd[0],fd[1]),fd[2]);
//清空fd_set
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;
//select函数会对fd_set产生修改,只保存变化的文件符,所以要用一个temp
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(buff,0,1024);
num=read(fd[i],buff,1024);
if(num<0){
return 1;
}else if(num == 0){
close(fd[i]);
FD_CLR(fd[i],&insert);
}else{
if(i == 0){
if((buff[0] == 'q') || (buff[0] == 'Q')){
return 0;
}
}
write(STDOUT_FILENO,buff,num);
}
}
}
}
}
}
poll用法与select很相似,只是在一些变量上有些不同:
[cpp] view plain copy print?
#include
#include
#include
#include
#include
#include
#include
#define MAX_BUFFER_SIZE 1024
#define IO_IN_FILES 3
#define TIME_DELAY 60000
int main() {
struct pollfd fds[IO_IN_FILES];
char buff[MAX_BUFFER_SIZE];
int i,res,real_read;
fds[0].fd=0;
if((fds[1].fd=open("in1",O_RDONLY|O_NONBLOCK))<0) {
printf("Open in1 error!\n");
return 1;
}
if((fds[2].fd=open("in2",O_RDONLY|O_NONBLOCK))<0) {
printf("Open in2 error!\n");
return 1;
}
for(i=0;i