/src/curl/lib/dynhds.c

Source
/***************************************************************************
 *                                  _   _ ____  _
 *  Project                     ___| | | |  _ \| |
 *                             / __| | | | |_) | |
 *                            | (__| |_| |  _ <| |___
 *                             \___|\___/|_| \_\_____|
 *
 * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
 *
 * This software is licensed as described in the file COPYING, which
 * you should have received as part of this distribution. The terms
 * are also available at https://curl.se/docs/copyright.html.
 *
 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
 * copies of the Software, and permit persons to whom the Software is
 * furnished to do so, under the terms of the COPYING file.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 * SPDX-License-Identifier: curl
 *
 ***************************************************************************/
#include "curl_setup.h"

#include "dynhds.h"
#include "strcase.h"

#ifdef USE_NGHTTP2
#include <stdint.h>
#include <nghttp2/nghttp2.h>
#endif /* USE_NGHTTP2 */

static struct dynhds_entry *entry_new(const char *name, size_t namelen,
                                      const char *value, size_t valuelen,
                                      int opts)
{
  struct dynhds_entry *e;
  char *p;

  DEBUGASSERT(name);
  DEBUGASSERT(value);
  e = curlx_calloc(1, sizeof(*e) + namelen + valuelen + 2);
  if(!e)
    return NULL;
  e->name = p = (char *)e + sizeof(*e);
  memcpy(p, name, namelen);
  e->namelen = namelen;
  e->value = p += namelen + 1; /* leave a \0 at the end of name */
  memcpy(p, value, valuelen);
  e->valuelen = valuelen;
  if(opts & DYNHDS_OPT_LOWERCASE)
    Curl_strntolower(e->name, e->name, e->namelen);
  return e;
}

static void entry_free(struct dynhds_entry *e)
{
  curlx_free(e);
}

void Curl_dynhds_init(struct dynhds *dynhds, size_t max_entries,
                      size_t max_strs_size)
{
  DEBUGASSERT(dynhds);
  DEBUGASSERT(max_strs_size);
  dynhds->hds = NULL;
  dynhds->hds_len = dynhds->hds_allc = dynhds->strs_len = 0;
  dynhds->max_entries = max_entries;
  dynhds->max_strs_size = max_strs_size;
  dynhds->opts = 0;
}

void Curl_dynhds_free(struct dynhds *dynhds)
{
  DEBUGASSERT(dynhds);
  if(dynhds->hds && dynhds->hds_len) {
    size_t i;
    DEBUGASSERT(dynhds->hds);
    for(i = 0; i < dynhds->hds_len; ++i) {
      entry_free(dynhds->hds[i]);
    }
  }
  Curl_safefree(dynhds->hds);
  dynhds->hds_len = dynhds->hds_allc = dynhds->strs_len = 0;
}

void Curl_dynhds_reset(struct dynhds *dynhds)
{
  DEBUGASSERT(dynhds);
  if(dynhds->hds_len) {
    size_t i;
    DEBUGASSERT(dynhds->hds);
    for(i = 0; i < dynhds->hds_len; ++i) {
      entry_free(dynhds->hds[i]);
      dynhds->hds[i] = NULL;
    }
  }
  dynhds->hds_len = dynhds->strs_len = 0;
}

size_t Curl_dynhds_count(struct dynhds *dynhds)
{
  return dynhds->hds_len;
}

void Curl_dynhds_set_opts(struct dynhds *dynhds, int opts)
{
  dynhds->opts = opts;
}

struct dynhds_entry *Curl_dynhds_getn(struct dynhds *dynhds, size_t n)
{
  DEBUGASSERT(dynhds);
  return (n < dynhds->hds_len) ? dynhds->hds[n] : NULL;
}

struct dynhds_entry *Curl_dynhds_get(struct dynhds *dynhds, const char *name,
                                     size_t namelen)
{
  size_t i;
  for(i = 0; i < dynhds->hds_len; ++i) {
    if(dynhds->hds[i]->namelen == namelen &&
       curl_strnequal(dynhds->hds[i]->name, name, namelen)) {
      return dynhds->hds[i];
    }
  }
  return NULL;
}

