X-Git-Url: http://git.silcnet.org/gitweb/?a=blobdiff_plain;f=lib%2Fsilccore%2Fsilcnotify.c;h=4f6feacbda0a14c42aaff2f1d3c0dc8196b7fd1e;hb=40f8443d8d3a6577336ee66d18e04d9ac4d956bb;hp=e3c8169dafb35166cdb7f1dad098faf1c38eb149;hpb=4b75fb20302adc84ef77c3bf4131473f5dc615ca;p=silc.git diff --git a/lib/silccore/silcnotify.c b/lib/silccore/silcnotify.c index e3c8169d..4f6feacb 100644 --- a/lib/silccore/silcnotify.c +++ b/lib/silccore/silcnotify.c @@ -2,15 +2,14 @@ silcnotify.c - Author: Pekka Riikonen + Author: Pekka Riikonen - Copyright (C) 2000 Pekka Riikonen + Copyright (C) 2000 - 2005 Pekka Riikonen This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by - the Free Software Foundation; either version 2 of the License, or - (at your option) any later version. - + the Free Software Foundation; version 2 of the License. + This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the @@ -19,7 +18,7 @@ */ /* $Id$ */ -#include "silcincludes.h" +#include "silc.h" #include "silcnotify.h" /****************************************************************************** @@ -36,76 +35,94 @@ struct SilcNotifyPayloadStruct { /* Parse notify payload buffer and return data into payload structure */ -SilcNotifyPayload silc_notify_payload_parse(SilcBuffer buffer) +SilcNotifyPayload silc_notify_payload_parse(const unsigned char *payload, + SilcUInt32 payload_len) { - SilcNotifyPayload new; - unsigned short len; + SilcBufferStruct buffer; + SilcNotifyPayload newp; + SilcUInt16 len; int ret; SILC_LOG_DEBUG(("Parsing Notify payload")); - new = silc_calloc(1, sizeof(*new)); + silc_buffer_set(&buffer, (unsigned char *)payload, payload_len); + newp = silc_calloc(1, sizeof(*newp)); + if (!newp) + return NULL; - ret = silc_buffer_unformat(buffer, - SILC_STR_UI_SHORT(&new->type), + ret = silc_buffer_unformat(&buffer, + SILC_STR_UI_SHORT(&newp->type), SILC_STR_UI_SHORT(&len), - SILC_STR_UI_CHAR(&new->argc), + SILC_STR_UI_CHAR(&newp->argc), SILC_STR_END); if (ret == -1) goto err; - if (len > buffer->len) + if (len > silc_buffer_len(&buffer)) goto err; - if (new->argc) { - silc_buffer_pull(buffer, 5); - new->args = silc_argument_payload_parse(buffer, new->argc); - silc_buffer_push(buffer, 5); + if (newp->argc) { + silc_buffer_pull(&buffer, 5); + newp->args = silc_argument_payload_parse(buffer.data, silc_buffer_len(&buffer), + newp->argc); + silc_buffer_push(&buffer, 5); } - return new; + return newp; err: - silc_free(new); + silc_free(newp); return NULL; } /* Encode notify payload with variable argument list. If `argc' is > 0 argument payloads will be associated to the notify payload. Variable - arguments must be {usigned char *, unsigned int (len)}. */ + arguments must be {usigned char *, SilcUInt32 (len)}. */ -SilcBuffer silc_notify_payload_encode(SilcNotifyType type, unsigned int argc, +SilcBuffer silc_notify_payload_encode(SilcNotifyType type, SilcUInt32 argc, va_list ap) { SilcBuffer buffer; SilcBuffer args = NULL; unsigned char **argv; - unsigned int *argv_lens = NULL, *argv_types = NULL; + SilcUInt32 *argv_lens = NULL, *argv_types = NULL; unsigned char *x; - unsigned int x_len; - int i, k = 0, len = 0; + SilcUInt32 x_len, len = 0; + int i, k = 0; if (argc) { argv = silc_calloc(argc, sizeof(unsigned char *)); - argv_lens = silc_calloc(argc, sizeof(unsigned int)); - argv_types = silc_calloc(argc, sizeof(unsigned int)); - + if (!argv) + return NULL; + argv_lens = silc_calloc(argc, sizeof(SilcUInt32)); + if (!argv_lens) { + silc_free(argv); + return NULL; + } + argv_types = silc_calloc(argc, sizeof(SilcUInt32)); + if (!argv_types) { + silc_free(argv_lens); + silc_free(argv); + return NULL; + } + for (i = 0, k = 0; i < argc; i++) { x = va_arg(ap, unsigned char *); - x_len = va_arg(ap, unsigned int); + x_len = va_arg(ap, SilcUInt32); if (!x || !x_len) continue; - - argv[k] = silc_calloc(x_len + 1, sizeof(unsigned char)); - memcpy(argv[k], x, x_len); + + argv[k] = silc_memdup(x, x_len); + if (!argv[k]) + return NULL; argv_lens[k] = x_len; argv_types[k] = i + 1; k++; } args = silc_argument_payload_encode(k, argv, argv_lens, argv_types); - len = args->len; + len = silc_buffer_len(args); for (i = 0; i < k; i++) silc_free(argv[i]); @@ -115,8 +132,9 @@ SilcBuffer silc_notify_payload_encode(SilcNotifyType type, unsigned int argc, } len += 5; - buffer = silc_buffer_alloc(len); - silc_buffer_pull_tail(buffer, SILC_BUFFER_END(buffer)); + buffer = silc_buffer_alloc_size(len); + if (!buffer) + return NULL; silc_buffer_format(buffer, SILC_STR_UI_SHORT(type), SILC_STR_UI_SHORT(len), @@ -126,7 +144,7 @@ SilcBuffer silc_notify_payload_encode(SilcNotifyType type, unsigned int argc, if (k) { silc_buffer_pull(buffer, 5); silc_buffer_format(buffer, - SILC_STR_UI_XNSTRING(args->data, args->len), + SILC_STR_UI_XNSTRING(args->data, silc_buffer_len(args)), SILC_STR_END); silc_buffer_push(buffer, 5); silc_buffer_free(args); @@ -135,7 +153,37 @@ SilcBuffer silc_notify_payload_encode(SilcNotifyType type, unsigned int argc, return buffer; } -/* Free's notify payload */ +/* Same as above but takes argument from the `args' Argument Payload. */ + +SilcBuffer silc_notify_payload_encode_args(SilcNotifyType type, + SilcUInt32 argc, + SilcBuffer args) +{ + SilcBuffer buffer; + SilcUInt32 len; + + len = 5 + (args ? silc_buffer_len(args) : 0); + buffer = silc_buffer_alloc_size(len); + if (!buffer) + return NULL; + silc_buffer_format(buffer, + SILC_STR_UI_SHORT(type), + SILC_STR_UI_SHORT(len), + SILC_STR_UI_CHAR(argc), + SILC_STR_END); + + if (args) { + silc_buffer_pull(buffer, 5); + silc_buffer_format(buffer, + SILC_STR_UI_XNSTRING(args->data, silc_buffer_len(args)), + SILC_STR_END); + silc_buffer_push(buffer, 5); + } + + return buffer; +} + +/* Frees notify payload */ void silc_notify_payload_free(SilcNotifyPayload payload) { @@ -154,7 +202,7 @@ SilcNotifyType silc_notify_get_type(SilcNotifyPayload payload) /* Return argument nums */ -unsigned int silc_notify_get_arg_num(SilcNotifyPayload payload) +SilcUInt32 silc_notify_get_arg_num(SilcNotifyPayload payload) { return payload->argc; }