diff options
Diffstat (limited to 'librabbitmq/amqp_connection.c')
| -rw-r--r-- | librabbitmq/amqp_connection.c | 436 |
1 files changed, 0 insertions, 436 deletions
diff --git a/librabbitmq/amqp_connection.c b/librabbitmq/amqp_connection.c deleted file mode 100644 index 34d12f8..0000000 --- a/librabbitmq/amqp_connection.c +++ /dev/null @@ -1,436 +0,0 @@ -/* - * ***** BEGIN LICENSE BLOCK ***** - * Version: MPL 1.1/GPL 2.0 - * - * The contents of this file are subject to the Mozilla Public License - * Version 1.1 (the "License"); you may not use this file except in - * compliance with the License. You may obtain a copy of the License at - * http://www.mozilla.org/MPL/ - * - * Software distributed under the License is distributed on an "AS IS" - * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See - * the License for the specific language governing rights and - * limitations under the License. - * - * The Original Code is librabbitmq. - * - * The Initial Developers of the Original Code are LShift Ltd, Cohesive - * Financial Technologies LLC, and Rabbit Technologies Ltd. Portions - * created before 22-Nov-2008 00:00:00 GMT by LShift Ltd, Cohesive - * Financial Technologies LLC, or Rabbit Technologies Ltd are Copyright - * (C) 2007-2008 LShift Ltd, Cohesive Financial Technologies LLC, and - * Rabbit Technologies Ltd. - * - * Portions created by LShift Ltd are Copyright (C) 2007-2009 LShift - * Ltd. Portions created by Cohesive Financial Technologies LLC are - * Copyright (C) 2007-2009 Cohesive Financial Technologies - * LLC. Portions created by Rabbit Technologies Ltd are Copyright (C) - * 2007-2009 Rabbit Technologies Ltd. - * - * Portions created by Tony Garnock-Jones are Copyright (C) 2009-2010 - * LShift Ltd and Tony Garnock-Jones. - * - * All Rights Reserved. - * - * Contributor(s): ______________________________________. - * - * Alternatively, the contents of this file may be used under the terms - * of the GNU General Public License Version 2 or later (the "GPL"), in - * which case the provisions of the GPL are applicable instead of those - * above. If you wish to allow use of your version of this file only - * under the terms of the GPL, and not to allow others to use your - * version of this file under the terms of the MPL, indicate your - * decision by deleting the provisions above and replace them with the - * notice and other provisions required by the GPL. If you do not - * delete the provisions above, a recipient may use your version of - * this file under the terms of any one of the MPL or the GPL. - * - * ***** END LICENSE BLOCK ***** - */ - -#include <stdlib.h> -#include <stdio.h> -#include <string.h> -#include <stdint.h> -#include <assert.h> - -#include "amqp.h" -#include "amqp_framing.h" -#include "amqp_private.h" - -#define INITIAL_FRAME_POOL_PAGE_SIZE 65536 -#define INITIAL_DECODING_POOL_PAGE_SIZE 131072 -#define INITIAL_INBOUND_SOCK_BUFFER_SIZE 131072 - -#define ENFORCE_STATE(statevec, statenum) \ - { \ - amqp_connection_state_t _check_state = (statevec); \ - int _wanted_state = (statenum); \ - if (_check_state->state != _wanted_state) \ - amqp_abort("Programming error: invalid AMQP connection state: expected %d, got %d", \ - _wanted_state, \ - _check_state->state); \ - } - -amqp_connection_state_t amqp_new_connection(void) { - amqp_connection_state_t state = - (amqp_connection_state_t) calloc(1, sizeof(struct amqp_connection_state_t_)); - - if (state == NULL) - return NULL; - - init_amqp_pool(&state->frame_pool, INITIAL_FRAME_POOL_PAGE_SIZE); - init_amqp_pool(&state->decoding_pool, INITIAL_DECODING_POOL_PAGE_SIZE); - - if (amqp_tune_connection(state, 0, INITIAL_FRAME_POOL_PAGE_SIZE, 0) != 0) - goto out_nomem; - - state->inbound_buffer.bytes = amqp_pool_alloc(&state->frame_pool, state->inbound_buffer.len); - if (state->inbound_buffer.bytes == NULL) - goto out_nomem; - - state->state = CONNECTION_STATE_INITIAL; - /* the server protocol version response is 8 bytes, which conveniently - is also the minimum frame size */ - state->target_size = 8; - - state->sockfd = -1; - state->sock_inbound_buffer.len = INITIAL_INBOUND_SOCK_BUFFER_SIZE; - state->sock_inbound_buffer.bytes = malloc(INITIAL_INBOUND_SOCK_BUFFER_SIZE); - if (state->sock_inbound_buffer.bytes == NULL) - goto out_nomem; - - return state; - - out_nomem: - free(state->sock_inbound_buffer.bytes); - empty_amqp_pool(&state->frame_pool); - empty_amqp_pool(&state->decoding_pool); - free(state); - return NULL; -} - -int amqp_get_sockfd(amqp_connection_state_t state) { - return state->sockfd; -} - -void amqp_set_sockfd(amqp_connection_state_t state, - int sockfd) -{ - state->sockfd = sockfd; -} - -int amqp_tune_connection(amqp_connection_state_t state, - int channel_max, - int frame_max, - int heartbeat) -{ - void *newbuf; - - ENFORCE_STATE(state, CONNECTION_STATE_IDLE); - - state->channel_max = channel_max; - state->frame_max = frame_max; - state->heartbeat = heartbeat; - - empty_amqp_pool(&state->frame_pool); - init_amqp_pool(&state->frame_pool, frame_max); - - state->inbound_buffer.len = frame_max; - state->outbound_buffer.len = frame_max; - newbuf = realloc(state->outbound_buffer.bytes, frame_max); - if (newbuf == NULL) { - amqp_destroy_connection(state); - return -ERROR_NO_MEMORY; - } - state->outbound_buffer.bytes = newbuf; - - return 0; -} - -int amqp_get_channel_max(amqp_connection_state_t state) { - return state->channel_max; -} - -int amqp_destroy_connection(amqp_connection_state_t state) { - int s = state->sockfd; - - empty_amqp_pool(&state->frame_pool); - empty_amqp_pool(&state->decoding_pool); - free(state->outbound_buffer.bytes); - free(state->sock_inbound_buffer.bytes); - free(state); - - if (s >= 0 && amqp_socket_close(s) < 0) - return -amqp_socket_error(); - else - return 0; -} - -static void return_to_idle(amqp_connection_state_t state) { - state->inbound_buffer.bytes = NULL; - state->inbound_offset = 0; - state->target_size = HEADER_SIZE; - state->state = CONNECTION_STATE_IDLE; -} - -static size_t consume_data(amqp_connection_state_t state, - amqp_bytes_t *received_data) -{ - /* how much data is available and will fit? */ - size_t bytes_consumed = state->target_size - state->inbound_offset; - if (received_data->len < bytes_consumed) - bytes_consumed = received_data->len; - - memcpy(amqp_offset(state->inbound_buffer.bytes, state->inbound_offset), - received_data->bytes, bytes_consumed); - state->inbound_offset += bytes_consumed; - received_data->bytes = amqp_offset(received_data->bytes, bytes_consumed); - received_data->len -= bytes_consumed; - - return bytes_consumed; -} - -int amqp_handle_input(amqp_connection_state_t state, - amqp_bytes_t received_data, - amqp_frame_t *decoded_frame) -{ - size_t bytes_consumed; - void *raw_frame; - - /* Returning frame_type of zero indicates either insufficient input, - or a complete, ignored frame was read. */ - decoded_frame->frame_type = 0; - - if (received_data.len == 0) - return 0; - - if (state->state == CONNECTION_STATE_IDLE) { - state->inbound_buffer.bytes = amqp_pool_alloc(&state->frame_pool, - state->inbound_buffer.len); - if (state->inbound_buffer.bytes == NULL) - /* state->inbound_buffer.len is always nonzero, because it - corresponds to frame_max, which is not permitted to be less - than AMQP_FRAME_MIN_SIZE (currently 4096 bytes). */ - return -ERROR_NO_MEMORY; - - state->state = CONNECTION_STATE_HEADER; - } - - bytes_consumed = consume_data(state, &received_data); - - /* do we have target_size data yet? if not, return with the - expectation that more will arrive */ - if (state->inbound_offset < state->target_size) - return bytes_consumed; - - raw_frame = state->inbound_buffer.bytes; - - switch (state->state) { - case CONNECTION_STATE_INITIAL: - /* check for a protocol header from the server */ - if (memcmp(raw_frame, "AMQP", 4) == 0) { - decoded_frame->frame_type = AMQP_PSEUDOFRAME_PROTOCOL_HEADER; - decoded_frame->channel = 0; - - decoded_frame->payload.protocol_header.transport_high - = amqp_d8(raw_frame, 4); - decoded_frame->payload.protocol_header.transport_low - = amqp_d8(raw_frame, 5); - decoded_frame->payload.protocol_header.protocol_version_major - = amqp_d8(raw_frame, 6); - decoded_frame->payload.protocol_header.protocol_version_minor - = amqp_d8(raw_frame, 7); - - return_to_idle(state); - return bytes_consumed; - } - - /* it's not a protocol header; fall through to process it as a - regular frame header */ - - case CONNECTION_STATE_HEADER: - /* frame length is 3 bytes in */ - state->target_size - = amqp_d32(raw_frame, 3) + HEADER_SIZE + FOOTER_SIZE; - state->state = CONNECTION_STATE_BODY; - - bytes_consumed += consume_data(state, &received_data); - - /* do we have target_size data yet? if not, return with the - expectation that more will arrive */ - if (state->inbound_offset < state->target_size) - return bytes_consumed; - - /* fall through to process body */ - - case CONNECTION_STATE_BODY: { - amqp_bytes_t encoded; - int res; - - /* Check frame end marker (footer) */ - if (amqp_d8(raw_frame, state->target_size - 1) != AMQP_FRAME_END) - return -ERROR_BAD_AMQP_DATA; - - decoded_frame->frame_type = amqp_d8(raw_frame, 0); - decoded_frame->channel = amqp_d16(raw_frame, 1); - - switch (decoded_frame->frame_type) { - case AMQP_FRAME_METHOD: - decoded_frame->payload.method.id = amqp_d32(raw_frame, HEADER_SIZE); - encoded.bytes = amqp_offset(raw_frame, HEADER_SIZE + 4); - encoded.len = state->target_size - HEADER_SIZE - 4 - FOOTER_SIZE; - - res = amqp_decode_method(decoded_frame->payload.method.id, - &state->decoding_pool, encoded, - &decoded_frame->payload.method.decoded); - if (res < 0) - return res; - - break; - - case AMQP_FRAME_HEADER: - decoded_frame->payload.properties.class_id - = amqp_d16(raw_frame, HEADER_SIZE); - /* unused 2-byte weight field goes here */ - decoded_frame->payload.properties.body_size - = amqp_d64(raw_frame, HEADER_SIZE + 4); - encoded.bytes = amqp_offset(raw_frame, HEADER_SIZE + 12); - encoded.len = state->target_size - HEADER_SIZE - 12 - FOOTER_SIZE; - decoded_frame->payload.properties.raw = encoded; - - res = amqp_decode_properties(decoded_frame->payload.properties.class_id, - &state->decoding_pool, encoded, - &decoded_frame->payload.properties.decoded); - if (res < 0) - return res; - - break; - - case AMQP_FRAME_BODY: - decoded_frame->payload.body_fragment.len - = state->target_size - HEADER_SIZE - FOOTER_SIZE; - decoded_frame->payload.body_fragment.bytes - = amqp_offset(raw_frame, HEADER_SIZE); - break; - - case AMQP_FRAME_HEARTBEAT: - break; - - default: - /* Ignore the frame */ - decoded_frame->frame_type = 0; - break; - } - - return_to_idle(state); - return bytes_consumed; - } - - default: - amqp_abort("Internal error: invalid amqp_connection_state_t->state %d", state->state); - return bytes_consumed; - } -} - -amqp_boolean_t amqp_release_buffers_ok(amqp_connection_state_t state) { - return (state->state == CONNECTION_STATE_IDLE) && (state->first_queued_frame == NULL); -} - -void amqp_release_buffers(amqp_connection_state_t state) { - ENFORCE_STATE(state, CONNECTION_STATE_IDLE); - - if (state->first_queued_frame) - amqp_abort("Programming error: attempt to amqp_release_buffers while waiting events enqueued"); - - recycle_amqp_pool(&state->frame_pool); - recycle_amqp_pool(&state->decoding_pool); -} - -void amqp_maybe_release_buffers(amqp_connection_state_t state) { - if (amqp_release_buffers_ok(state)) { - amqp_release_buffers(state); - } -} - -int amqp_send_frame(amqp_connection_state_t state, - const amqp_frame_t *frame) -{ - void *out_frame = state->outbound_buffer.bytes; - int res; - - amqp_e8(out_frame, 0, frame->frame_type); - amqp_e16(out_frame, 1, frame->channel); - - if (frame->frame_type == AMQP_FRAME_BODY) { - /* For a body frame, rather than copying data around, we use - writev to compose the frame */ - struct iovec iov[3]; - uint8_t frame_end_byte = AMQP_FRAME_END; - const amqp_bytes_t *body = &frame->payload.body_fragment; - - amqp_e32(out_frame, 3, body->len); - - iov[0].iov_base = out_frame; - iov[0].iov_len = HEADER_SIZE; - iov[1].iov_base = body->bytes; - iov[1].iov_len = body->len; - iov[2].iov_base = &frame_end_byte; - iov[2].iov_len = FOOTER_SIZE; - - res = amqp_socket_writev(state->sockfd, iov, 3); - } - else { - size_t out_frame_len; - amqp_bytes_t encoded; - - switch (frame->frame_type) { - case AMQP_FRAME_METHOD: - amqp_e32(out_frame, HEADER_SIZE, frame->payload.method.id); - - encoded.bytes = amqp_offset(out_frame, HEADER_SIZE + 4); - encoded.len = state->outbound_buffer.len - HEADER_SIZE - 4 - FOOTER_SIZE; - - res = amqp_encode_method(frame->payload.method.id, - frame->payload.method.decoded, encoded); - if (res < 0) - return res; - - out_frame_len = res + 4; - break; - - case AMQP_FRAME_HEADER: - amqp_e16(out_frame, HEADER_SIZE, frame->payload.properties.class_id); - amqp_e16(out_frame, HEADER_SIZE+2, 0); /* "weight" */ - amqp_e64(out_frame, HEADER_SIZE+4, frame->payload.properties.body_size); - - encoded.bytes = amqp_offset(out_frame, HEADER_SIZE + 12); - encoded.len = state->outbound_buffer.len - HEADER_SIZE - 12 - FOOTER_SIZE; - - res = amqp_encode_properties(frame->payload.properties.class_id, - frame->payload.properties.decoded, encoded); - if (res < 0) - return res; - - out_frame_len = res + 12; - break; - - case AMQP_FRAME_HEARTBEAT: - out_frame_len = 0; - break; - - default: - abort(); - } - - amqp_e32(out_frame, 3, out_frame_len); - amqp_e8(out_frame, out_frame_len + HEADER_SIZE, AMQP_FRAME_END); - res = send(state->sockfd, out_frame, - out_frame_len + HEADER_SIZE + FOOTER_SIZE, 0); - } - - if (res < 0) - return -amqp_socket_error(); - else - return 0; -} |