/*
|
* 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 writing 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;
|
}
|
