linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段

转载 2016年10月26日 15:31:00

生产者消费者问题:该问题描述了两个共享固定大小缓冲区的进程——即所谓的“生产者”和“消费者”——在实际运行时会发生的问题。生产者的主要作用是生成一定量的数据放到缓冲区中,然后重复此过程。与此同时,消费者也在缓冲区消耗这些数据。该问题的关键就是要保证生产者不会在缓冲区满时加入数据,消费者也不会在缓冲区中空时消耗数据。
我们可以用信号量解决生产者消费者问题,如下图:
这里写图片描述
定义3个信号量,sem_full 和 sem_empty 用于生产者进程和消费者进程之间同步,即缓冲区为空才能生产,缓冲区不为空才能消费。由于共享同一块缓冲区,在生产一个产品过程中不能生产/消费产品,在消费一个产品的过程中不能生产/消费产品,故再使用一个 sem_mutex 信号量来约束行为,即进程间互斥。

下面基于生产者消费者模型,来实现一个先进先出的共享内存段:
这里写图片描述
如上图所示,定义两个结构体,shmhead 是共享内存段的头部,保存了块大小,块数,读写索引。shmfifo 保存了共享内存头部的指针,有效负载的起始地址,创建的共享内存段的shmid,以及3个信号量集的semid。
下面来封装几个函数:

#include "shmfifo.h"
#include <assert.h>

shmfifo_t *shmfifo_init(int key, int blksize, int blocks)
{
    shmfifo_t *fifo = (shmfifo_t *)malloc(sizeof(shmfifo_t));
    assert(fifo != NULL);
    memset(fifo, 0, sizeof(shmfifo_t));


    int shmid;
    shmid = shmget(key, 0, 0);
    int size = sizeof(shmhead_t) + blksize * blocks;
    if (shmid == -1)
    {
        fifo->shmid = shmget(key, size, IPC_CREAT | 0666);
        if (fifo->shmid == -1)
            ERR_EXIT("shmget");

        fifo->p_shm = (shmhead_t *)shmat(fifo->shmid, NULL, 0);
        if (fifo->p_shm == (shmhead_t *) - 1)
            ERR_EXIT("shmat");

        fifo->p_payload = (char *)(fifo->p_shm + 1);

        fifo->p_shm->blksize = blksize;
        fifo->p_shm->blocks = blocks;
        fifo->p_shm->rd_index = 0;
        fifo->p_shm->wr_index = 0;

        fifo->sem_mutex = sem_create(key);
        fifo->sem_full = sem_create(key + 1);
        fifo->sem_empty = sem_create(key + 2);

        sem_setval(fifo->sem_mutex, 1);
        sem_setval(fifo->sem_full, blocks);
        sem_setval(fifo->sem_empty, 0);
    }
    else
    {
        fifo->shmid = shmid;
        fifo->p_shm = (shmhead_t *)shmat(fifo->shmid, NULL, 0);
        if (fifo->p_shm == (shmhead_t *) - 1)
            ERR_EXIT("shmat");

        fifo->p_payload = (char *)(fifo->p_shm + 1);

        fifo->sem_mutex = sem_open(key);
        fifo->sem_full = sem_open(key + 1);
        fifo->sem_empty = sem_open(key + 2);
    }

    return fifo;
}

void shmfifo_put(shmfifo_t *fifo, const void *buf)
{
    sem_p(fifo->sem_full);
    sem_p(fifo->sem_mutex);

    memcpy(fifo->p_payload + fifo->p_shm->blksize * fifo->p_shm->wr_index,
           buf, fifo->p_shm->blksize);
    fifo->p_shm->wr_index = (fifo->p_shm->wr_index + 1) % fifo->p_shm->blocks;
    sem_v(fifo->sem_mutex);
    sem_v(fifo->sem_empty);
}

void shmfifo_get(shmfifo_t *fifo, void *buf)
{
    sem_p(fifo->sem_empty);
    sem_p(fifo->sem_mutex);

    memcpy(buf, fifo->p_payload + fifo->p_shm->blksize * fifo->p_shm->rd_index,
           fifo->p_shm->blksize);
    fifo->p_shm->rd_index = (fifo->p_shm->rd_index + 1) % fifo->p_shm->blocks;
    sem_v(fifo->sem_mutex);
    sem_v(fifo->sem_full);
}

void shmfifo_destroy(shmfifo_t *fifo)
{
    sem_d(fifo->sem_mutex);
    sem_d(fifo->sem_full);
    sem_d(fifo->sem_empty);

    shmdt(fifo->p_shm);
    shmctl(fifo->shmid, IPC_RMID, 0);
    free(fifo);
}

1、shmfifo_init:先分配shmfifo 结构体的内存,如果尝试打开共享内存失败则创建,创建的共享内存段大小 = shmhead大小 + 块大小×块数目,然后shmat将此共享内存段映射到进程地址空间,然后使用sem_create 创建3个信号量集,每个信号集只有一个信号量,即上面提到的3个信号量,设置每个信号量的资源初始值。如果共享内存已经存在,则直接shmat映射下即可,此时3个信号量集也已经存在,sem_open 打开即可。sem_xxx 系列封装函数参考这里。
2、shmfifo_put:参照第一个生产者消费者的图,除去sem_p,sem_v 操作之外,中间就将buf 的内容memcpy 到对应缓冲区块,然后移动wr_index。
3、shmfifo_get:与shmfifo_put 类似,执行的是相反的操作。
4、shmfifo_destroy:删除3个信号量集,将共享内存段从进程地址空间剥离,删除共享内存段,释放shmfifo 结构体的内存。

