Coverage Report

Created: 2023-06-07 06:38

/src/httpd/server/apreq_parser_header.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
**  Licensed to the Apache Software Foundation (ASF) under one or more
3
** contributor license agreements.  See the NOTICE file distributed with
4
** this work for additional information regarding copyright ownership.
5
** The ASF licenses this file to You under the Apache License, Version 2.0
6
** (the "License"); you may not use this file except in compliance with
7
** the License.  You may obtain a copy of the License at
8
**
9
**      http://www.apache.org/licenses/LICENSE-2.0
10
**
11
**  Unless required by applicable law or agreed to in writing, software
12
**  distributed under the License is distributed on an "AS IS" BASIS,
13
**  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
**  See the License for the specific language governing permissions and
15
**  limitations under the License.
16
*/
17
#include <assert.h>
18
#include "apreq_parser.h"
19
#include "apreq_error.h"
20
#include "apreq_util.h"
21
22
#include "apr_lib.h" /* for apr_iscntrl() & co */
23
24
961
#define PARSER_STATUS_CHECK(PREFIX)   do {         \
25
961
    if (ctx->status == PREFIX##_ERROR)             \
26
961
        return APREQ_ERROR_GENERAL;                \
27
961
    else if (ctx->status == PREFIX##_COMPLETE)     \
28
961
        return APR_SUCCESS;                        \
29
961
    else if (bb == NULL)                           \
30
961
        return APR_INCOMPLETE;                     \
31
961
} while (0);
32
33
34
struct hdr_ctx {
35
    apr_bucket_brigade *bb;
36
    apr_size_t          nlen;
37
    apr_size_t          glen;
38
    apr_size_t          vlen;
39
    enum {
40
        HDR_NAME,
41
        HDR_GAP,
42
        HDR_VALUE,
43
        HDR_NEWLINE,
44
        HDR_ENDLINE,
45
        HDR_FOLDLINE,
46
        HDR_LASTLINE,
47
        HDR_COMPLETE,
48
        HDR_ERROR
49
    }                   status;
50
};
51
52
/********************* header parsing utils ********************/
53
54
55
static apr_bucket *split_header_line(apr_bucket *e, apr_size_t *off,
56
                                     const char **data, apr_size_t *dlen)
57
4.28k
{
58
4.28k
    if (*off > 1) {
59
4.28k
        apr_bucket_split(e, *off - 1);
60
4.28k
        e = APR_BUCKET_NEXT(e);
61
4.28k
        *dlen -= *off - 1;
62
4.28k
        *data += *off - 1;
63
4.28k
        *off = 1;
64
4.28k
    }
65
4.28k
    return e;
66
4.28k
}
67
68
static apr_status_t consume_header_line(apreq_param_t **p,
69
                                        apr_pool_t *pool,
70
                                        apr_bucket_brigade *bb,
71
                                        apr_size_t nlen,
72
                                        apr_size_t glen,
73
                                        apr_size_t vlen)
74
1.38k
{
75
1.38k
    apreq_param_t *param;
76
1.38k
    apreq_value_t *v;
77
1.38k
    apr_bucket *e, *f;
78
1.38k
    apr_status_t s;
79
1.38k
    struct iovec vec[APREQ_DEFAULT_NELTS], *iov;
80
1.38k
    apr_array_header_t arr;
81
1.38k
    char *dest;
82
1.38k
    const char *data;
83
1.38k
    apr_size_t dlen;
84
1.38k
    int i, eol = 0;
85
86
1.38k
    param = apreq_param_make(pool, NULL, nlen, NULL, vlen);
87
1.38k
    if (param == NULL)
88
0
        return APR_ENOMEM;
89
1.38k
    *(const apreq_value_t **)&v = &param->v;
90
91
1.38k
    arr.pool     = pool;
92
1.38k
    arr.elt_size = sizeof(struct iovec);
93
1.38k
    arr.nelts    = 0;
94
1.38k
    arr.nalloc   = APREQ_DEFAULT_NELTS;
95
1.38k
    arr.elts     = (char *)vec;
96
97
1.38k
    e = APR_BRIGADE_FIRST(bb);
98
99
    /* store name in a temporary iovec array */
100
1.39k
    do {
101
1.39k
        apr_size_t len;
102
103
1.39k
        assert(e != APR_BRIGADE_SENTINEL(bb));
104
1.39k
        iov = (struct iovec *)apr_array_push(&arr);
105
1.39k
        s = apr_bucket_read(e, (const char **)&iov->iov_base,
106
1.39k
                            &len, APR_BLOCK_READ);
107
1.39k
        if (s != APR_SUCCESS)
108
0
            return s;
109
1.39k
        iov->iov_len = len;
110
111
1.39k
        assert(nlen >= len);
112
1.39k
        nlen -= len;
113
114
1.39k
        e = APR_BUCKET_NEXT(e);
115
1.39k
    } while (nlen > 0);
116
117
    /* skip gap */
118
1.39k
    do {
119
1.39k
        assert(e != APR_BRIGADE_SENTINEL(bb));
120
1.39k
        s = apr_bucket_read(e, &data, &dlen, APR_BLOCK_READ);
121
1.39k
        if (s != APR_SUCCESS)
122
0
            return s;
123
124
1.39k
        assert(glen >= dlen);
125
1.39k
        glen -= dlen;
126
127
1.39k
        e = APR_BUCKET_NEXT(e);
128
1.39k
    } while (glen > 0);
129
130
    /* copy value */
131
1.38k
    dest = v->data;
132
1.39k
    do {
133
1.39k
        apr_size_t off;
134
135
1.39k
        assert(e != APR_BRIGADE_SENTINEL(bb));
136
1.39k
        s = apr_bucket_read(e, &data, &dlen, APR_BLOCK_READ);
137
1.39k
        if (s != APR_SUCCESS)
138
0
            return s;
139
140
14.4k
        for (off = 0; off < dlen; ++off) {
141
13.1k
            const char ch = data[off];
142
13.1k
            if (ch == '\r' || ch == '\n') {
143
                /* Eat [CR]LF of continuation or end of line */
144
1.81k
                if (!vlen && ch == '\n')
145
1.38k
                    eol = 1; /* done */
146
1.81k
                continue;
147
1.81k
            }
148
11.2k
            assert(vlen > 0);
149
11.2k
            *dest = ch;
150
11.2k
            ++dest;
151
11.2k
            --vlen;
152
11.2k
        }
153
154
1.39k
        e = APR_BUCKET_NEXT(e);
155
1.39k
    } while (!eol);
156
1.38k
    v->dlen = dest - v->data;
157
1.38k
    *dest++ = 0;
158
159
    /* write name */
160
1.38k
    v->name = dest;
161
2.78k
    for (i = 0; i < arr.nelts; ++i) {
162
1.39k
        iov = &((struct iovec *)arr.elts)[i];
163
1.39k
        memcpy(dest, iov->iov_base, iov->iov_len);
164
1.39k
        dest += iov->iov_len;
165
1.39k
        ++iov;
166
1.39k
    }
167
1.38k
    v->nlen = dest - v->name;
168
1.38k
    *dest = 0;
169
170
5.56k
    while ((f = APR_BRIGADE_FIRST(bb)) != e)
171
4.18k
        apr_bucket_delete(f);
172
173
1.38k
    apreq_param_tainted_on(param);
174
1.38k
    *p = param;
175
1.38k
    return APR_SUCCESS;
176
177
1.38k
}
178
179
10.7k
#define IS_TOKEN_CHAR(c) (apr_isalnum(c) \
180
10.7k
                          || ((c) && strchr("!#$%&'*+-.^_`|~", (c))))
