nginx共享内存的方式

最新推荐文章于 2022-10-04 20:33:23 发布

我要精通C++

最新推荐文章于 2022-10-04 20:33:23 发布

阅读量607

点赞数 1

分类专栏： nginx 文章标签： nginx 共享内存

本文链接：https://blog.csdn.net/Edidaughter/article/details/125558430

版权

nginx 专栏收录该内容

54 篇文章 1 订阅

订阅专栏

1.看下源码

1.1 src\os\unix\ngx_shmem.h

/*
 * Copyright (C) Igor Sysoev
 * Copyright (C) Nginx, Inc.
 */
#ifndef _NGX_SHMEM_H_INCLUDED_
#define _NGX_SHMEM_H_INCLUDED_

#include <ngx_config.h>
#include <ngx_core.h>
typedef struct 
{
    u_char      *addr;
    size_t       size;
    ngx_str_t    name;
    ngx_log_t   *log;
    ngx_uint_t   exists;   /* unsigned  exists:1;  */
} ngx_shm_t;
ngx_int_t ngx_shm_alloc(ngx_shm_t *shm);
void ngx_shm_free(ngx_shm_t *shm);

#endif /* _NGX_SHMEM_H_INCLUDED_ */

1.2 src\os\unix\ngx_shmem.c


/*
 * Copyright (C) Igor Sysoev
 * Copyright (C) Nginx, Inc.
 */

#include <ngx_config.h>
#include <ngx_core.h>

#if (NGX_HAVE_MAP_ANON)

ngx_int_t
ngx_shm_alloc(ngx_shm_t *shm)
{
    shm->addr = (u_char *)mmap(NULL, shm->size, PROT_READ | PROT_WRITE, MAP_ANON | MAP_SHARED, -1, 0);
    if (shm->addr == MAP_FAILED)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "mmap(MAP_ANON|MAP_SHARED, %uz) failed", shm->size);
        return NGX_ERROR;
    }
    return NGX_OK;
}

void ngx_shm_free(ngx_shm_t *shm)
{
    if (munmap((void *)shm->addr, shm->size) == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "munmap(%p, %uz) failed", shm->addr, shm->size);
    }
}

#elif (NGX_HAVE_MAP_DEVZERO)

ngx_int_t
ngx_shm_alloc(ngx_shm_t *shm)
{
    ngx_fd_t fd;
    fd = open("/dev/zero", O_RDWR);
    if (fd == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "open(\"/dev/zero\") failed");
        return NGX_ERROR;
    }
    shm->addr = (u_char *)mmap(NULL, shm->size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
    if (shm->addr == MAP_FAILED)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "mmap(/dev/zero, MAP_SHARED, %uz) failed", shm->size);
    }
    if (close(fd) == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "close(\"/dev/zero\") failed");
    }
    return (shm->addr == MAP_FAILED) ? NGX_ERROR : NGX_OK;
}

void ngx_shm_free(ngx_shm_t *shm)
{
    if (munmap((void *)shm->addr, shm->size) == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "munmap(%p, %uz) failed", shm->addr, shm->size);
    }
}

#elif (NGX_HAVE_SYSVSHM)

#include <sys/ipc.h>
#include <sys/shm.h>

ngx_int_t
ngx_shm_alloc(ngx_shm_t *shm)
{
    int id;
    id = shmget(IPC_PRIVATE, shm->size, (SHM_R | SHM_W | IPC_CREAT));
    if (id == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "shmget(%uz) failed", shm->size);
        return NGX_ERROR;
    }
    ngx_log_debug1(NGX_LOG_DEBUG_CORE, shm->log, 0, "shmget id: %d", id);
    shm->addr = shmat(id, NULL, 0);
    if (shm->addr == (void *)-1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "shmat() failed");
    }
    if (shmctl(id, IPC_RMID, NULL) == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "shmctl(IPC_RMID) failed");
    }
    return (shm->addr == (void *)-1) ? NGX_ERROR : NGX_OK;
}

void ngx_shm_free(ngx_shm_t *shm)
{
    if (shmdt(shm->addr) == -1)
    {
        ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "shmdt(%p) failed", shm->addr);
    }
}

#endif