struct dynhds_entry *Curl_dynhds_cget(struct dynhds *dynhds, const char *name)
{
  return Curl_dynhds_get(dynhds, name, strlen(name));
}

CURLcode Curl_dynhds_add(struct dynhds *dynhds,
                         const char *name, size_t namelen,
                         const char *value, size_t valuelen)
{
  struct dynhds_entry *entry = NULL;
  CURLcode result = CURLE_OUT_OF_MEMORY;

  DEBUGASSERT(dynhds);
  if(dynhds->max_entries && dynhds->hds_len >= dynhds->max_entries)
    return CURLE_OUT_OF_MEMORY;
  if(dynhds->strs_len + namelen + valuelen > dynhds->max_strs_size)
    return CURLE_OUT_OF_MEMORY;

  entry = entry_new(name, namelen, value, valuelen, dynhds->opts);
  if(!entry)
    goto out;

  if(dynhds->hds_len + 1 >= dynhds->hds_allc) {
    size_t nallc = dynhds->hds_len + 16;
    struct dynhds_entry **nhds;

    if(dynhds->max_entries && nallc > dynhds->max_entries)
      nallc = dynhds->max_entries;

    nhds = curlx_calloc(nallc, sizeof(struct dynhds_entry *));
    if(!nhds)
      goto out;
    if(dynhds->hds) {
      memcpy(nhds, dynhds->hds,
             dynhds->hds_len * sizeof(struct dynhds_entry *));
      Curl_safefree(dynhds->hds);
    }
    dynhds->hds = nhds;
    dynhds->hds_allc = nallc;
  }
  dynhds->hds[dynhds->hds_len++] = entry;
  entry = NULL;
  dynhds->strs_len += namelen + valuelen;
  result = CURLE_OK;

out:
  if(entry)
    entry_free(entry);
  return result;
}

CURLcode Curl_dynhds_cadd(struct dynhds *dynhds,
                          const char *name, const char *value)
{
  return Curl_dynhds_add(dynhds, name, strlen(name), value, strlen(value));
}

CURLcode Curl_dynhds_h1_add_line(struct dynhds *dynhds,
                                 const char *line, size_t line_len)
{
  const char *p;
  const char *name;
  size_t namelen;
  const char *value;
  size_t valuelen, i;

  if(!line || !line_len)
    return CURLE_OK;

  p = memchr(line, ':', line_len);
  if(!p)
    return CURLE_BAD_FUNCTION_ARGUMENT;
  name = line;
  namelen = p - line;
  p++; /* move past the colon */
  for(i = namelen + 1; i < line_len; ++i, ++p) {
    if(!ISBLANK(*p))
      break;
  }
  value = p;
  valuelen = line_len - i;

  p = memchr(value, '\r', valuelen);
  if(!p)
    p = memchr(value, '\n', valuelen);
  if(p)
    valuelen = (size_t)(p - value);

  return Curl_dynhds_add(dynhds, name, namelen, value, valuelen);
}

CURLcode Curl_dynhds_h1_cadd_line(struct dynhds *dynhds, const char *line)
{
  return Curl_dynhds_h1_add_line(dynhds, line, line ? strlen(line) : 0);
}

#ifdef UNITTESTS
/* used by unit2602.c */

bool Curl_dynhds_contains(struct dynhds *dynhds,
                          const char *name, size_t namelen)
{
  return !!Curl_dynhds_get(dynhds, name, namelen);
}

bool Curl_dynhds_ccontains(struct dynhds *dynhds, const char *name)
{
  return Curl_dynhds_contains(dynhds, name, strlen(name));
}

size_t Curl_dynhds_count_name(struct dynhds *dynhds,
                              const char *name, size_t namelen)
{
  size_t n = 0;
  if(dynhds->hds_len) {
    size_t i;
    for(i = 0; i < dynhds->hds_len; ++i) {
      if((namelen == dynhds->hds[i]->namelen) &&
         curl_strnequal(name, dynhds->hds[i]->name, namelen))
        ++n;
    }
  }
  return n;
}

