多個生產者及多個消費者..問題??

angus203

請問這樣能否平衡兩者所需??

信號量及共享記憶體 初始化

1. 
   
2. #include <sys/types.h>
   
3. #include <sys/ipc.h>
   
4. #include <sys/shm.h>
   
5. #include <stdio.h>
   
6. #include <unistd.h>
   
7. #include <stdlib.h>
   
8. #include <sys/sem.h>
   
9. 
   
10. #define KEY (1492)
    
11. 
    
12. void sem_lock(int sem_set_id)
    
13. {
    
14.     /* structure for semaphore operations.   */
    
15.     struct sembuf sem_op;
    
16. 
    
17.     /* wait on the semaphore, unless it's value is non-negative. */
    
18.     sem_op.sem_num = 0;
    
19.     sem_op.sem_op = -1;
    
20.     sem_op.sem_flg = 0;
    
21.     semop(sem_set_id, &sem_op, 1);
    
22. }
    
23. 
    
24. void sem_unlock(int sem_set_id)
    
25. {
    
26.     /* structure for semaphore operations.   */
    
27.     struct sembuf sem_op;
    
28. 
    
29.     /* signal the semaphore - increase its value by one. */
    
30.     sem_op.sem_num = 0;
    
31.     sem_op.sem_op = 1;   /* <-- Comment 3 */
    
32.     sem_op.sem_flg = 0;
    
33.     semop(sem_set_id, &sem_op, 1);
    
34. }
    
35. 
    
36. void shared_memory()
    
37. {
    
38.   int shmid;
    
39.   int *shm;
    
40.   key_t key = IPC_PRIVATE;
    
41.   int flag = SHM_R | SHM_W;
    
42. 
    
43. 
    
44.    key = 1234;
    
45. 
    
46. 
    
47.   shmid = shmget(key, 1000, flag);
    
48.   printf("shared memory for key %d: %d\n", key, shmid);
    
49.   if (shmid < 0) {
    
50.     perror("shmget");
    
51.     printf("Try to create this segment\n");
    
52.     shmid = shmget(key, 1000, flag | IPC_CREAT);
    
53.     if (shmid < 0) {
    
54.       perror("shmget | IPC_CREAT");
    
55.     }
    
56.   }
    
57. 
    
58.   shm = (int *)shmat(shmid, /*addr*/0, /*flag*/0);
    
59.   if (shm == (int *)-1) {
    
60.     perror("shmat");
    
61.     exit(1);
    
62.   }
    
63.   [color=red]*shm = 5;[/color]
    
64.   printf("shared memory content: %d\n", *shm);
    
65. 
    
66. }
    
67. 
    
68. void sema_init()
    
69. {
    
70.    int id;
    
71.    union semun {
    
72.            int val;
    
73.            struct semid_ds *buf;
    
74.            ushort * array;
    
75.            } argument;
    
76. 
    
77.    argument.val = 0;
    
78. 
    
79.    id = semget(KEY, 1, 0666 | IPC_CREAT);
    
80. 
    
81.    if(id < 0)
    
82.    {
    
83.       fprintf(stderr, "Unable to obtain semaphore.\n");
    
84.       exit(0);
    
85.    }
    
86. 
    
87.    if( semctl(id, 0, SETVAL, argument) < 0)
    
88.    {
    
89.       fprintf( stderr, "Cannot set semaphore value.\n");
    
90.    }
    
91.    else
    
92.    {
    
93.       fprintf(stderr, "Semaphore %d initialized.\n", KEY);
    
94.    }
    
95. }
    
96. int main(){
    
97.         sema_init();
    
98.         shared_memory();
    
99.         return 0;
    
100. }
     
101. 
     

复制代码

生產者

1. 
   
2. #include <sys/types.h>
   
3. #include <sys/ipc.h>
   
4. #include <sys/shm.h>
   
5. #include <stdio.h>
   
6. #include <unistd.h>
   
7. #include <stdlib.h>
   
8. #include <sys/sem.h>
   
9. 
   
10. #define KEY (1492)
    
11. 
    
12. void sem_lock(int sem_set_id)
    
13. {
    
14.     /* structure for semaphore operations.   */
    
15.     struct sembuf sem_op;
    
16. 
    
17.     /* wait on the semaphore, unless it's value is non-negative. */
    
18.     sem_op.sem_num = 0;
    
19.     sem_op.sem_op = -1;
    
20.     sem_op.sem_flg = 0;
    
21.     semop(sem_set_id, &sem_op, 1);
    
22. }
    
23. 
    
24. void sem_unlock(int sem_set_id)
    
25. {
    
26.     /* structure for semaphore operations.   */
    
27.     struct sembuf sem_op;
    
28. 
    
29.     /* signal the semaphore - increase its value by one. */
    
30.     sem_op.sem_num = 0;
    
31.     sem_op.sem_op = 1;   /* <-- Comment 3 */
    
32.     sem_op.sem_flg = 0;
    
33.     semop(sem_set_id, &sem_op, 1);
    
34. }
    
35. 
    
36. int main()
    
