X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=lib%2Fpb-protocol%2Fpb-protocol.c;h=6f278c89ac00e24185b8d8071152b3825076b413;hp=69c9bfea63e5017051e9a66509c9aa9fc36c02bc;hb=192291bf0e5ae7ad714c863926ed0b8981102e9c;hpb=08b9b747db92ce682e094ef86f48321d29fc44d6 diff --git a/lib/pb-protocol/pb-protocol.c b/lib/pb-protocol/pb-protocol.c index 69c9bfe..6f278c8 100644 --- a/lib/pb-protocol/pb-protocol.c +++ b/lib/pb-protocol/pb-protocol.c @@ -1,11 +1,13 @@ #include +#include #include #include #include #include #include +#include #include "pb-protocol.h" @@ -38,26 +40,6 @@ * 4-byte len, id */ -/* Deep copy a boot_option. -*/ -struct boot_option *boot_option_copy(void* ctx, const struct boot_option *opt) -{ - struct boot_option *new = talloc(ctx, struct boot_option); - - if (new) { - new->id = talloc_strdup(new, opt->id); - new->name = talloc_strdup(new, opt->name); - new->description = talloc_strdup(new, opt->description); - new->icon_file = talloc_strdup(new, opt->icon_file); - new->boot_image_file = talloc_strdup(new, opt->boot_image_file); - new->initrd_file = talloc_strdup(new, opt->initrd_file); - new->boot_args = talloc_strdup(new, opt->boot_args); - memset(&new->list, 0, sizeof(new->list)); - } - - return new; -} - int pb_protocol_device_cmp(const struct device *a, const struct device *b) { return !strcmp(a->id, b->id); @@ -232,7 +214,11 @@ int pb_protocol_write_message(int fd, struct pb_protocol_message *message) talloc_free(message); - return total_len ? -1 : 0; + if (!total_len) + return 0; + + pb_log("%s: failed: %s\n", __func__, strerror(errno)); + return -1; } struct pb_protocol_message *pb_protocol_create_message(void *ctx, @@ -240,8 +226,11 @@ struct pb_protocol_message *pb_protocol_create_message(void *ctx, { struct pb_protocol_message *message; - if (payload_len > PB_PROTOCOL_MAX_PAYLOAD_SIZE) + if (payload_len > PB_PROTOCOL_MAX_PAYLOAD_SIZE) { + pb_log("%s: payload too big %u/%u\n", __func__, payload_len, + PB_PROTOCOL_MAX_PAYLOAD_SIZE); return NULL; + } message = talloc_size(ctx, sizeof(*message) + payload_len); @@ -268,8 +257,11 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd) m.payload_len = __be32_to_cpu(m.payload_len); m.action = __be32_to_cpu(m.action); - if (m.payload_len > PB_PROTOCOL_MAX_PAYLOAD_SIZE) + if (m.payload_len > PB_PROTOCOL_MAX_PAYLOAD_SIZE) { + pb_log("%s: payload too big %u/%u\n", __func__, m.payload_len, + PB_PROTOCOL_MAX_PAYLOAD_SIZE); return NULL; + } message = talloc_size(ctx, sizeof(m) + m.payload_len); memcpy(message, &m, sizeof(m)); @@ -279,6 +271,8 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd) if (rc <= 0) { talloc_free(message); + pb_log("%s: failed (%u): %s\n", __func__, len, + strerror(errno)); return NULL; }