/
ngx_http_rds_csv_util.c
105 lines (78 loc) · 1.86 KB
/
ngx_http_rds_csv_util.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/*
* Copyright (C) Yichun Zhang (agentzh)
*/
#ifndef DDEBUG
#define DDEBUG 0
#endif
#include "ddebug.h"
#include "resty_dbd_stream.h"
#include "ngx_http_rds_csv_util.h"
uintptr_t
ngx_http_rds_csv_escape_csv_str(u_char field_sep, u_char *dst, u_char *src,
size_t size, unsigned *need_quotes)
{
ngx_uint_t n;
if (dst == NULL) {
*need_quotes = 0;
/* find the number of characters to be escaped */
n = 0;
while (size) {
switch (*src) {
case '"':
n++;
/* fallthrough */
case '\r':
case '\n':
*need_quotes = 1;
break;
default:
if (*src == field_sep) {
*need_quotes = 1;
}
break;
}
src++;
size--;
}
return (uintptr_t) n;
}
while (size) {
if (*src == '"') {
*dst++ = '"';
*dst++ = '"';
src++;
} else {
*dst++ = *src++;
}
size--;
}
return (uintptr_t) dst;
}
ngx_int_t
ngx_http_rds_csv_test_content_type(ngx_http_request_t *r)
{
ngx_str_t *type;
type = &r->headers_out.content_type;
if (type->len != rds_content_type_len
|| ngx_strncmp(type->data, rds_content_type, rds_content_type_len)
!= 0)
{
return NGX_DECLINED;
}
return NGX_OK;
}
void
ngx_http_rds_csv_discard_bufs(ngx_pool_t *pool, ngx_chain_t *in)
{
ngx_chain_t *cl;
for (cl = in; cl; cl = cl->next) {
#if 0
if (cl->buf->temporary
&& ngx_buf_size(cl->buf) > 0)
{
ngx_pfree(pool, cl->buf->start);
}
#endif
cl->buf->pos = cl->buf->last;
}
}