Kristian Høgsberg
16 years ago
parent
c5089878cb
commit
680f1c7025
@ -0,0 +1,193 @@ |
||||
#include <stdlib.h> |
||||
#include <stdint.h> |
||||
#include <string.h> |
||||
#include <stdio.h> |
||||
#include <errno.h> |
||||
#include <sys/uio.h> |
||||
|
||||
#include "connection.h" |
||||
|
||||
#define ARRAY_LENGTH(a) (sizeof (a) / sizeof (a)[0]) |
||||
|
||||
struct wl_buffer { |
||||
char data[4096]; |
||||
int head, tail; |
||||
}; |
||||
|
||||
struct wl_connection { |
||||
struct wl_buffer in, out; |
||||
int fd; |
||||
void *data; |
||||
wl_connection_update_func_t update; |
||||
}; |
||||
|
||||
struct wl_connection * |
||||
wl_connection_create(int fd, |
||||
wl_connection_update_func_t update, |
||||
void *data) |
||||
{ |
||||
struct wl_connection *connection; |
||||
|
||||
connection = malloc(sizeof *connection); |
||||
memset(connection, 0, sizeof *connection); |
||||
connection->fd = fd; |
||||
connection->update = update; |
||||
connection->data = data; |
||||
|
||||
connection->update(connection, |
||||
WL_CONNECTION_READABLE, |
||||
connection->data); |
||||
|
||||
return connection; |
||||
} |
||||
|
||||
void |
||||
wl_connection_destroy(struct wl_connection *connection) |
||||
{ |
||||
free(connection); |
||||
} |
||||
|
||||
void |
||||
wl_connection_copy(struct wl_connection *connection, void *data, size_t size) |
||||
{ |
||||
struct wl_buffer *b; |
||||
int tail, rest; |
||||
|
||||
b = &connection->in; |
||||
tail = b->tail; |
||||
if (tail + size <= ARRAY_LENGTH(b->data)) { |
||||
memcpy(data, b->data + tail, size); |
||||
} else {
|
||||
rest = ARRAY_LENGTH(b->data) - tail; |
||||
memcpy(data, b->data + tail, rest); |
||||
memcpy(data + rest, b->data, size - rest); |
||||
} |
||||
} |
||||
|
||||
void |
||||
wl_connection_consume(struct wl_connection *connection, size_t size) |
||||
{ |
||||
struct wl_buffer *b; |
||||
int tail, rest; |
||||
|
||||
b = &connection->in; |
||||
tail = b->tail; |
||||
if (tail + size <= ARRAY_LENGTH(b->data)) { |
||||
b->tail += size; |
||||
} else {
|
||||
rest = ARRAY_LENGTH(b->data) - tail; |
||||
b->tail = size - rest; |
||||
} |
||||
} |
||||
|
||||
int wl_connection_data(struct wl_connection *connection, uint32_t mask) |
||||
{ |
||||
struct wl_buffer *b; |
||||
struct iovec iov[2]; |
||||
int len, head, tail, count, size, available; |
||||
|
||||
if (mask & WL_CONNECTION_READABLE) { |
||||
b = &connection->in; |
||||
head = connection->in.head; |
||||
if (head < b->tail) { |
||||
iov[0].iov_base = b->data + head; |
||||
iov[0].iov_len = b->tail - head; |
||||
count = 1; |
||||
} else { |
||||
size = ARRAY_LENGTH(b->data) - head; |
||||
iov[0].iov_base = b->data + head; |
||||
iov[0].iov_len = size; |
||||
iov[1].iov_base = b->data; |
||||
iov[1].iov_len = b->tail; |
||||
count = 2; |
||||
} |
||||
len = readv(connection->fd, iov, count); |
||||
if (len < 0) { |
||||
fprintf(stderr, |
||||
"read error from connection %p: %m (%d)\n", |
||||
connection, errno); |
||||
return -1; |
||||
} else if (len == 0) { |
||||
/* FIXME: Handle this better? */ |
||||
return -1; |
||||
} else if (head + len <= ARRAY_LENGTH(b->data)) { |
||||
b->head += len; |
||||
} else { |
||||
b->head = head + len - ARRAY_LENGTH(b->data); |
||||
} |
||||
|
||||
/* We know we have data in the buffer at this point,
|
||||
* so if head equals tail, it means the buffer is |
||||
* full. */ |
||||
|
||||
available = b->head - b->tail; |
||||
if (available == 0) |
||||
available = sizeof b->data; |
||||
else if (available < 0) |
||||
available += ARRAY_LENGTH(b->data); |
||||
} else { |
||||
available = 0; |
||||
}
|
||||
|
||||
if (mask & WL_CONNECTION_WRITABLE) { |
||||
b = &connection->out; |
||||
tail = b->tail; |
||||
if (tail < b->head) { |
||||
iov[0].iov_base = b->data + tail; |
||||
iov[0].iov_len = b->head - tail; |
||||
count = 1; |
||||
} else { |
||||
size = ARRAY_LENGTH(b->data) - tail; |
||||
iov[0].iov_base = b->data + tail; |
||||
iov[0].iov_len = size; |
||||
iov[1].iov_base = b->data; |
||||
iov[1].iov_len = b->head; |
||||
count = 2; |
||||
} |
||||
len = writev(connection->fd, iov, count); |
||||
if (len < 0) { |
||||
fprintf(stderr, "write error for connection %p: %m\n", connection); |
||||
return -1; |
||||
} else if (tail + len <= ARRAY_LENGTH(b->data)) { |
||||
b->tail += len; |
||||
} else { |
||||
b->tail = tail + len - ARRAY_LENGTH(b->data); |
||||
} |
||||
|
||||
/* We just took data out of the buffer, so at this
|
||||
* point if head equals tail, the buffer is empty. */ |
||||
|
||||
if (b->tail == b->head) |
||||
connection->update(connection, |
||||
WL_CONNECTION_READABLE, |
||||
connection->data); |
||||
} |
||||
|
||||
return available; |
||||
} |
||||
|
||||
void |
||||
wl_connection_write(struct wl_connection *connection, const void *data, size_t count) |
||||
{ |
||||
struct wl_buffer *b; |
||||
size_t size; |
||||
int head; |
||||
|
||||
b = &connection->out; |
||||
head = b->head; |
||||
if (head + count <= ARRAY_LENGTH(b->data)) { |
||||
memcpy(b->data + head, data, count); |
||||
b->head += count; |
||||
} else { |
||||
size = ARRAY_LENGTH(b->data) - head; |
||||
memcpy(b->data + head, data, size); |
||||
memcpy(b->data, data + size, count - size); |
||||
b->head = count - size; |
||||
} |
||||
|
||||
if (b->tail == head) |
||||
connection->update(connection, |
||||
WL_CONNECTION_READABLE | |
||||
WL_CONNECTION_WRITABLE, |
||||
connection->data); |
||||
} |
||||
@ -0,0 +1,21 @@ |
||||
#ifndef _CONNECTION_H_ |
||||
#define _CONNECTION_H_ |
||||
|
||||
struct wl_connection; |
||||
|
||||
#define WL_CONNECTION_READABLE 0x01 |
||||
#define WL_CONNECTION_WRITABLE 0x02 |
||||
|
||||
typedef int (*wl_connection_update_func_t)(struct wl_connection *connection, |
||||
uint32_t mask, void *data); |
||||
|
||||
struct wl_connection *wl_connection_create(int fd, |
||||
wl_connection_update_func_t update, |
||||
void *data); |
||||
void wl_connection_destroy(struct wl_connection *connection); |
||||
void wl_connection_copy(struct wl_connection *connection, void *data, size_t size); |
||||
void wl_connection_consume(struct wl_connection *connection, size_t size); |
||||
int wl_connection_data(struct wl_connection *connection, uint32_t mask); |
||||
void wl_connection_write(struct wl_connection *connection, const void *data, size_t count); |
||||
|
||||
#endif |
||||
Loading…
Reference in new issue