下面是生产者程序和消费者程序:
shmfifo_send.c

#include "shmfifo.h"

typedef struct stu
{
    char name[32];
    int age;
} STU;
int main(void)
{
    shmfifo_t *fifo = shmfifo_init(1234, sizeof(STU), 3);

    STU s;
    memset(&s, 0, sizeof(STU));
    s.name[0] = 'A';
    int i;

    for (i = 0; i < 5; i++)
    {
        s.age = 20 + i;
        shmfifo_put(fifo, &s);
        s.name[0] = s.name[0] + 1;

        printf("send ok\n");
    }

    free(fifo);

    return 0;
}

shmfifo_recv.c

#include "shmfifo.h"

typedef struct stu
{
    char name[32];
    int age;
} STU;

int main(void)
{
    shmfifo_t *fifo = shmfifo_init(1234, sizeof(STU), 3);

    STU s;
    memset(&s, 0, sizeof(STU));
    int i;

    for (i = 0; i < 5; i++)
    {
        shmfifo_get(fifo, &s);
        printf("name = %s age = %d\n", s.name, s.age);
    }

    shmfifo_destroy(fifo);

    return 0;
}

先运行生产者进程,输出如下:
simba@ubuntu:~/Documents/code/linux_programming/UNP/system_v/shmfifo$ ./shmfifo_send
send ok
send ok
send ok

因为共享内存只有3块block,故发送了3次后再次P(semfull)就阻塞了,等待消费者读取数据,现在运行消费者进程
simba@ubuntu:~/Documents/code/linux_programming/UNP/system_v/shmfifo$ ./shmfifo_recv
name = A age = 20
name = B age = 21
name = C age = 22
name = D age = 23
name = E age = 24
因为生产者已经创建了一块共享内存,故消费者只是打开而已,当读取了第一块数据之后,生产者会再次写入,依次输出后两个 send ok,可以推论的是D是重新写到共享内存开始的第一块,E是第二块,类似环形队列。
从输出可以看出,的确实现了数据的先进先出。

PS:在生产实践中也可以看到利用共享内存实现环形缓冲区 or 哈希表 的例子。

参考:《UNP》
转载自http://blog.csdn.net/jnu_simba/article/details/9103059

linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段

生产者消费者问题:该问题描述了两个共享固定大小缓冲区的进程——即所谓的“生产者”和“消费者”——在实际运行时会发生的问题。生产者的主要作用是生成一定量的数据放到缓冲区中,然后重复此过程。与此同时,消费...
  • Simba888888
  • Simba888888
  • 2013年06月15日 22:35
  • 4512

Linux进程间通信与生产者消费者问题

生产者消费者问题(英语:Producer-consumerproblem),也称有限缓冲问题(英语:Bounded-bufferproblem),是一个多线程同步问题的经典案例。该问题描述了两个共享固...
  • yongmi
  • yongmi
  • 2013年01月20日 20:43
  • 4559

Linux 系统应用编程——多线程经典问题(生产者-消费者)

“生产者——消费者”问题是Linux多线程编程中的经典问题,主要是利用信号量处理线程间的同步和互斥问题。          “生产者——消费者”问题描述如下:           有一个有限缓冲区...
  • zqixiao_09
  • zqixiao_09
  • 2016年01月10日 12:47
  • 2747

linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段

生产者消费者问题:该问题描述了两个共享固定大小缓冲区的进程——即所谓的“生产者”和“消费者”——在实际运行时会发生的问题。生产者的主要作用是生成一定量的数据放到缓冲区中,然后重复此过程。与此同时,消费...
  • Simba888888
  • Simba888888
  • 2013年06月15日 22:35
  • 4512

linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段 .

linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段 分类: linux网络编程 2013-06-15 22:35 563人阅读 评论(...
  • linuxarmsummary
  • linuxarmsummary
  • 2013年09月29日 08:49
  • 635

Linux进程间通信(IPC)编程实践(十一)System V信号量---实现一个先进先出的共享内存shmfifo

使用消息队列即可实现消息的先进先出(FIFO), 但是使用共享内存实现消息的先进先出则更加快速;    我们首先完成C语言版本的shmfifo(基于过程调用), 然后在此基础上实现C++版本的S...
  • NK_test
  • NK_test
  • 2015年12月08日 14:25
  • 2015

System V共享内存与信号量综合应用之生产者与消费者问题解决

  • 2015年05月27日 00:34
  • 5KB
  • 下载

system V 共享内存和信号量的组合使用 实现生产者消费者问题

看到别人写的,可是不是我想要的形式,改写下。 先运行消费者进程 producer.c #include #include #include #include #inc...
  • worshiper
  • worshiper
  • 2012年04月17日 23:34
  • 575

用阻塞队列LinkedBlockingQueue实现生产者消费者先进先出

在Java多线程应用中,队列的使用率很高,多数生产消费模型的首选数据结构就是队列(先进先出)。Java提供的线程安全的Queue可以分为阻塞队列和非阻塞队列,其中阻塞队列的典型例子是BlockingQ...
  • done58
  • done58
  • 2016年03月28日 11:18
  • 2408

61-System V 信号量(生产者与消费者模型)

学过 OS 的同学一定知道这个模型,这里再简单回忆一下。最后给出基于 System V 信号量的完整实现。1. PV 原语 P(S) :表示将资源 S 数量减 1,即 S = S - 1. 如果 S ...
  • q1007729991
  • q1007729991
  • 2017年01月15日 10:44
  • 449
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:linux网络编程之System V 信号量(三):基于生产者-消费者模型实现先进先出的共享内存段
举报原因:
原因补充:

(最多只允许输入30个字)