|
- /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
- * All rights reserved.
- *
- * This package is an SSL implementation written
- * by Eric Young (eay@cryptsoft.com).
- * The implementation was written so as to conform with Netscapes SSL.
- *
- * This library is free for commercial and non-commercial use as long as
- * the following conditions are aheared to. The following conditions
- * apply to all code found in this distribution, be it the RC4, RSA,
- * lhash, DES, etc., code; not just the SSL code. The SSL documentation
- * included with this distribution is covered by the same copyright terms
- * except that the holder is Tim Hudson (tjh@cryptsoft.com).
- *
- * Copyright remains Eric Young's, and as such any Copyright notices in
- * the code are not to be removed.
- * If this package is used in a product, Eric Young should be given attribution
- * as the author of the parts of the library used.
- * This can be in the form of a textual message at program startup or
- * in documentation (online or textual) provided with the package.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- * must display the following acknowledgement:
- * "This product includes cryptographic software written by
- * Eric Young (eay@cryptsoft.com)"
- * The word 'cryptographic' can be left out if the rouines from the library
- * being used are not cryptographic related :-).
- * 4. If you include any Windows specific code (or a derivative thereof) from
- * the apps directory (application code) you must include an acknowledgement:
- * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
- *
- * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- *
- * The licence and distribution terms for any publically available version or
- * derivative of this code cannot be changed. i.e. this code cannot simply be
- * copied and put under another distribution licence
- * [including the GNU Public Licence.] */
-
- #include <openssl/bio.h>
-
- #include <assert.h>
- #include <errno.h>
- #include <limits.h>
- #include <string.h>
-
- #include <openssl/err.h>
- #include <openssl/mem.h>
- #include <openssl/thread.h>
-
- #include "../internal.h"
-
-
- /* BIO_set initialises a BIO structure to have the given type and sets the
- * reference count to one. It returns one on success or zero on error. */
- static int bio_set(BIO *bio, const BIO_METHOD *method) {
- /* This function can be called with a stack allocated |BIO| so we have to
- * assume that the contents of |BIO| are arbitary. This also means that it'll
- * leak memory if you call |BIO_set| twice on the same BIO. */
- memset(bio, 0, sizeof(BIO));
-
- bio->method = method;
- bio->shutdown = 1;
- bio->references = 1;
-
- if (method->create != NULL && !method->create(bio)) {
- return 0;
- }
-
- return 1;
- }
-
- BIO *BIO_new(const BIO_METHOD *method) {
- BIO *ret = OPENSSL_malloc(sizeof(BIO));
- if (ret == NULL) {
- OPENSSL_PUT_ERROR(BIO, ERR_R_MALLOC_FAILURE);
- return NULL;
- }
-
- if (!bio_set(ret, method)) {
- OPENSSL_free(ret);
- ret = NULL;
- }
-
- return ret;
- }
-
- int BIO_free(BIO *bio) {
- BIO *next_bio;
-
- for (; bio != NULL; bio = next_bio) {
- if (!CRYPTO_refcount_dec_and_test_zero(&bio->references)) {
- return 0;
- }
-
- if (bio->callback != NULL) {
- int i = (int)bio->callback(bio, BIO_CB_FREE, NULL, 0, 0, 1);
- if (i <= 0) {
- return i;
- }
- }
-
- next_bio = BIO_pop(bio);
-
- if (bio->method != NULL && bio->method->destroy != NULL) {
- bio->method->destroy(bio);
- }
-
- OPENSSL_free(bio);
- }
- return 1;
- }
-
- BIO *BIO_up_ref(BIO *bio) {
- CRYPTO_refcount_inc(&bio->references);
- return bio;
- }
-
- void BIO_vfree(BIO *bio) {
- BIO_free(bio);
- }
-
- void BIO_free_all(BIO *bio) {
- BIO_free(bio);
- }
-
- static int bio_io(BIO *bio, void *buf, int len, size_t method_offset,
- int callback_flags, size_t *num) {
- int i;
- typedef int (*io_func_t)(BIO *, char *, int);
- io_func_t io_func = NULL;
-
- if (bio != NULL && bio->method != NULL) {
- io_func =
- *((const io_func_t *)(((const uint8_t *)bio->method) + method_offset));
- }
-
- if (io_func == NULL) {
- OPENSSL_PUT_ERROR(BIO, BIO_R_UNSUPPORTED_METHOD);
- return -2;
- }
-
- if (bio->callback != NULL) {
- i = (int) bio->callback(bio, callback_flags, buf, len, 0L, 1L);
- if (i <= 0) {
- return i;
- }
- }
-
- if (!bio->init) {
- OPENSSL_PUT_ERROR(BIO, BIO_R_UNINITIALIZED);
- return -2;
- }
-
- i = 0;
- if (buf != NULL && len > 0) {
- i = io_func(bio, buf, len);
- }
-
- if (i > 0) {
- *num += i;
- }
-
- if (bio->callback != NULL) {
- i = (int)(bio->callback(bio, callback_flags | BIO_CB_RETURN, buf, len, 0L,
- (long)i));
- }
-
- return i;
- }
-
- int BIO_read(BIO *bio, void *buf, int len) {
- return bio_io(bio, buf, len, offsetof(BIO_METHOD, bread), BIO_CB_READ,
- &bio->num_read);
- }
-
- int BIO_gets(BIO *bio, char *buf, int len) {
- return bio_io(bio, buf, len, offsetof(BIO_METHOD, bgets), BIO_CB_GETS,
- &bio->num_read);
- }
-
- int BIO_write(BIO *bio, const void *in, int inl) {
- return bio_io(bio, (char *)in, inl, offsetof(BIO_METHOD, bwrite),
- BIO_CB_WRITE, &bio->num_write);
- }
-
- int BIO_puts(BIO *bio, const char *in) {
- return BIO_write(bio, in, strlen(in));
- }
-
- int BIO_flush(BIO *bio) {
- return BIO_ctrl(bio, BIO_CTRL_FLUSH, 0, NULL);
- }
-
- long BIO_ctrl(BIO *bio, int cmd, long larg, void *parg) {
- long ret;
-
- if (bio == NULL) {
- return 0;
- }
-
- if (bio->method == NULL || bio->method->ctrl == NULL) {
- OPENSSL_PUT_ERROR(BIO, BIO_R_UNSUPPORTED_METHOD);
- return -2;
- }
-
- if (bio->callback != NULL) {
- ret = bio->callback(bio, BIO_CB_CTRL, parg, cmd, larg, 1);
- if (ret <= 0) {
- return ret;
- }
- }
-
- ret = bio->method->ctrl(bio, cmd, larg, parg);
-
- if (bio->callback != NULL) {
- ret = bio->callback(bio, BIO_CB_CTRL | BIO_CB_RETURN, parg, cmd, larg, ret);
- }
-
- return ret;
- }
-
- char *BIO_ptr_ctrl(BIO *b, int cmd, long larg) {
- char *p = NULL;
-
- if (BIO_ctrl(b, cmd, larg, (void *)&p) <= 0) {
- return NULL;
- }
-
- return p;
- }
-
- long BIO_int_ctrl(BIO *b, int cmd, long larg, int iarg) {
- int i = iarg;
-
- return BIO_ctrl(b, cmd, larg, (void *)&i);
- }
-
- int BIO_reset(BIO *bio) {
- return BIO_ctrl(bio, BIO_CTRL_RESET, 0, NULL);
- }
-
- void BIO_set_flags(BIO *bio, int flags) {
- bio->flags |= flags;
- }
-
- int BIO_test_flags(const BIO *bio, int flags) {
- return bio->flags & flags;
- }
-
- int BIO_should_read(const BIO *bio) {
- return BIO_test_flags(bio, BIO_FLAGS_READ);
- }
-
- int BIO_should_write(const BIO *bio) {
- return BIO_test_flags(bio, BIO_FLAGS_WRITE);
- }
-
- int BIO_should_retry(const BIO *bio) {
- return BIO_test_flags(bio, BIO_FLAGS_SHOULD_RETRY);
- }
-
- int BIO_should_io_special(const BIO *bio) {
- return BIO_test_flags(bio, BIO_FLAGS_IO_SPECIAL);
- }
-
- int BIO_get_retry_reason(const BIO *bio) { return bio->retry_reason; }
-
- void BIO_clear_flags(BIO *bio, int flags) {
- bio->flags &= ~flags;
- }
-
- void BIO_set_retry_read(BIO *bio) {
- bio->flags |= BIO_FLAGS_READ | BIO_FLAGS_SHOULD_RETRY;
- }
-
- void BIO_set_retry_write(BIO *bio) {
- bio->flags |= BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY;
- }
-
- static const int kRetryFlags = BIO_FLAGS_RWS | BIO_FLAGS_SHOULD_RETRY;
-
- int BIO_get_retry_flags(BIO *bio) {
- return bio->flags & kRetryFlags;
- }
-
- void BIO_clear_retry_flags(BIO *bio) {
- bio->flags &= ~kRetryFlags;
- bio->retry_reason = 0;
- }
-
- int BIO_method_type(const BIO *bio) { return bio->method->type; }
-
- void BIO_copy_next_retry(BIO *bio) {
- BIO_clear_retry_flags(bio);
- BIO_set_flags(bio, BIO_get_retry_flags(bio->next_bio));
- bio->retry_reason = bio->next_bio->retry_reason;
- }
-
- long BIO_callback_ctrl(BIO *bio, int cmd, bio_info_cb fp) {
- long ret;
- bio_info_cb cb;
-
- if (bio == NULL) {
- return 0;
- }
-
- if (bio->method == NULL || bio->method->callback_ctrl == NULL) {
- OPENSSL_PUT_ERROR(BIO, BIO_R_UNSUPPORTED_METHOD);
- return 0;
- }
-
- cb = bio->callback;
-
- if (cb != NULL) {
- ret = cb(bio, BIO_CB_CTRL, (void *)&fp, cmd, 0, 1L);
- if (ret <= 0) {
- return ret;
- }
- }
-
- ret = bio->method->callback_ctrl(bio, cmd, fp);
-
- if (cb != NULL) {
- ret = cb(bio, BIO_CB_CTRL | BIO_CB_RETURN, (void *)&fp, cmd, 0, ret);
- }
-
- return ret;
- }
-
- size_t BIO_pending(const BIO *bio) {
- return BIO_ctrl((BIO *) bio, BIO_CTRL_PENDING, 0, NULL);
- }
-
- size_t BIO_ctrl_pending(const BIO *bio) {
- return BIO_pending(bio);
- }
-
- size_t BIO_wpending(const BIO *bio) {
- return BIO_ctrl((BIO *) bio, BIO_CTRL_WPENDING, 0, NULL);
- }
-
- int BIO_set_close(BIO *bio, int close_flag) {
- return BIO_ctrl(bio, BIO_CTRL_SET_CLOSE, close_flag, NULL);
- }
-
- void BIO_set_callback(BIO *bio, bio_info_cb callback_func) {
- bio->callback = callback_func;
- }
-
- void BIO_set_callback_arg(BIO *bio, char *arg) {
- bio->cb_arg = arg;
- }
-
- char *BIO_get_callback_arg(const BIO *bio) {
- return bio->cb_arg;
- }
-
- OPENSSL_EXPORT size_t BIO_number_read(const BIO *bio) {
- return bio->num_read;
- }
-
- OPENSSL_EXPORT size_t BIO_number_written(const BIO *bio) {
- return bio->num_write;
- }
-
- BIO *BIO_push(BIO *bio, BIO *appended_bio) {
- BIO *last_bio;
-
- if (bio == NULL) {
- return bio;
- }
-
- last_bio = bio;
- while (last_bio->next_bio != NULL) {
- last_bio = last_bio->next_bio;
- }
-
- last_bio->next_bio = appended_bio;
- return bio;
- }
-
- BIO *BIO_pop(BIO *bio) {
- BIO *ret;
-
- if (bio == NULL) {
- return NULL;
- }
- ret = bio->next_bio;
- bio->next_bio = NULL;
- return ret;
- }
-
- BIO *BIO_next(BIO *bio) {
- if (!bio) {
- return NULL;
- }
- return bio->next_bio;
- }
-
- BIO *BIO_find_type(BIO *bio, int type) {
- int method_type, mask;
-
- if (!bio) {
- return NULL;
- }
- mask = type & 0xff;
-
- do {
- if (bio->method != NULL) {
- method_type = bio->method->type;
-
- if (!mask) {
- if (method_type & type) {
- return bio;
- }
- } else if (method_type == type) {
- return bio;
- }
- }
- bio = bio->next_bio;
- } while (bio != NULL);
-
- return NULL;
- }
-
- int BIO_indent(BIO *bio, unsigned indent, unsigned max_indent) {
- if (indent > max_indent) {
- indent = max_indent;
- }
-
- while (indent--) {
- if (BIO_puts(bio, " ") != 1) {
- return 0;
- }
- }
- return 1;
- }
-
- static int print_bio(const char *str, size_t len, void *bio) {
- return BIO_write((BIO *)bio, str, len);
- }
-
- void BIO_print_errors(BIO *bio) {
- ERR_print_errors_cb(print_bio, bio);
- }
-
- void ERR_print_errors(BIO *bio) {
- BIO_print_errors(bio);
- }
-
- /* bio_read_all reads everything from |bio| and prepends |prefix| to it. On
- * success, |*out| is set to an allocated buffer (which should be freed with
- * |OPENSSL_free|), |*out_len| is set to its length and one is returned. The
- * buffer will contain |prefix| followed by the contents of |bio|. On failure,
- * zero is returned.
- *
- * The function will fail if the size of the output would equal or exceed
- * |max_len|. */
- static int bio_read_all(BIO *bio, uint8_t **out, size_t *out_len,
- const uint8_t *prefix, size_t prefix_len,
- size_t max_len) {
- static const size_t kChunkSize = 4096;
-
- size_t len = prefix_len + kChunkSize;
- if (len > max_len) {
- len = max_len;
- }
- if (len < prefix_len) {
- return 0;
- }
- *out = OPENSSL_malloc(len);
- if (*out == NULL) {
- return 0;
- }
- memcpy(*out, prefix, prefix_len);
- size_t done = prefix_len;
-
- for (;;) {
- if (done == len) {
- OPENSSL_free(*out);
- return 0;
- }
- const size_t todo = len - done;
- assert(todo < INT_MAX);
- const int n = BIO_read(bio, *out + done, todo);
- if (n == 0) {
- *out_len = done;
- return 1;
- } else if (n == -1) {
- OPENSSL_free(*out);
- return 0;
- }
-
- done += n;
- if (len < max_len && len - done < kChunkSize / 2) {
- len += kChunkSize;
- if (len < kChunkSize || len > max_len) {
- len = max_len;
- }
- uint8_t *new_buf = OPENSSL_realloc(*out, len);
- if (new_buf == NULL) {
- OPENSSL_free(*out);
- return 0;
- }
- *out = new_buf;
- }
- }
- }
-
- int BIO_read_asn1(BIO *bio, uint8_t **out, size_t *out_len, size_t max_len) {
- uint8_t header[6];
-
- static const size_t kInitialHeaderLen = 2;
- if (BIO_read(bio, header, kInitialHeaderLen) != kInitialHeaderLen) {
- return 0;
- }
-
- const uint8_t tag = header[0];
- const uint8_t length_byte = header[1];
-
- if ((tag & 0x1f) == 0x1f) {
- /* Long form tags are not supported. */
- return 0;
- }
-
- size_t len, header_len;
- if ((length_byte & 0x80) == 0) {
- /* Short form length. */
- len = length_byte;
- header_len = kInitialHeaderLen;
- } else {
- const size_t num_bytes = length_byte & 0x7f;
-
- if ((tag & 0x20 /* constructed */) != 0 && num_bytes == 0) {
- /* indefinite length. */
- return bio_read_all(bio, out, out_len, header, kInitialHeaderLen,
- max_len);
- }
-
- if (num_bytes == 0 || num_bytes > 4) {
- return 0;
- }
-
- if (BIO_read(bio, header + kInitialHeaderLen, num_bytes) != num_bytes) {
- return 0;
- }
- header_len = kInitialHeaderLen + num_bytes;
-
- uint32_t len32 = 0;
- unsigned i;
- for (i = 0; i < num_bytes; i++) {
- len32 <<= 8;
- len32 |= header[kInitialHeaderLen + i];
- }
-
- if (len32 < 128) {
- /* Length should have used short-form encoding. */
- return 0;
- }
-
- if ((len32 >> ((num_bytes-1)*8)) == 0) {
- /* Length should have been at least one byte shorter. */
- return 0;
- }
-
- len = len32;
- }
-
- if (len + header_len < len ||
- len + header_len > max_len) {
- return 0;
- }
- len += header_len;
- *out_len = len;
-
- *out = OPENSSL_malloc(len);
- if (*out == NULL) {
- return 0;
- }
- memcpy(*out, header, header_len);
- if (BIO_read(bio, (*out) + header_len, len - header_len) !=
- len - header_len) {
- OPENSSL_free(*out);
- return 0;
- }
-
- return 1;
- }
|