nginx/src/os/unix/ngx_shmem.c

   1
   2 /*
   3  * Copyright (C) Igor Sysoev
   4  */
   5
   6
   7 #include <ngx_config.h>
   8 #include <ngx_core.h>
   9
  10
  11 #if (NGX_HAVE_MAP_ANON)
  12
  13 ngx_int_t
  14 ngx_shm_alloc(ngx_shm_t *shm)
  15 {
  16     shm->addr = (u_char *) mmap(NULL, shm->size,
  17                                 PROT_READ|PROT_WRITE,
  18                                 MAP_ANON|MAP_SHARED, -1, 0);
  19
  20     if (shm->addr == MAP_FAILED) {
  21         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  22                       "mmap(MAP_ANON|MAP_SHARED, %uz) failed", shm->size);
  23         return NGX_ERROR;
  24     }
  25
  26     return NGX_OK;
  27 }
  28
  29
  30 void
  31 ngx_shm_free(ngx_shm_t *shm)
  32 {
  33     if (munmap((void *) shm->addr, shm->size) == -1) {
  34         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  35                       "munmap(%p, %uz) failed", shm->addr, shm->size);
  36     }
  37 }
  38
  39 #elif (NGX_HAVE_MAP_DEVZERO)
  40
  41 ngx_int_t
  42 ngx_shm_alloc(ngx_shm_t *shm)
  43 {
  44     ngx_fd_t  fd;
  45
  46     fd = open("/dev/zero", O_RDWR);
  47
  48     if (fd == -1) {
  49         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  50                       "open(\"/dev/zero\") failed");
  51         return NGX_ERROR;
  52     }
  53
  54     shm->addr = (u_char *) mmap(NULL, shm->size, PROT_READ|PROT_WRITE,
  55                                 MAP_SHARED, fd, 0);
  56
  57     if (shm->addr == MAP_FAILED) {
  58         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  59                       "mmap(/dev/zero, MAP_SHARED, %uz) failed", shm->size);
  60     }
  61
  62     if (close(fd) == -1) {
  63         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  64                       "close(\"/dev/zero\") failed");
  65     }
  66
  67     return (shm->addr == MAP_FAILED) ? NGX_ERROR : NGX_OK;
  68 }
  69
  70
  71 void
  72 ngx_shm_free(ngx_shm_t *shm)
  73 {
  74     if (munmap((void *) shm->addr, shm->size) == -1) {
  75         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  76                       "munmap(%p, %uz) failed", shm->addr, shm->size);
  77     }
  78 }
  79
  80 #elif (NGX_HAVE_SYSVSHM)
  81
  82 #include <sys/ipc.h>
  83 #include <sys/shm.h>
  84
  85
  86 ngx_int_t
  87 ngx_shm_alloc(ngx_shm_t *shm)
  88 {
  89     int  id;
  90
  91     id = shmget(IPC_PRIVATE, shm->size, (SHM_R|SHM_W|IPC_CREAT));
  92
  93     if (id == -1) {
  94         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
  95                       "shmget(%uz) failed", shm->size);
  96         return NGX_ERROR;
  97     }
  98
  99     ngx_log_debug1(NGX_LOG_DEBUG_CORE, shm->log, 0, "shmget id: %d", id);
 100
 101     shm->addr = shmat(id, NULL, 0);
 102
 103     if (shm->addr == (void *) -1) {
 104         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno, "shmat() failed");
 105     }
 106
 107     if (shmctl(id, IPC_RMID, NULL) == -1) {
 108         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
 109                       "shmctl(IPC_RMID) failed");
 110     }
 111
 112     return (shm->addr == (void *) -1) ? NGX_ERROR : NGX_OK;
 113 }
 114
 115
 116 void
 117 ngx_shm_free(ngx_shm_t *shm)
 118 {
 119     if (shmdt(shm->addr) == -1) {
 120         ngx_log_error(NGX_LOG_ALERT, shm->log, ngx_errno,
 121                       "shmdt(%p) failed", shm->addr);
 122     }
 123 }
 124
 125 #endif