| 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 |  | 
|---|
| 25 | #include "curl_setup.h" | 
|---|
| 26 | #include "urldata.h" | 
|---|
| 27 | #include "bufref.h" | 
|---|
| 28 |  | 
|---|
| 29 | #include "curl_memory.h" | 
|---|
| 30 | #include "memdebug.h" | 
|---|
| 31 |  | 
|---|
| 32 | #define SIGNATURE 0x5c48e9b2    /* Random pattern. */ | 
|---|
| 33 |  | 
|---|
| 34 | /* | 
|---|
| 35 | * Init a bufref struct. | 
|---|
| 36 | */ | 
|---|
| 37 | void Curl_bufref_init(struct bufref *br) | 
|---|
| 38 | { | 
|---|
| 39 | DEBUGASSERT(br); | 
|---|
| 40 | br->dtor = NULL; | 
|---|
| 41 | br->ptr = NULL; | 
|---|
| 42 | br->len = 0; | 
|---|
| 43 |  | 
|---|
| 44 | #ifdef DEBUGBUILD | 
|---|
| 45 | br->signature = SIGNATURE; | 
|---|
| 46 | #endif | 
|---|
| 47 | } | 
|---|
| 48 |  | 
|---|
| 49 | /* | 
|---|
| 50 | * Free the buffer and re-init the necessary fields. It doesn't touch the | 
|---|
| 51 | * 'signature' field and thus this buffer reference can be reused. | 
|---|
| 52 | */ | 
|---|
| 53 |  | 
|---|
| 54 | void Curl_bufref_free(struct bufref *br) | 
|---|
| 55 | { | 
|---|
| 56 | DEBUGASSERT(br); | 
|---|
| 57 | DEBUGASSERT(br->signature == SIGNATURE); | 
|---|
| 58 | DEBUGASSERT(br->ptr || !br->len); | 
|---|
| 59 |  | 
|---|
| 60 | if(br->ptr && br->dtor) | 
|---|
| 61 | br->dtor((void *) br->ptr); | 
|---|
| 62 |  | 
|---|
| 63 | br->dtor = NULL; | 
|---|
| 64 | br->ptr = NULL; | 
|---|
| 65 | br->len = 0; | 
|---|
| 66 | } | 
|---|
| 67 |  | 
|---|
| 68 | /* | 
|---|
| 69 | * Set the buffer reference to new values. The previously referenced buffer | 
|---|
| 70 | * is released before assignment. | 
|---|
| 71 | */ | 
|---|
| 72 | void Curl_bufref_set(struct bufref *br, const void *ptr, size_t len, | 
|---|
| 73 | void (*dtor)(void *)) | 
|---|
| 74 | { | 
|---|
| 75 | DEBUGASSERT(ptr || !len); | 
|---|
| 76 | DEBUGASSERT(len <= CURL_MAX_INPUT_LENGTH); | 
|---|
| 77 |  | 
|---|
| 78 | Curl_bufref_free(br); | 
|---|
| 79 | br->ptr = (const unsigned char *) ptr; | 
|---|
| 80 | br->len = len; | 
|---|
| 81 | br->dtor = dtor; | 
|---|
| 82 | } | 
|---|
| 83 |  | 
|---|
| 84 | /* | 
|---|
| 85 | * Get a pointer to the referenced buffer. | 
|---|
| 86 | */ | 
|---|
| 87 | const unsigned char *Curl_bufref_ptr(const struct bufref *br) | 
|---|
| 88 | { | 
|---|
| 89 | DEBUGASSERT(br); | 
|---|
| 90 | DEBUGASSERT(br->signature == SIGNATURE); | 
|---|
| 91 | DEBUGASSERT(br->ptr || !br->len); | 
|---|
| 92 |  | 
|---|
| 93 | return br->ptr; | 
|---|
| 94 | } | 
|---|
| 95 |  | 
|---|
| 96 | /* | 
|---|
| 97 | * Get the length of the referenced buffer data. | 
|---|
| 98 | */ | 
|---|
| 99 | size_t Curl_bufref_len(const struct bufref *br) | 
|---|
| 100 | { | 
|---|
| 101 | DEBUGASSERT(br); | 
|---|
| 102 | DEBUGASSERT(br->signature == SIGNATURE); | 
|---|
| 103 | DEBUGASSERT(br->ptr || !br->len); | 
|---|
| 104 |  | 
|---|
| 105 | return br->len; | 
|---|
| 106 | } | 
|---|
| 107 |  | 
|---|
| 108 | CURLcode Curl_bufref_memdup(struct bufref *br, const void *ptr, size_t len) | 
|---|
| 109 | { | 
|---|
| 110 | unsigned char *cpy = NULL; | 
|---|
| 111 |  | 
|---|
| 112 | DEBUGASSERT(br); | 
|---|
| 113 | DEBUGASSERT(br->signature == SIGNATURE); | 
|---|
| 114 | DEBUGASSERT(br->ptr || !br->len); | 
|---|
| 115 | DEBUGASSERT(ptr || !len); | 
|---|
| 116 | DEBUGASSERT(len <= CURL_MAX_INPUT_LENGTH); | 
|---|
| 117 |  | 
|---|
| 118 | if(ptr) { | 
|---|
| 119 | cpy = malloc(len + 1); | 
|---|
| 120 | if(!cpy) | 
|---|
| 121 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 122 | if(len) | 
|---|
| 123 | memcpy(dest: cpy, src: ptr, n: len); | 
|---|
| 124 | cpy[len] = '\0'; | 
|---|
| 125 | } | 
|---|
| 126 |  | 
|---|
| 127 | Curl_bufref_set(br, ptr: cpy, len, dtor: curl_free); | 
|---|
| 128 | return CURLE_OK; | 
|---|
| 129 | } | 
|---|
| 130 |  | 
|---|