size_t Curl_dynhds_ccount_name(struct dynhds *dynhds, const char *name)
{
  return Curl_dynhds_count_name(dynhds, name, strlen(name));
}

CURLcode Curl_dynhds_set(struct dynhds *dynhds,
                         const char *name, size_t namelen,
                         const char *value, size_t valuelen)
{
  Curl_dynhds_remove(dynhds, name, namelen);
  return Curl_dynhds_add(dynhds, name, namelen, value, valuelen);
}

size_t Curl_dynhds_remove(struct dynhds *dynhds,
                          const char *name, size_t namelen)
{
  size_t n = 0;
  if(dynhds->hds_len) {
    size_t i, len;
    for(i = 0; i < dynhds->hds_len; ++i) {
      if((namelen == dynhds->hds[i]->namelen) &&
         curl_strnequal(name, dynhds->hds[i]->name, namelen)) {
        ++n;
        --dynhds->hds_len;
        dynhds->strs_len -= (dynhds->hds[i]->namelen +
                             dynhds->hds[i]->valuelen);
        entry_free(dynhds->hds[i]);
        len = dynhds->hds_len - i; /* remaining entries */
        if(len) {
          memmove(&dynhds->hds[i], &dynhds->hds[i + 1],
                  len * sizeof(dynhds->hds[i]));
        }
        --i; /* do this index again */
      }
    }
  }
  return n;
}

size_t Curl_dynhds_cremove(struct dynhds *dynhds, const char *name)
{
  return Curl_dynhds_remove(dynhds, name, strlen(name));
}

#endif

CURLcode Curl_dynhds_h1_dprint(struct dynhds *dynhds, struct dynbuf *dbuf)
{
  CURLcode result = CURLE_OK;
  size_t i;

  if(!dynhds->hds_len)
    return result;

  for(i = 0; i < dynhds->hds_len; ++i) {
    result = curlx_dyn_addf(dbuf, "%.*s: %.*s\r\n",
                            (int)dynhds->hds[i]->namelen, dynhds->hds[i]->name,
                            (int)dynhds->hds[i]->valuelen,
                            dynhds->hds[i]->value);
    if(result)
      break;
  }

  return result;
}

#ifdef USE_NGHTTP2

nghttp2_nv *Curl_dynhds_to_nva(struct dynhds *dynhds, size_t *pcount)
{
  nghttp2_nv *nva = curlx_calloc(1, sizeof(nghttp2_nv) * dynhds->hds_len);
  size_t i;

  *pcount = 0;
  if(!nva)
    return NULL;

  for(i = 0; i < dynhds->hds_len; ++i) {
    struct dynhds_entry *e = dynhds->hds[i];
    DEBUGASSERT(e);
    nva[i].name = (unsigned char *)e->name;
    nva[i].namelen = e->namelen;
    nva[i].value = (unsigned char *)e->value;
    nva[i].valuelen = e->valuelen;
    nva[i].flags = NGHTTP2_NV_FLAG_NONE;
  }
  *pcount = dynhds->hds_len;
  return nva;
}

#endif /* USE_NGHTTP2 */

Coverage Report

Created: 2026-01-10 06:51

