test_kfifo_thread_safe.c

// Copyright (c) 2024 cheng-zhongliang. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

#define _GNU_SOURCE

#include <assert.h>
#include <pthread.h>
#include <sched.h>

#include "kfifo.h"

void set_cpu_affinity(int cpu_no) {
    pthread_t t = pthread_self();
    cpu_set_t cpu;
    CPU_ZERO(&cpu);
    CPU_SET(cpu_no, &cpu);
    assert(pthread_setaffinity_np(t, sizeof(cpu_set_t), &cpu) == 0);
}

void* producer_thread(void* arg) {
    set_cpu_affinity(0);
    _KFIFO(kfifo, int)* kfifo = arg;
    int n = 0;
    do {
        if(KFIFO_FULL(kfifo)) {
            continue;
        }
        int* nn = calloc(1, sizeof(int));
        *nn = n;
        KFIFO_ENQUEUE(kfifo, nn);
        n++;
    } while(n != 10000);
    return NULL;
}

void* consumer_thread(void* arg) {
    set_cpu_affinity(1);
    _KFIFO(kfifo, int)* kfifo = arg;
    int n = 0;
    do {
        if(KFIFO_EMPTY(kfifo)) {
            continue;
        }
        int* nn;
        KFIFO_DEQUEUE(kfifo, nn);
        assert(*nn == n);
        free(nn);
        n++;
    } while(n != 10000);
    return NULL;
}

int main(void) {
    _KFIFO(kfifo, int) kfifo;
    KFIFO_INIT(&kfifo, 16);
    pthread_t producer, consumer;
    pthread_create(&producer, NULL, producer_thread, &kfifo);
    pthread_create(&consumer, NULL, consumer_thread, &kfifo);
    pthread_join(producer, NULL);
    pthread_join(consumer, NULL);
    return 0;
}