使用共享內(nèi)存和記錄鎖實例。本例中，父進(jìn)程創(chuàng)建一段共享內(nèi)存，然后向其中追加字符串"Parent"(在寫入“Parent”的時候，采用一個字符一個字符的寫入，目的是為了驗證記錄鎖對父子進(jìn)程同步的正確性)；子進(jìn)程向共享內(nèi)存中追加字符串“Child”（同樣是一個一個的字符寫入）。由于記錄鎖是針對文件的，所以得先創(chuàng)建一個空文件作為記錄鎖的操作對象，作為共享內(nèi)存訪問的輔助工具，如果一個進(jìn)程對這個空文件加寫鎖成功后，就開始訪問共享內(nèi)存，訪問結(jié)束就對文件解鎖。
代碼如下：
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <fcntl.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#define SHM_SIZE 1024 /* shared memory size(bytes) */
#define SHM_MODE 0600 /* user read/write */
#define WRITE_LOCK(fd) regLock((fd), F_SETLKW, F_WRLCK, 0, SEEK_SET, 0)
#define UN_LOCK(fd) regLock((fd), F_SETLK, F_UNLCK, 0, SEEK_SET, 0)
/* 創(chuàng)建一個文件 */
int createFile()
{
int fd;
if ( (fd = open("/tmp/emptyfile4shm", O_RDWR | O_CREAT, 0666)) < 0) {
fprintf(stderr, "Create a empty file failed!\n");
exit(EXIT_FAILURE);
}
return fd;
}
/* 在文件fd上加鎖或解鎖*/
int regLock(int fd, int cmd, int type, off_t offset, int whence, off_t len)
{
struct flock lock;
lock.l_type = type; /* lock type: F_RDLCK, F_WRLCK, F_UNLCK */
lock.l_start = offset; /* byte offset relative to l_whence */
lock.l_whence = whence; /* it's value : SEEK_SET, SEEK_CUR, SEEK_END */
lock.l_len = len; /* bytes (0 means to EOF) */
return fcntl(fd, cmd, &lock);
}
int main(void)
{
int shmid;
int filed;
pid_t pid;
char *shmptr = NULL;
filed = createFile();
if ( (shmid = shmget(IPC_PRIVATE, SHM_SIZE, SHM_MODE)) < 0) { /* Create shared memory */
fprintf(stderr, "Create shared memory failed!\n");
exit(EXIT_FAILURE);
}
if ( (pid = fork()) < 0) { /* Create a child process */
fprintf(stderr, "Create child process failed!\n");
exit(EXIT_FAILURE);
}
if (pid == 0) { /* In child process */
if ( (shmptr = shmat(shmid, 0, 0)) == (void *)-1) {
fprintf(stderr, "Attached shared memory failed!\n");
exit(EXIT_FAILURE);
}
while (1) {
WRITE_LOCK(filed); /* add a write lock to filed */
/* shared memory 剩余的空間不能存儲"Child"字符串和結(jié)束符0時就退出循環(huán) */
if (SHM_SIZE - strlen(shmptr) < strlen("Child") + 1) break;
strcat(shmptr, "C"); /* 由于加了鎖，每個Child將會連續(xù)出現(xiàn) */
strcat(shmptr, "h");
strcat(shmptr, "i");
strcat(shmptr, "l");
strcat(shmptr, "d");
UN_LOCK(filed); /* release lock */
}
printf("child process:\n\t%s\n", shmptr); /* child process print shared memory */
exit(0);
}
/* In parent process */
if ( (shmptr = shmat(shmid, 0, 0)) == (void *)-1) { /* Attached shared memory */
fprintf(stderr, "Attached shared memory failed!\n");
exit(EXIT_FAILURE);
}
while (1) {
WRITE_LOCK(filed);
/* shared memory 剩余的空間不能存儲"Parent"字符串和結(jié)束符0時就退出循環(huán) */
if (SHM_SIZE - strlen(shmptr) < strlen("Parent") + 1) break;
strcat(shmptr, "P");/* 由于加了鎖，每個Parent將會連續(xù)出現(xiàn) */
strcat(shmptr, "a");
strcat(shmptr, "r");
strcat(shmptr, "e");
strcat(shmptr, "n");
strcat(shmptr, "t");
UN_LOCK(filed);
}
printf("parent process:\n\t%s\n", shmptr); /* parent print shared memory */
wait(0);
exit(0);
}