1
0
mirror of git://sourceware.org/git/lvm2.git synced 2024-12-22 17:35:59 +03:00
lvm2/daemons/common/daemon-shared.c
Petr Rockai aac236f4a8 Improve format_buffer in daemon-shared.c, adding block formatting in addition
to string/integer (this propagates to the *simple* family of request/response
functionality).
2011-07-18 14:42:44 +00:00

117 lines
2.9 KiB
C

#include <errno.h>
#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include <assert.h>
#include "daemon-shared.h"
/*
* Read a single message from a (socket) filedescriptor. Messages are delimited
* by blank lines. This call will block until all of a message is received. The
* memory will be allocated from heap. Upon error, all memory is freed and the
* buffer pointer is set to NULL.
*
* See also write_buffer about blocking (read_buffer has identical behaviour).
*/
int read_buffer(int fd, char **buffer) {
int bytes = 0;
int buffersize = 32;
*buffer = malloc(buffersize + 1);
while (1) {
int result = read(fd, (*buffer) + bytes, buffersize - bytes);
if (result > 0)
bytes += result;
if (result == 0)
goto fail; /* we should never encounter EOF here */
if (result < 0 && errno != EAGAIN && errno != EWOULDBLOCK)
goto fail;
if (bytes == buffersize) {
buffersize += 1024;
char *new = realloc(*buffer, buffersize + 1);
if (new)
*buffer = new;
else
goto fail;
} else {
(*buffer)[bytes] = 0;
char *end;
if ((end = strstr((*buffer) + bytes - 2, "\n\n"))) {
*end = 0;
break; /* success, we have the full message now */
}
/* TODO call select here if we encountered EAGAIN/EWOULDBLOCK */
}
}
return 1;
fail:
free(*buffer);
*buffer = NULL;
return 0;
}
/*
* Write a buffer to a filedescriptor. Keep trying. Blocks (even on
* SOCK_NONBLOCK) until all of the write went through.
*
* TODO use select on EWOULDBLOCK/EAGAIN to avoid useless spinning
*/
int write_buffer(int fd, char *buffer, int length) {
int written = 0;
while (1) {
int result = write(fd, buffer + written, length - written);
if (result > 0)
written += result;
if (result < 0 && errno != EWOULDBLOCK && errno != EAGAIN)
break; /* too bad */
if (written == length)
return 1; /* done */
}
return 0;
}
char *format_buffer(const char *what, const char *id, va_list ap)
{
char *buffer, *old;
char *next;
int keylen;
dm_asprintf(&buffer, "%s = \"%s\"\n", what, id);
if (!buffer) goto fail;
while (next = va_arg(ap, char *)) {
old = buffer;
assert(strchr(next, '='));
keylen = strchr(next, '=') - next;
if (strstr(next, "%d")) {
int value = va_arg(ap, int);
dm_asprintf(&buffer, "%s%.*s= %d\n", buffer, keylen, next, value);
dm_free(old);
} else if (strstr(next, "%s")) {
char *value = va_arg(ap, char *);
dm_asprintf(&buffer, "%s%.*s= \"%s\"\n", buffer, keylen, next, value);
dm_free(old);
} else if (strstr(next, "%b")) {
char *block = va_arg(ap, char *);
if (!block)
continue;
dm_asprintf(&buffer, "%s%.*s%s", buffer, keylen, next, block);
dm_free(old);
} else {
dm_asprintf(&buffer, "%s%s", buffer, next);
dm_free(old);
}
if (!buffer) goto fail;
}
old = buffer;
dm_asprintf(&buffer, "%s\n", buffer);
dm_free(old);
return buffer;
fail:
dm_free(buffer);
return NULL;
}