头部文件代码: mylock.h
struct myflock
{
short l_type; /*文件锁类型: F_RDLOCK 读取锁;F_WRLCK 写入锁;F_UNLCK 解锁 */
off_t l_start; /*相对位移量*/
short l_whence; /*相对位移量的起点SEEK_SET;SEEK_CUR;SEEK_END: */
off_t l_len; /* 加锁区域长度 */
pid_t l_pid; /* */
} ;
/* lock_set */
int lock_set(int fd, int type)
{
struct myflock old_lock, lock;
lock.l_whence = SEEK_SET;
lock.l_start = 0;
lock.l_len = 0;
lock.l_type = type;
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 setby %d\n", lock.l_pid);
}
else if (lock.l_type == F_WRLCK) /* 该文件已有写入锁 */
{
printf("Write lock already setby %d\n", lock.l_pid);
}
}
/* l_type 可能已被F_GETLK 修改过*/
lock.l_type = type;
/* 根据不同的type 值进行阻塞式上锁或解锁*/
if ((fcntl(fd, F_SETLKW,&lock)) < 0)
{
printf("Lock failed:type = %d\n", lock.l_type); return 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("Release lock by %d\n", getpid()); return 1;
break;
default:
break;
} /* end of switch */
return 0;
}
生产者程序的源代码: producer.c
/* producer.c */
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include "mylock.h"
#define MAXLEN 10 /* 缓冲区大小最大值*/
#define ALPHABET 1 /* 表示使用英文字符*/
#define ALPHABET_START 'a' /* 头一个字符,可以用'A'*/
#define COUNT_OF_ALPHABET 26 /* 字母字符的个数*/
#define DIGIT 2 /* 表示使用数字字符 */
#define DIGIT_START '0' /* 头一个数字字符 */
#define COUNT_OF_DIGIT 10 /* 数字字符的个数 */
#define SIGN_TYPE ALPHABET /*本实例用英文字符 */
const char *fifo_file = "./myfifo"; /* !" FIFO 文件名 */
char buff[MAXLEN]; /* 缓冲区 */
/* 函数product() 产生一个字符并写入仿真 FIFO 文件中 */
int roduct(void)
{
int fd;
unsigned int sign_type, sign_start, sign_count, size;
static unsigned int counter = 0;
/* 打开!"FIFO 文件*/
if ((fd = open(fifo_file,O_CREAT|O_RDWR|O_APPEND, 0644)) < 0)
{
printf("Open fifo fileerror\n"); exit(1);
}
sign_type = SIGN_TYPE;
switch(sign_type)
{
case ALPHABET: /* 英文字符 */
{
sign_start = ALPHABET_START;
sign_count = COUNT_OF_ALPHABET;
}
break;
case DIGIT: /* 数字字符 */
{
sign_start = DIGIT_START;
sign_count = COUNT_OF_DIGIT;
}
break;
default:
{
return -1;
}
} /*end of switch*/
sprintf(buff, "%c", (sign_start + counter));counter = (counter + 1) % sign_count;
lock_set(fd, F_WRLCK); /* 上写锁*/
if ((size = write(fd, buff, strlen(buff))) < 0)
{
printf("Producer: writeerror\n"); return -1;
}
lock_set(fd, F_UNLCK); /* 解锁 */
close(fd);
return 0;
}
int main(int argc ,char *argv[])
{
int time_step = 1; /* 生产周期 */
int time_life = 10; /* 需要生产的资源数 */
if (argc > 1)
{/* 第一个参数表示生产周期 */
sscanf(argv[1], "%d",&time_step);
}
if (argc > 2)
{/* 第二个参数表示需要生产的资源数 */
sscanf(argv[2], "%d",&time_life);
}
while (time_life--)
{
if (product() < 0)
{
break;
}
sleep(time_step);
}
exit(EXIT_SUCCESS);
}
消费者程序的源代码: customer.c
/* customer.c */
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include "mylock.h"
#define MAX_FILE_SIZE 100 * 1024 * 1024 /* 100M*/
const char *fifo_file = "./myfifo"; /* 仿真 FIFO 文件名 */
const char *tmp_file = "./tmp"; /* 临时文件名 */
/* 资源消费函数customing */
int customing(const char *myfifo,int need)
{
int fd;
char buff;
int counter = 0;
if ((fd =open(myfifo, O_RDONLY)) < 0)
{
printf("Functioncustoming error\n");
return -1;
}
printf("Enjoy:");
lseek(fd, SEEK_SET, 0);
while (counter < need)
{
while ((read(fd, &buff, 1) == 1) && (counter < need))
{
fputc(buff,stdout); /* -.就是在屏幕上/0的显示*/
counter++;
}
}
fputs("\n", stdout);
close(fd);
return 0;
}
/* myfilecopy()函数: 实现从sour_file 文件的offset 偏移处开始将count 个字节数据复制到dest_file 文件*/
int myfilecopy(const char *sour_file,const char *dest_file,int offset, int count, int copy_mode)
{
int in_file, out_file;
int counter = 0;
char buff_unit;
if ((in_file = open(sour_file, O_RDONLY|O_NONBLOCK)) < 0)
{
printf("Functionmyfilecopy error in source file\n"); return -1;
}
if ((out_file = open(dest_file, O_CREAT|O_RDWR|O_TRUNC|O_NONBLOCK, 0644)) < 0)
{
printf("Functionmyfilecopy error in destination file:"); return -1;
}
lseek(in_file, offset, SEEK_SET);
while ((read(in_file, &buff_unit, 1) == 1) && (counter < count))
{
write(out_file,&buff_unit, 1); counter++;
}
close(in_file);
close(out_file);
return 0;
}
/* custom()函数:实现FIFO 消费者*/
int custom(int need)
{
int fd;
/* 对资源进行消费,need 表示该消费的资源数目 */
customing(fifo_file, need);
if ((fd =open(fifo_file, O_RDWR)) < 0)
{
printf("Functionmyfilecopy error in source_file:");
return -1;
}
/* 为了模拟FIFO 结构,对整个文件进平行行移动 */
lock_set(fd, F_WRLCK);
myfilecopy(fifo_file, tmp_file,need, MAX_FILE_SIZE, 0);
myfilecopy(tmp_file, fifo_file, 0, MAX_FILE_SIZE, 0);
lock_set(fd, F_UNLCK);
unlink(tmp_file);
close(fd);
return 0;
}
int main(int argc ,char *argv[])
{
int customer_capacity = 10;
if (argc > 1) /* 第一个参数指定需要消费的资源数目,默认值为10 */
{
sscanf(argv[1],"%d", &customer_capacity);
}
if (customer_capacity > 0)
{
custom(customer_capacity);
}
exit(EXIT_SUCCESS);
}
(3) 分别编译生产者程序producer.c 和消费者程序customer.c
(4) 确保编译没有错误后,先在控制台终端1上运行生产者程序:./producer 1 20
再在控制台终端2上运行消费者程序:./customer 5
观察两终端的输入输出情况