ssize_t
ngx_parse_size(ngx_str_t *line)
{
u_char unit;
size_t len;
ssize_t size, scale, max;
len = line->len;
unit = line->data[len - 1];
switch (unit)
{
case 'K':
case 'k':
len--;
max = NGX_MAX_SIZE_T_VALUE / 1024;
scale = 1024;
break;
case 'M':
case 'm':
len--;
max = NGX_MAX_SIZE_T_VALUE / (1024 * 1024);
scale = 1024 * 1024;
break;
default:
max = NGX_MAX_SIZE_T_VALUE;
scale = 1;
}
size = ngx_atosz(line->data, len);
if (size == NGX_ERROR || size > max)
{
return NGX_ERROR;
}
size *= scale;
return size;
}
static ngx_int_t
ngx_http_set_content_length_header(ngx_http_request_t *r,
ngx_http_headers_more_header_val_t *hv, ngx_str_t *value)
{
off_t len;
if (value->len == 0) {
return ngx_http_clear_content_length_header(r, hv, value);
}
len = ngx_atosz(value->data, value->len);
if (len == NGX_ERROR) {
return NGX_ERROR;
}
r->headers_out.content_length_n = len;
return ngx_http_set_builtin_header(r, hv, value);
}
ssize_t
ngx_parse_size(ngx_str_t *line)
{
u_char unit;
size_t len;
ssize_t size;
ngx_int_t scale;
len = line->len;
unit = line->data[len - 1];
switch (unit) {
case 'K':
case 'k':
len--;
scale = 1024;
break;
case 'M':
case 'm':
len--;
scale = 1024 * 1024;
break;
default:
scale = 1;
}
size = ngx_atosz(line->data, len);
if (size == NGX_ERROR) {
return NGX_ERROR;
}
size *= scale;
return size;
}
ngx_int_t
ngx_getmeminfo(ngx_meminfo_t *meminfo, ngx_log_t *log)
{
u_char buf[2048];
u_char *p, *start, *last;
size_t *sz = NULL;
ssize_t n, len;
ngx_fd_t fd;
enum {
sw_name = 0,
sw_value_start,
sw_value,
sw_skipline,
sw_newline,
} state;
ngx_memzero(meminfo, sizeof(ngx_meminfo_t));
if (ngx_meminfo_file.fd == 0) {
fd = ngx_open_file(NGX_MEMINFO_FILE, NGX_FILE_RDONLY,
NGX_FILE_OPEN,
NGX_FILE_DEFAULT_ACCESS);
if (fd == NGX_INVALID_FILE) {
ngx_log_error(NGX_LOG_EMERG, log, ngx_errno,
ngx_open_file_n " \"%s\" failed",
NGX_MEMINFO_FILE);
return NGX_ERROR;
}
ngx_meminfo_file.name.data = (u_char *) NGX_MEMINFO_FILE;
ngx_meminfo_file.name.len = ngx_strlen(NGX_MEMINFO_FILE);
ngx_meminfo_file.fd = fd;
}
ngx_meminfo_file.log = log;
n = ngx_read_file(&ngx_meminfo_file, buf, sizeof(buf) - 1, 0);
if (n == NGX_ERROR) {
ngx_log_error(NGX_LOG_ALERT, log, ngx_errno,
ngx_read_file_n " \"%s\" failed",
NGX_MEMINFO_FILE);
return NGX_ERROR;
}
p = buf;
start = buf;
last = buf + n;
state = sw_name;
for (; p < last; p++) {
if (*p == '\n') {
state = sw_newline;
}
switch (state) {
case sw_name:
if (*p != ':') {
continue;
}
len = p - start;
sz = NULL;
switch (len) {
case 6:
/* Cached */
if (meminfo->cachedram == 0 &&
ngx_strncmp(start, "Cached", len) == 0)
{
sz = &meminfo->cachedram;
}
break;
case 7:
/* Buffers MemFree */
if (meminfo->bufferram == 0 &&
ngx_strncmp(start, "Buffers", len) == 0)
{
sz = &meminfo->bufferram;
} else if (meminfo->freeram == 0 &&
ngx_strncmp(start, "MemFree", len) == 0)
{
sz = &meminfo->freeram;
}
break;
case 8:
/* MemTotal SwapFree */
if (meminfo->totalram == 0 &&
ngx_strncmp(start, "MemTotal", len) == 0)
{
sz = &meminfo->totalram;
} else if (meminfo->freeswap == 0 &&
ngx_strncmp(start, "SwapFree", len) == 0)
{
sz = &meminfo->freeswap;
}
break;
case 9:
/* SwapTotal */
if (meminfo->totalswap == 0 &&
ngx_strncmp(start, "SwapTotal", len) == 0)
{
sz = &meminfo->totalswap;
}
break;
}
if (sz == NULL) {
state = sw_skipline;
continue;
}
state = sw_value_start;
continue;
case sw_value_start:
if (*p == ' ') {
continue;
}
start = p;
state = sw_value;
continue;
case sw_value:
if (*p >= '0' && *p <= '9') {
continue;
}
*(sz) = ngx_atosz(start, p - start) * 1024;
state = sw_skipline;
continue;
case sw_skipline:
continue;
case sw_newline:
state = sw_name;
start = p + 1;
continue;
}
}
return NGX_OK;
}
