71 lines
1.3 KiB
C
71 lines
1.3 KiB
C
#include "pid_manager.h"
|
|
#include <pthread.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#define MIN_PID 300
|
|
#define MAX_PID 5000
|
|
#define SIZE (MAX_PID - MIN_PID + 1)
|
|
|
|
bool pids[SIZE];
|
|
|
|
void *runner(void *param);
|
|
|
|
int allocate_map(void) {
|
|
for (int i = 0; i < SIZE; i++) {
|
|
pids[i] = false;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
int allocate_pid(void) {
|
|
for (int i = 0; i < SIZE; i++) {
|
|
if (!pids[i]) {
|
|
pids[i] = true;
|
|
return i + MIN_PID;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
void release_pid(int pid) {
|
|
if (pid >= MIN_PID && pid <= MAX_PID) {
|
|
pids[pid - MIN_PID] = false;
|
|
}
|
|
}
|
|
|
|
void *runner(void *param) {
|
|
const pid_t self = gettid();
|
|
const int pid = allocate_pid();
|
|
printf("Run thread %d handle PID %d.\n", self, pid);
|
|
sleep(rand() % 31);
|
|
release_pid(pid);
|
|
printf("Thread %d ended\n", self);
|
|
pthread_exit(0);
|
|
}
|
|
|
|
int main(void) {
|
|
const int thread_count = 100;
|
|
pthread_t threads[thread_count];
|
|
|
|
if (!allocate_map()) {
|
|
perror("Failed to allocate PID Manager\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
for (int i = 0; i < thread_count; i++) {
|
|
printf("Created thread %d\n", i);
|
|
pthread_create(&threads[i], NULL, runner, NULL);
|
|
}
|
|
|
|
for (int i = 0; i < thread_count; i++) {
|
|
printf("Waiting thread %d\n", i);
|
|
pthread_join(threads[i], NULL);
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|