37. {
    
38.   int shmid;
    
39.   int *shm;
    
40.   key_t key;
    
41. 
    
42.   key = 1234;
    
43. 
    
44.   printf("Trying shared memory %d\n", key);
    
45.   shmid = shmget(key, 1000, SHM_R | SHM_W);
    
46.   if (shmid < 0) {
    
47.     perror("shmget");
    
48.     shmid = key;
    
49.   }
    
50.   else {
    
51.     printf("shared memory: %d\n", shmid);
    
52.   }
    
53.   shm = (int *)shmat(shmid, /*addr*/0, /*flag*/0);
    
54.   if (shm == (int *)-1) {
    
55.     perror("shmat");
    
56.     exit(1);
    
57.   }
    
58.   printf("shared memory: %p\n", *shm);
    
59.   if (shm != 0) {
    
60.     printf("shared memory content: %d\n", *shm);
    
61.   }
    
62. 
    
63.         int sem_set_id;
    
64.         sem_set_id = semget(KEY, 1, 0666);
    
65. 
    
66.         sem_unlock(sem_set_id);
    
67. 
    
68.         int pause_time;
    
69.         int i;
    
70.         for(i=0; i<10; i++){
    
71.                 sem_lock(sem_set_id);
    
72.                 pause_time = rand() % 3;
    
73.             sleep(pause_time);
    
74.                 if(*shm < 5){
    
75.                         *shm = *shm + 1;
    
76.                         printf("producer item: %d\n", *shm);
    
77.                 }else {printf("Full: %d\n", *shm);}
    
78.                 //printf("C");
    
79.                 fflush(stdout);
    
80.                 sem_unlock(sem_set_id);
    
81.                 if(*shm < 5){
    
82.                         *shm = *shm + 1;
    
83.                         printf("producer item: %d\n", *shm);
    
84.                 }else {printf("Full: %d\n", *shm);}
    
85.                 //printf("C");
    
86.                 fflush(stdout);
    
87.                 pause_time = rand() % 2;
    
88.             sleep(pause_time);
    
89. 
    
90.         }
    
91. 
    
92.   return 0;
    
93. } /* main */
    
94. 
    

复制代码

消費者

1. 
   
2. #include <sys/types.h>
   
3. #include <sys/ipc.h>
   
4. #include <sys/shm.h>
   
5. #include <stdio.h>
   
6. #include <unistd.h>
   
7. #include <stdlib.h>
   
8. #include <sys/sem.h>
   
9. 
   
10. #define KEY (1492)
    
11. 
    
12. void sem_lock(int sem_set_id)
    
13. {
    
14.     /* structure for semaphore operations.   */
    
15.     struct sembuf sem_op;
    
16. 
    
17.     /* wait on the semaphore, unless it's value is non-negative. */
    
18.     sem_op.sem_num = 0;
    
19.     sem_op.sem_op = -1;
    
20.     sem_op.sem_flg = 0;
    
21.     semop(sem_set_id, &sem_op, 1);
    
22. }
    
23. 
    
24. void sem_unlock(int sem_set_id)
    
25. {
    
26.     /* structure for semaphore operations.   */
    
27.     struct sembuf sem_op;
    
28. 
    
29.     /* signal the semaphore - increase its value by one. */
    
30.     sem_op.sem_num = 0;
    
31.     sem_op.sem_op = 1;   /* <-- Comment 3 */
    
32.     sem_op.sem_flg = 0;
    
33.     semop(sem_set_id, &sem_op, 1);
    
34. }
    
35. 
    
36. int main()
    
37. {
    
38.   int shmid;
    
39.   int *shm;
    
40.   key_t key;
    
41. 
    
42.   key = 1234;
    
43. 
    
44.   printf("Trying shared memory %d\n", key);
    
45.   shmid = shmget(key, 1000, SHM_R | SHM_W);
    
46.   if (shmid < 0) {
    
47.     perror("shmget");
    
48.     shmid = key;
    
49.   }
    
50.   else {
    
51.     printf("shared memory: %d\n", shmid);
    
52.   }
    
53.   shm = (int *)shmat(shmid, /*addr*/0, /*flag*/0);
    
54.   if (shm == (int *)-1) {
    
55.     perror("shmat");
    
56.     exit(1);
    
57.   }
    
58.   printf("shared memory: %p\n", *shm);
    
59.   if (shm != 0) {
    
60.     printf("shared memory content: %d\n", *shm);
    
61.   }
    
62. 
    
63.         int sem_set_id;
    
64.         sem_set_id = semget(KEY, 1, 0666);
    
65. 
    
66.         sem_unlock(sem_set_id);
    
67. 
    
68.         int pause_time;
    
69.         int i;
    
70.         sleep(2);
    
71.         for(i=0; i<10; i++){
    
72. 
    
73.                 sem_lock(sem_set_id);
    
74.                 pause_time = rand() % 3;
    
75.             sleep(pause_time);
    
76.                 if(*shm > 0){
    
77.                         *shm = *shm - 1;
    
78.                         printf("Consuemr item: %d\n", *shm);
    
79.                 }else {printf("Out of Stock: %d\n", *shm);}
    
80.                 //printf("D");
    
81.                 fflush(stdout);
    
82.                 sem_unlock(sem_set_id);
    
83.                 if(*shm > 0){
    
84.                         *shm = *shm - 1;
    
85.                         printf("Consuemr item: %d\n", *shm);
    
86.                 }else {printf("Out of Stock: %d\n", *shm);}
    
87.                 //printf("D");
    
88.                 fflush(stdout);
    
89.                 pause_time = rand() % 2;
    
90.             sleep(pause_time);
    
91.         }
    
92. 
    
93.   return 0;
    
94. } /* main */
    
95. 
    

复制代码