0,0 → 1,168 |
/* |
* Copyright 2009 VMware, Inc. |
* All Rights Reserved. |
* |
* Permission is hereby granted, free of charge, to any person obtaining a |
* copy of this software and associated documentation files (the "Software"), |
* to deal in the Software without restriction, including without limitation |
* on the rights to use, copy, modify, merge, publish, distribute, sub |
* license, and/or sell copies of the Software, and to permit persons to whom |
* the Software is furnished to do so, subject to the following conditions: |
* |
* The above copyright notice and this permission notice (including the next |
* paragraph) shall be included in all copies or substantial portions of the |
* Software. |
* |
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
* FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL |
* VMWARE AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, |
* DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
* OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
* USE OR OTHER DEALINGS IN THE SOFTWARE. |
*/ |
|
#include "rbug.h" |
#include "rbug_internal.h" |
|
#include "util/u_network.h" |
|
struct rbug_connection |
{ |
int socket; |
uint32_t send_serial; |
uint32_t recv_serial; |
enum rbug_opcode opcode; |
}; |
|
/** |
* Create a rbug connection from a socket created with u_socket. |
* |
* Result: |
* A new allocated connection using socket as communication path |
*/ |
struct rbug_connection * |
rbug_from_socket(int socket) |
{ |
struct rbug_connection *c = CALLOC_STRUCT(rbug_connection); |
c->socket = socket; |
return c; |
} |
|
/** |
* Free a connection, also closes socket. |
*/ |
void |
rbug_disconnect(struct rbug_connection *c) |
{ |
u_socket_close(c->socket); |
FREE(c); |
} |
|
/** |
* Waits for a message to be fully received. |
* Also returns the serial for the message, serial is not touched for replys. |
* |
* Result: |
* demarshaled message on success, NULL on connection error |
*/ |
struct rbug_header * |
rbug_get_message(struct rbug_connection *c, uint32_t *serial) |
{ |
struct rbug_proto_header header; |
struct rbug_header *out; |
struct rbug_proto_header *data; |
size_t length = 0; |
size_t read = 0; |
int ret; |
|
|
ret = u_socket_peek(c->socket, &header, sizeof(header)); |
if (ret <= 0) { |
return NULL; |
} |
|
length = (size_t)header.length * 4; |
data = MALLOC(length); |
if (!data) { |
return NULL; |
} |
data->opcode = 0; |
|
do { |
uint8_t *ptr = ((uint8_t*)data) + read; |
ret = u_socket_recv(c->socket, ptr, length - read); |
|
if (ret <= 0) { |
FREE(data); |
return NULL; |
} |
|
read += ret; |
} while(read < length); |
|
out = rbug_demarshal(data); |
if (!out) |
FREE(data); |
else if (serial) |
*serial = c->recv_serial++; |
else |
c->recv_serial++; |
|
return out; |
} |
|
/** |
* Frees a message and associated data. |
*/ |
void |
rbug_free_header(struct rbug_header *header) |
{ |
if (!header) |
return; |
|
FREE(header->__message); |
FREE(header); |
} |
|
/** |
* Internal function used by rbug_send_* functions. |
* |
* Start sending a message. |
*/ |
int |
rbug_connection_send_start(struct rbug_connection *c, enum rbug_opcode opcode, uint32_t length) |
{ |
c->opcode = opcode; |
return 0; |
} |
|
/** |
* Internal function used by rbug_send_* functions. |
* |
* Write data to the socket. |
*/ |
int |
rbug_connection_write(struct rbug_connection *c, void *to, uint32_t size) |
{ |
int ret = u_socket_send(c->socket, to, size); |
return ret; |
} |
|
/** |
* Internal function used by rbug_send_* functions. |
* |
* Finish writeing data to the socket. |
* Ups the send_serial and sets the serial argument if supplied. |
*/ |
int rbug_connection_send_finish(struct rbug_connection *c, uint32_t *serial) |
{ |
if (c->opcode < 0) |
return 0; |
else if (serial) |
*serial = c->send_serial++; |
else |
c->send_serial++; |
|
return 0; |
} |