181
182
APREQ_DECLARE_PARSER(apreq_parse_headers)
183
961
{
184
961
    apr_pool_t *pool = parser->pool;
185
961
    apr_bucket *e;
186
961
    struct hdr_ctx *ctx;
187
961
    char ch;
188
189
961
    if (parser->ctx == NULL) {
190
961
        ctx = apr_pcalloc(pool, sizeof *ctx);
191
961
        ctx->bb = apr_brigade_create(pool, parser->bucket_alloc);
192
961
        parser->ctx = ctx;
193
961
        ctx->status = HDR_NAME;
194
961
    }
195
0
    else
196
0
        ctx = parser->ctx;
197
198
961
    PARSER_STATUS_CHECK(HDR);
199
961
    e = APR_BRIGADE_LAST(ctx->bb);
200
961
    APR_BRIGADE_CONCAT(ctx->bb, bb);
201
202
    /* parse the brigade for CRLF_CRLF-terminated header block,
203
     * each time starting from the front of the brigade.
204
     */
205
206
961
    for (e = APR_BUCKET_NEXT(e);
207
1.05k
         e != APR_BRIGADE_SENTINEL(ctx->bb);
208
961
         e = APR_BUCKET_NEXT(e))
209
979
    {
210
979
        apr_size_t off = 0, dlen;
211
979
        const char *data;
212
979
        apr_status_t s;
213
979
        apreq_param_t *param = NULL; /* silences gcc-4.0 warning */
214
215
979
        if (APR_BUCKET_IS_EOS(e)) {
216
0
            ctx->status = HDR_COMPLETE;
217
0
            APR_BRIGADE_CONCAT(bb, ctx->bb);
218
0
            return APR_SUCCESS;
219
0
        }
220
221
979
        s = apr_bucket_read(e, &data, &dlen, APR_BLOCK_READ);
222
979
        if ( s != APR_SUCCESS ) {
223
0
            ctx->status = HDR_ERROR;
224
0
            return s;
225
0
        }
226
979
        if (dlen == 0)
227
0
            continue;
228
229
6.70k
    parse_hdr_bucket:
230
231
        /*              gap           nlen = 13
232
         *              vvv           glen =  3
233
         * Sample-Header:  grape      vlen =  5
234
         * ^^^^^^^^^^^^^   ^^^^^
235
         *     name        value
236
         */
237
238
6.70k
        switch (ctx->status) {
239
240
1.52k
        case HDR_NAME:
241
242
11.6k
            while (off < dlen) {
243
11.6k
                ch = data[off++];
244
11.6k
                switch (ch) {
245
1.46k
                case ':':
246
1.46k
                    if (!ctx->nlen) {
247
7
                        ctx->status = HDR_ERROR;
248
7
                        return APREQ_ERROR_BADHEADER;
249
7
                    }
250
1.45k
                    e = split_header_line(e, &off, &data, &dlen);
251
1.45k
                    ++ctx->glen;
252
1.45k
                    ctx->status = HDR_GAP;
253
1.45k
                    goto parse_hdr_bucket;
254
255
10.1k
                default:
256
10.1k
                    if (!IS_TOKEN_CHAR(ch)) {
257
34
                        ctx->status = HDR_ERROR;
258
34
                        return APREQ_ERROR_BADCHAR;
259
34
                    }
260
10.1k
                    ++ctx->nlen;
261
11.6k
                }
262
11.6k
            }
263
31
            break;
264
265
266
1.46k
        case HDR_GAP:
267
268
2.12k
            while (off < dlen) {
269
2.10k
                ch = data[off++];
270
2.10k
                switch (ch) {
271
344
                case ' ':
272
657
                case '\t':
273
657
                    ++ctx->glen;
274
657
                    break;
275
276
532
                case '\n':
277
532
                    e = split_header_line(e, &off, &data, &dlen);
278
532
                    ctx->status = HDR_NEWLINE;
279
532
                    goto parse_hdr_bucket;
280
281
80
                case '\r':
282
80
                    e = split_header_line(e, &off, &data, &dlen);
283
80
                    ctx->status = HDR_ENDLINE;
284
80
                    goto parse_hdr_bucket;
285
286
833
                default:
287
833
                    if (apr_iscntrl(ch)) {
288
2
                        ctx->status = HDR_ERROR;
289
2
                        return APREQ_ERROR_BADCHAR;
290
2
                    }
291
831
                    e = split_header_line(e, &off, &data, &dlen);
292
831
                    ++ctx->vlen;
293
831
                    ctx->status = HDR_VALUE;
294
831
                    goto parse_hdr_bucket;
295
2.10k
                }
296
2.10k
            }
297
21
            break;
298
299
300
959
        case HDR_VALUE:
301
302
11.2k
            while (off < dlen) {
303
11.2k
                ch = data[off++];
304
11.2k
                switch (ch) {
305
757
                case '\n':
306
757
                    ctx->status = HDR_NEWLINE;
307
757
                    goto parse_hdr_bucket;
308
309
175
                case '\r':
310
175
                    ctx->status = HDR_ENDLINE;
311
175
                    goto parse_hdr_bucket;
312
313
10.3k
                default:
314
10.3k
                    if (apr_iscntrl(ch)) {
315
9
                        ctx->status = HDR_ERROR;
316
9
                        return APREQ_ERROR_BADCHAR;
317
9
                    }
318
10.3k
                    ++ctx->vlen;
319
11.2k
                }
320
11.2k
            }
321
18
            break;
322
323
324
316
        case HDR_ENDLINE:
325
494
        case HDR_LASTLINE:
326
327
494
            if (off == dlen)
328
10
                break;
329
330
484
            if (data[off++] != '\n') {
331
12
                ctx->status = HDR_ERROR;
332
12
                return APREQ_ERROR_BADHEADER;
333
12
            }
334
335
472
            if (ctx->status == HDR_LASTLINE) {
336
174
                ctx->status = HDR_COMPLETE;
337
174
                goto parse_hdr_bucket;
338
174
            }
339
340
            /* fall thru */
341
298
            ctx->status = HDR_NEWLINE;
342
343
1.74k
        case HDR_NEWLINE:
344
345
1.74k
            if (off == dlen)
346
11
                break;
347
348
1.73k
            ch = data[off++];
349
1.73k
            switch (ch) {
350
230
            case ' ':
351
346
            case '\t':
352
346
                ++ctx->vlen;
353
346
                break;
354
355
1.38k
            default:
356
1.38k
                e = split_header_line(e, &off, &data, &dlen);
357
358
                /* consume from splitted brigade now */
359
1.38k
                s = consume_header_line(&param, pool, ctx->bb,
360
1.38k
                                        ctx->nlen, ctx->glen, ctx->vlen);
361
1.38k
                if (parser->hook != NULL && s == APR_SUCCESS)
362
0
                    s = apreq_hook_run(parser->hook, param, NULL);
363
1.38k
                if (s != APR_SUCCESS) {
364
0
                    ctx->status = HDR_ERROR;
365
0
                    return s;
366
0
                }
367
1.38k
                apreq_value_table_add(&param->v, t);
368
1.38k
                ctx->nlen = 0;
369
1.38k
                ctx->vlen = 0;
370
1.38k
                ctx->glen = 0;
371
372
1.38k
                switch (ch) {
373
634
                case '\n':
374
634
                    ctx->status = HDR_COMPLETE;
375
634
                    break;
376
377
177
                case '\r':
378
177
                    ctx->status = HDR_LASTLINE;
379
177
                    break;
380
381
576
                default:
382
576
                    if (!IS_TOKEN_CHAR(ch)) {
383
8
                        ctx->status = HDR_ERROR;
384
8
                        return APREQ_ERROR_BADCHAR;
385
8
                    }
386
568
                    ++ctx->nlen;
387
568
                    ctx->status = HDR_NAME;
388
568
                    break;
389
1.38k
                }
390
1.37k
                goto parse_hdr_bucket;
391
1.73k
            }
392
393
            /* fall thru */
394
346
            ctx->status = HDR_FOLDLINE;
395
396
349
        case HDR_FOLDLINE:
397
398
734
            while (off < dlen) {
399
727
                ch = data[off++];
400
727
                switch (ch) {
401
260
                case ' ':
402
385
                case '\t':
403
385
                    ++ctx->vlen;
404
385
                    break;
405
406
156
                case '\n':
407
156
                    ctx->status = HDR_NEWLINE;
408
156
                    goto parse_hdr_bucket;
409
410
60
                case '\r':
411
60
                    ctx->status = HDR_ENDLINE;
412
60
                    goto parse_hdr_bucket;
413
414
126
                default:
415
126
                    if (apr_iscntrl(ch)) {
416
1
                        ctx->status = HDR_ERROR;
417
1
                        return APREQ_ERROR_BADCHAR;
418
1
                    }
419
125
                    ctx->status = HDR_VALUE;
420
125
                    ++ctx->vlen;
421
125
                    goto parse_hdr_bucket;
422
727
                }
423
727
            }
424
7
            break;
425
426
427
808
        case HDR_COMPLETE:
428
429
808
            if (off < dlen)
430
803
                apr_bucket_split(e, off);
431
432
808
            e = APR_BUCKET_NEXT(e);
433
809
            do {
434
809
                apr_bucket *f = APR_BRIGADE_FIRST(ctx->bb);
435
809
                apr_bucket_delete(f);
436
809
            } while (e != APR_BRIGADE_FIRST(ctx->bb));
437
438
808
            APR_BRIGADE_CONCAT(bb, ctx->bb);
439
808
            return APR_SUCCESS;
440
441
442
0
        default:
443
0
            assert(0); /* not reached */
444
6.70k
        }
445
6.70k
    }
446
80
    apreq_brigade_setaside(ctx->bb,pool);
447
80
    return APR_INCOMPLETE;
448
961
}