#include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <fcntl.h> #include <sys/mman.h> #include <pthread.h> pthread_mutex_t* g_mutex; //創建共享的mutex void init_mutex(void) { int ret; //g_mutex一定要是進程間可以共享的,否則無法達到進程間互斥 g_mutex=(pthread_mutex_t*)mmap(NULL, sizeof(pthread_mutex_t), PROT_READ|PROT_WRITE, MAP_SHARED|MAP_ANONYMOUS, -1, 0); if( MAP_FAILED==g_mutex ) { perror("mmap"); exit(1); } //設置attr的屬性 pthread_mutexattr_t attr; pthread_mutexattr_init(&attr); //一定要設置為PTHREAD_PROCESS_SHARED //具體可以參考http://blog.chinaunix.net/u/22935/showart_340408.html ret=pthread_mutexattr_setpshared(&attr,PTHREAD_PROCESS_SHARED); if( ret!=0 ) { perror("init_mutex pthread_mutexattr_setpshared"); exit(1); } pthread_mutex_init(g_mutex, &attr); } int main(int argc, char *argv[]) { init_mutex(); int ret; char str1[]="this is child process/r/n"; char str2[]="this is father process/r/n"; int fd=open("tmp", O_RDWR|O_CREAT|O_TRUNC, 0666); if( -1==fd ) { perror("open"); exit(1); } pid_t pid; pid=fork(); if( pid<0 ) { perror("fork"); exit(1); } else if( 0==pid ) { ret=pthread_mutex_lock(g_mutex); if( ret!=0 ) { perror("child pthread_mutex_lock"); } sleep(10);//測試是否能夠阻止父進程的寫入 write(fd, str1, sizeof(str1)); ret=pthread_mutex_unlock(g_mutex); if( ret!=0 ) { perror("child pthread_mutex_unlock"); } } else { sleep(2);//保證子進程先執行 ret=pthread_mutex_lock(g_mutex); if( ret!=0 ) { perror("father pthread_mutex_lock"); } write(fd, str2, sizeof(str2)); ret=pthread_mutex_unlock(g_mutex); if( ret!=0 ) { perror("father pthread_mutex_unlock"); } } wait(NULL); munmap(g_mutex, sizeof(pthread_mutex_t)); }
運行後tmp文件內容為:
this is child process
this is father process