Line	Count	Source
1		/***************************************************************************
2		* _ _ ____ _
3		* Project ___\| \| \| \| _ \\| \|
4		* / __\| \| \| \| \|_) \| \|
5		* \| (__\| \|_\| \| _ <\| \|___
6		* \___\|\___/\|_\| \_\_____\|
7		*
8		* Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
9		*
10		* This software is licensed as described in the file COPYING, which
11		* you should have received as part of this distribution. The terms
12		* are also available at https://curl.se/docs/copyright.html.
13		*
14		* You may opt to use, copy, modify, merge, publish, distribute and/or sell
15		* copies of the Software, and permit persons to whom the Software is
16		* furnished to do so, under the terms of the COPYING file.
17		*
18		* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19		* KIND, either express or implied.
20		*
21		* SPDX-License-Identifier: curl
22		*
23		***************************************************************************/
24		#include "curl_setup.h"
25
26		#include "dynhds.h"
27		#include "strcase.h"
28
29		#ifdef USE_NGHTTP2
30		#include <stdint.h>
31		#include <nghttp2/nghttp2.h>
32		#endif /* USE_NGHTTP2 */
33
34		static struct dynhds_entry entry_new(const char name, size_t namelen,
35		const char *value, size_t valuelen,
36		int opts)
37	0	{
38	0	struct dynhds_entry *e;
39	0	char *p;
40
41	0	DEBUGASSERT(name);
42	0	DEBUGASSERT(value);
43	0	e = curlx_calloc(1, sizeof(*e) + namelen + valuelen + 2);
44	0	if(!e)
45	0	return NULL;
46	0	e->name = p = (char )e + sizeof(e);
47	0	memcpy(p, name, namelen);
48	0	e->namelen = namelen;
49	0	e->value = p += namelen + 1; /* leave a \0 at the end of name */
50	0	memcpy(p, value, valuelen);
51	0	e->valuelen = valuelen;
52	0	if(opts & DYNHDS_OPT_LOWERCASE)
53	0	Curl_strntolower(e->name, e->name, e->namelen);
54	0	return e;
55	0	}
56
57		static void entry_free(struct dynhds_entry *e)
58	0	{
59	0	curlx_free(e);
60	0	}
61
62		void Curl_dynhds_init(struct dynhds *dynhds, size_t max_entries,
63		size_t max_strs_size)
64	0	{
65	0	DEBUGASSERT(dynhds);
66	0	DEBUGASSERT(max_strs_size);
67	0	dynhds->hds = NULL;
68	0	dynhds->hds_len = dynhds->hds_allc = dynhds->strs_len = 0;
69	0	dynhds->max_entries = max_entries;
70	0	dynhds->max_strs_size = max_strs_size;
71	0	dynhds->opts = 0;
72	0	}
73
74		void Curl_dynhds_free(struct dynhds *dynhds)
75	0	{
76	0	DEBUGASSERT(dynhds);
77	0	if(dynhds->hds && dynhds->hds_len) {
78	0	size_t i;
79	0	DEBUGASSERT(dynhds->hds);
80	0	for(i = 0; i < dynhds->hds_len; ++i) {
81	0	entry_free(dynhds->hds[i]);
82	0	}
83	0	}
84	0	Curl_safefree(dynhds->hds);
85	0	dynhds->hds_len = dynhds->hds_allc = dynhds->strs_len = 0;
86	0	}
87
88		void Curl_dynhds_reset(struct dynhds *dynhds)
89	0	{
90	0	DEBUGASSERT(dynhds);
91	0	if(dynhds->hds_len) {
92	0	size_t i;
93	0	DEBUGASSERT(dynhds->hds);
94	0	for(i = 0; i < dynhds->hds_len; ++i) {
95	0	entry_free(dynhds->hds[i]);
96	0	dynhds->hds[i] = NULL;
97	0	}
98	0	}
99	0	dynhds->hds_len = dynhds->strs_len = 0;
100	0	}
101
102		size_t Curl_dynhds_count(struct dynhds *dynhds)
103	0	{
104	0	return dynhds->hds_len;
105	0	}
106
107		void Curl_dynhds_set_opts(struct dynhds *dynhds, int opts)
108	0	{
109	0	dynhds->opts = opts;
110	0	}
111
112		struct dynhds_entry Curl_dynhds_getn(struct dynhds dynhds, size_t n)
113	0	{
114	0	DEBUGASSERT(dynhds);
115	0	return (n < dynhds->hds_len) ? dynhds->hds[n] : NULL;
116	0	}
117
118		struct dynhds_entry Curl_dynhds_get(struct dynhds dynhds, const char *name,
119		size_t namelen)
120	0	{
121	0	size_t i;
122	0	for(i = 0; i < dynhds->hds_len; ++i) {
123	0	if(dynhds->hds[i]->namelen == namelen &&
124	0	curl_strnequal(dynhds->hds[i]->name, name, namelen)) {
125	0	return dynhds->hds[i];
126	0	}
127	0	}
128	0	return NULL;
129	0	}
130
131		struct dynhds_entry Curl_dynhds_cget(struct dynhds dynhds, const char *name)
132	0	{
133	0	return Curl_dynhds_get(dynhds, name, strlen(name));
134	0	}
135
136		CURLcode Curl_dynhds_add(struct dynhds *dynhds,
137		const char *name, size_t namelen,
138		const char *value, size_t valuelen)
139	0	{
140	0	struct dynhds_entry *entry = NULL;
141	0	CURLcode result = CURLE_OUT_OF_MEMORY;
142
143	0	DEBUGASSERT(dynhds);
144	0	if(dynhds->max_entries && dynhds->hds_len >= dynhds->max_entries)
145	0	return CURLE_OUT_OF_MEMORY;
146	0	if(dynhds->strs_len + namelen + valuelen > dynhds->max_strs_size)
147	0	return CURLE_OUT_OF_MEMORY;
148
149	0	entry = entry_new(name, namelen, value, valuelen, dynhds->opts);
150	0	if(!entry)
151	0	goto out;
152
153	0	if(dynhds->hds_len + 1 >= dynhds->hds_allc) {
154	0	size_t nallc = dynhds->hds_len + 16;
155	0	struct dynhds_entry **nhds;
156
157	0	if(dynhds->max_entries && nallc > dynhds->max_entries)
158	0	nallc = dynhds->max_entries;
159
160	0	nhds = curlx_calloc(nallc, sizeof(struct dynhds_entry *));
161	0	if(!nhds)
162	0	goto out;
163	0	if(dynhds->hds) {
164	0	memcpy(nhds, dynhds->hds,
165	0	dynhds->hds_len * sizeof(struct dynhds_entry *));
166	0	Curl_safefree(dynhds->hds);
167	0	}
168	0	dynhds->hds = nhds;
169	0	dynhds->hds_allc = nallc;
170	0	}
171	0	dynhds->hds[dynhds->hds_len++] = entry;
172	0	entry = NULL;
173	0	dynhds->strs_len += namelen + valuelen;
174	0	result = CURLE_OK;
175
176	0	out:
177	0	if(entry)
178	0	entry_free(entry);
179	0	return result;
180	0	}
181
182		CURLcode Curl_dynhds_cadd(struct dynhds *dynhds,
183		const char name, const char value)
184	0	{
185	0	return Curl_dynhds_add(dynhds, name, strlen(name), value, strlen(value));
186	0	}
187
188		CURLcode Curl_dynhds_h1_add_line(struct dynhds *dynhds,
189		const char *line, size_t line_len)
190	0	{
191	0	const char *p;
192	0	const char *name;
193	0	size_t namelen;
194	0	const char *value;
195	0	size_t valuelen, i;
196
197	0	if(!line \|\| !line_len)
198	0	return CURLE_OK;
199
200	0	p = memchr(line, ':', line_len);
201	0	if(!p)
202	0	return CURLE_BAD_FUNCTION_ARGUMENT;
203	0	name = line;
204	0	namelen = p - line;
205	0	p++; /* move past the colon */
206	0	for(i = namelen + 1; i < line_len; ++i, ++p) {
207	0	if(!ISBLANK(*p))
208	0	break;
209	0	}
210	0	value = p;
211	0	valuelen = line_len - i;
212
213	0	p = memchr(value, '\r', valuelen);
214	0	if(!p)
215	0	p = memchr(value, '\n', valuelen);
216	0	if(p)
217	0	valuelen = (size_t)(p - value);
218
219	0	return Curl_dynhds_add(dynhds, name, namelen, value, valuelen);
220	0	}
221
222		CURLcode Curl_dynhds_h1_cadd_line(struct dynhds dynhds, const char line)
223	0	{
224	0	return Curl_dynhds_h1_add_line(dynhds, line, line ? strlen(line) : 0);
225	0	}
226
227		#ifdef UNITTESTS
228		/* used by unit2602.c */
229
230		bool Curl_dynhds_contains(struct dynhds *dynhds,
231		const char *name, size_t namelen)
232		{
233		return !!Curl_dynhds_get(dynhds, name, namelen);
234		}
235
236		bool Curl_dynhds_ccontains(struct dynhds dynhds, const char name)
237		{
238		return Curl_dynhds_contains(dynhds, name, strlen(name));
239		}
240
241		size_t Curl_dynhds_count_name(struct dynhds *dynhds,
242		const char *name, size_t namelen)
243		{
244		size_t n = 0;
245		if(dynhds->hds_len) {
246		size_t i;
247		for(i = 0; i < dynhds->hds_len; ++i) {
248		if((namelen == dynhds->hds[i]->namelen) &&
249		curl_strnequal(name, dynhds->hds[i]->name, namelen))
250		++n;
251		}
252		}
253		return n;
254		}
255
256		size_t Curl_dynhds_ccount_name(struct dynhds dynhds, const char name)
257		{
258		return Curl_dynhds_count_name(dynhds, name, strlen(name));
259		}
260
261		CURLcode Curl_dynhds_set(struct dynhds *dynhds,
262		const char *name, size_t namelen,
263		const char *value, size_t valuelen)
264		{
265		Curl_dynhds_remove(dynhds, name, namelen);
266		return Curl_dynhds_add(dynhds, name, namelen, value, valuelen);
267		}
268
269		size_t Curl_dynhds_remove(struct dynhds *dynhds,
270		const char *name, size_t namelen)
271		{
272		size_t n = 0;
273		if(dynhds->hds_len) {
274		size_t i, len;
275		for(i = 0; i < dynhds->hds_len; ++i) {
276		if((namelen == dynhds->hds[i]->namelen) &&
277		curl_strnequal(name, dynhds->hds[i]->name, namelen)) {
278		++n;
279		--dynhds->hds_len;
280		dynhds->strs_len -= (dynhds->hds[i]->namelen +
281		dynhds->hds[i]->valuelen);
282		entry_free(dynhds->hds[i]);
283		len = dynhds->hds_len - i; /* remaining entries */
284		if(len) {
285		memmove(&dynhds->hds[i], &dynhds->hds[i + 1],
286		len * sizeof(dynhds->hds[i]));
287		}
288		--i; /* do this index again */
289		}
290		}
291		}
292		return n;
293		}
294
295		size_t Curl_dynhds_cremove(struct dynhds dynhds, const char name)
296		{
297		return Curl_dynhds_remove(dynhds, name, strlen(name));
298		}
299
300		#endif
301
302		CURLcode Curl_dynhds_h1_dprint(struct dynhds dynhds, struct dynbuf dbuf)
303	0	{
304	0	CURLcode result = CURLE_OK;
305	0	size_t i;
306
307	0	if(!dynhds->hds_len)
308	0	return result;
309
310	0	for(i = 0; i < dynhds->hds_len; ++i) {
311	0	result = curlx_dyn_addf(dbuf, "%.s: %.s\r\n",
312	0	(int)dynhds->hds[i]->namelen, dynhds->hds[i]->name,
313	0	(int)dynhds->hds[i]->valuelen,
314	0	dynhds->hds[i]->value);
315	0	if(result)
316	0	break;
317	0	}
318
319	0	return result;
320	0	}
321
322		#ifdef USE_NGHTTP2
323
324		nghttp2_nv Curl_dynhds_to_nva(struct dynhds dynhds, size_t *pcount)
325	0	{
326	0	nghttp2_nv nva = curlx_calloc(1, sizeof(nghttp2_nv) dynhds->hds_len);
327	0	size_t i;
328
329	0	*pcount = 0;
330	0	if(!nva)
331	0	return NULL;
332
333	0	for(i = 0; i < dynhds->hds_len; ++i) {
334	0	struct dynhds_entry *e = dynhds->hds[i];
335	0	DEBUGASSERT(e);
336	0	nva[i].name = (unsigned char *)e->name;
337	0	nva[i].namelen = e->namelen;
338	0	nva[i].value = (unsigned char *)e->value;
339	0	nva[i].valuelen = e->valuelen;
340	0	nva[i].flags = NGHTTP2_NV_FLAG_NONE;
341	0	}
342	0	*pcount = dynhds->hds_len;
343	0	return nva;
344	0	}
345
346		#endif /* USE_NGHTTP2 */