Fixed string formatting crashbug.
[silc.git] / lib / silccore / silcattrs.c
1 /*
2
3   silcattrs.c 
4
5   Author: Pekka Riikonen <priikone@silcnet.org>
6
7   Copyright (C) 2002 Pekka Riikonen
8
9   This program is free software; you can redistribute it and/or modify
10   it under the terms of the GNU General Public License as published by
11   the Free Software Foundation; version 2 of the License.
12
13   This program is distributed in the hope that it will be useful,
14   but WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16   GNU General Public License for more details.
17
18 */
19 /* Implementation of Attribute Payload routines */
20 /* $Id$ */
21
22 #include "silcincludes.h"
23 #include "silcattrs.h"
24
25 /******************************************************************************
26
27                              Attribute Payload
28
29 ******************************************************************************/
30
31 struct SilcAttributePayloadStruct {
32   SilcAttribute attribute;
33   SilcAttributeFlags flags;
34   SilcUInt16 data_len;
35   unsigned char *data;
36 };
37
38 /* Internal routine for encoding a attribute */
39
40 static unsigned char *
41 silc_attribute_payload_encode_int(SilcAttribute attribute,
42                                   SilcAttributeFlags flags,
43                                   void *object,
44                                   SilcUInt32 object_size,
45                                   SilcUInt32 *ret_len)
46 {
47   SilcBuffer tmpbuf = NULL;
48   unsigned char tmp[4], *str = NULL, *ret;
49   int len;
50
51   /* Encode according to attribute type */
52   if (flags & SILC_ATTRIBUTE_FLAG_VALID) {
53     if (!object && !object_size)
54       return NULL;
55
56     switch (attribute) {
57
58     case SILC_ATTRIBUTE_USER_INFO:
59       {
60         SilcVCard vcard = object;
61         if (object_size != sizeof(*vcard))
62           return NULL;
63         str = silc_vcard_encode(vcard, &object_size);
64         if (!str)
65           return NULL;
66         object = str;
67       }
68       break;
69
70     case SILC_ATTRIBUTE_SERVICE:
71       {
72         SilcAttributeObjService *service = object;
73         if (object_size != sizeof(*service))
74           return NULL;
75         len = strlen(service->address);
76         str = silc_malloc(7 + len);
77         if (!str)
78           return NULL;
79         SILC_PUT32_MSB(service->port, str);
80         SILC_PUT16_MSB(len, str + 4);
81         memcpy(str + 6, service->address, len);
82         str[6 + len] = service->status;
83         object = str;
84         object_size = 7 + len;
85       }
86       break;
87
88     case SILC_ATTRIBUTE_STATUS_MOOD:
89     case SILC_ATTRIBUTE_PREFERRED_CONTACT:
90       {
91         SilcUInt32 mask = (SilcUInt32)object;
92         if (object_size != sizeof(SilcUInt32))
93           return NULL;
94         SILC_PUT32_MSB(mask, tmp);
95         object = tmp;
96         object_size = sizeof(SilcUInt32);
97       }
98       break;
99
100     case SILC_ATTRIBUTE_STATUS_FREETEXT:
101     case SILC_ATTRIBUTE_PREFERRED_LANGUAGE:
102     case SILC_ATTRIBUTE_TIMEZONE:
103       {
104         unsigned char *string = object;
105         str = silc_malloc(2 + object_size);
106         if (!str)
107           return NULL;
108         SILC_PUT16_MSB(object_size, str);
109         memcpy(str + 2, string, object_size);
110         object = str;
111         object_size += 2;
112       }
113       break;
114
115     case SILC_ATTRIBUTE_STATUS_MESSAGE:
116     case SILC_ATTRIBUTE_EXTENSION:
117       {
118         SilcAttributeObjMime *mime = object;
119         if (object_size != sizeof(*mime))
120           return NULL;
121         object = (void *)mime->mime;
122         object_size = mime->mime_len;
123       }
124       break;
125
126     case SILC_ATTRIBUTE_GEOLOCATION:
127       {
128         SilcAttributeObjGeo *geo = object;
129         int len1, len2, len3, len4;
130         if (object_size != sizeof(*geo))
131           return NULL;
132         len1 = (geo->longitude ? strlen(geo->longitude) : 0);
133         len2 = (geo->latitude  ? strlen(geo->latitude)  : 0);
134         len3 = (geo->altitude  ? strlen(geo->altitude)  : 0);
135         len4 = (geo->accuracy  ? strlen(geo->accuracy)  : 0);
136         if (len1 + len2 + len3 + len4 == 0)
137           return NULL;
138         len = len1 + len2 + len3 + len4;
139         tmpbuf = silc_buffer_alloc_size(8 + len);
140         if (!tmpbuf)
141           return NULL;
142         silc_buffer_format(tmpbuf,
143                            SILC_STR_UI_SHORT(len1),
144                            SILC_STR_UI16_STRING(len1 ? geo->longitude : ""),
145                            SILC_STR_UI_SHORT(len2),
146                            SILC_STR_UI16_STRING(len ? geo->latitude : ""),
147                            SILC_STR_UI_SHORT(len3),
148                            SILC_STR_UI16_STRING(len3 ? geo->altitude : ""),
149                            SILC_STR_UI_SHORT(len4),
150                            SILC_STR_UI16_STRING(len4 ? geo->accuracy : ""),
151                            SILC_STR_END);
152         object = tmpbuf->data;
153         object_size = tmpbuf->len;
154       }
155       break;
156
157     case SILC_ATTRIBUTE_DEVICE_INFO:
158       {
159         SilcAttributeObjDevice *dev = object;
160         int len1, len2, len3, len4;
161         if (object_size != sizeof(*dev))
162           return NULL;
163         len1 = (dev->manufacturer ? strlen(dev->manufacturer) : 0);
164         len2 = (dev->version      ? strlen(dev->version)      : 0);
165         len3 = (dev->model        ? strlen(dev->model)        : 0);
166         len4 = (dev->language     ? strlen(dev->language)     : 0);
167         if (len1 + len2 + len3 + len4 == 0)
168           return NULL;
169         len = len1 + len2 + len3 + len4;
170         tmpbuf = silc_buffer_alloc_size(4 + 8 + len);
171         if (!tmpbuf)
172           return NULL;
173         silc_buffer_format(tmpbuf,
174                            SILC_STR_UI_INT(dev->type),
175                            SILC_STR_UI_SHORT(len1),
176                            SILC_STR_UI16_STRING(len ? dev->manufacturer : ""),
177                            SILC_STR_UI_SHORT(len2),
178                            SILC_STR_UI16_STRING(len ? dev->version : ""),
179                            SILC_STR_UI_SHORT(len3),
180                            SILC_STR_UI16_STRING(len3 ? dev->model : ""),
181                            SILC_STR_UI_SHORT(len4),
182                            SILC_STR_UI16_STRING(len4 ? dev->language : ""),
183                            SILC_STR_END);
184         object = tmpbuf->data;
185         object_size = tmpbuf->len;
186       }
187       break;
188
189     case SILC_ATTRIBUTE_USER_PUBLIC_KEY:
190     case SILC_ATTRIBUTE_SERVER_PUBLIC_KEY:
191       {
192         SilcAttributeObjPk *pk = object;
193         if (object_size != sizeof(*pk))
194           return NULL;
195         len = (pk->type ? strlen(pk->type) : 0);
196         tmpbuf = silc_buffer_alloc_size(2 + len + pk->data_len);
197         if (!tmpbuf)
198           return NULL;
199         silc_buffer_format(tmpbuf,
200                            SILC_STR_UI_SHORT(len),
201                            SILC_STR_UI16_STRING(pk->type),
202                            SILC_STR_UI_XNSTRING(pk->data, pk->data_len),
203                            SILC_STR_END);
204         object = tmpbuf->data;
205         object_size = tmpbuf->len;
206       }
207       break;
208
209     case SILC_ATTRIBUTE_USER_DIGITAL_SIGNATURE:
210     case SILC_ATTRIBUTE_SERVER_DIGITAL_SIGNATURE:
211       {
212         SilcAttributeObjPk *pk = object;
213         if (object_size != sizeof(*pk))
214           return NULL;
215         object = pk->data;
216         object_size = pk->data_len;
217       }
218       break;
219
220     default:
221       return NULL;
222       break;
223     }
224
225     ret = silc_memdup(object, object_size);
226
227     if (tmpbuf)
228       silc_buffer_free(tmpbuf);
229     silc_free(str);
230
231     if (ret_len)
232       *ret_len = object_size;
233
234     return ret;
235   }
236
237   return NULL;
238 }
239
240 /* Allocates attribute payload and encodes the attribute there */
241
242 SilcAttributePayload silc_attribute_payload_alloc(SilcAttribute attribute,
243                                                   SilcAttributeFlags flags,
244                                                   void *object,
245                                                   SilcUInt32 object_size)
246 {
247   SilcAttributePayload attr;
248
249   attr = silc_calloc(1, sizeof(*attr));
250   if (!attr)
251     return NULL;
252
253   attr->attribute = attribute;
254   attr->flags = flags;
255   attr->data =
256     silc_attribute_payload_encode_int(attribute, flags, object,
257                                       object_size,
258                                       (SilcUInt32 *)&attr->data_len);
259   if (!attr->data) {
260     silc_free(attr);
261     return NULL;
262   }
263
264   return attr;
265 }
266
267 /* Parse list of payloads */
268
269 SilcDList silc_attribute_payload_parse(const unsigned char *payload,
270                                        SilcUInt32 payload_len)
271 {
272   SilcBufferStruct buffer;
273   SilcDList list;
274   SilcAttributePayload newp;
275   int len, ret;
276
277   SILC_LOG_DEBUG(("Parsing Attribute Payload list"));
278
279   silc_buffer_set(&buffer, (unsigned char *)payload, payload_len);
280   list = silc_dlist_init();
281
282   while (buffer.len) {
283     newp = silc_calloc(1, sizeof(*newp));
284     if (!newp)
285       goto err;
286     ret = silc_buffer_unformat(&buffer,
287                                SILC_STR_UI_CHAR(&newp->attribute),
288                                SILC_STR_UI_CHAR(&newp->flags),
289                                SILC_STR_UI16_NSTRING_ALLOC(&newp->data, 
290                                                            &newp->data_len),
291                                SILC_STR_END);
292     if (ret == -1)
293       goto err;
294
295     if (newp->data_len > buffer.len) {
296       SILC_LOG_ERROR(("Incorrect attribute payload in list"));
297       goto err;
298     }
299
300     len = 4 + newp->data_len;
301     if (buffer.len < len)
302       break;
303     silc_buffer_pull(&buffer, len);
304
305     silc_dlist_add(list, newp);
306   }
307   
308   return list;
309
310  err:
311   silc_attribute_payload_list_free(list);
312   return NULL;
313 }
314
315 /* Encode one attribute payload to buffer */
316
317 SilcBuffer silc_attribute_payload_encode(SilcBuffer attrs,
318                                          SilcAttribute attribute,
319                                          SilcAttributeFlags flags,
320                                          void *object,
321                                          SilcUInt32 object_size)
322 {
323   object = silc_attribute_payload_encode_int(attribute, flags, object,
324                                              object_size, &object_size);
325   attrs = silc_attribute_payload_encode_data(attrs, attribute, flags,
326                                              (const unsigned char *)object,
327                                              object_size);
328   silc_free(object);
329   return attrs;
330 }
331
332 /* Encoded the attribute data directly to buffer */
333
334 SilcBuffer silc_attribute_payload_encode_data(SilcBuffer attrs,
335                                               SilcAttribute attribute,
336                                               SilcAttributeFlags flags,
337                                               const unsigned char *data,
338                                               SilcUInt32 data_len)
339 {
340   SilcBuffer buffer = attrs;
341   int len;
342
343   len = 4 + data_len;
344   buffer = silc_buffer_realloc(buffer,
345                                (buffer ? buffer->truelen + len : len));
346   if (!buffer)
347     return NULL;
348   silc_buffer_pull(buffer, buffer->len);
349   silc_buffer_pull_tail(buffer, len);
350   silc_buffer_format(buffer, 
351                      SILC_STR_UI_CHAR(attribute),
352                      SILC_STR_UI_CHAR(flags),
353                      SILC_STR_UI_SHORT((SilcUInt16)data_len),
354                      SILC_STR_UI_XNSTRING(data, data_len),
355                      SILC_STR_END);
356   silc_buffer_push(buffer, buffer->data - buffer->head);
357
358   return buffer;
359 }
360
361 /* Free Attribute Payload */
362
363 void silc_attribute_payload_free(SilcAttributePayload payload)
364 {
365   silc_free(payload->data);
366   silc_free(payload);
367 }
368
369 /* Free's list of Attribute Payloads */
370
371 void silc_attribute_payload_list_free(SilcDList list)
372 {
373   SilcAttributePayload entry;
374
375   silc_dlist_start(list);
376   while ((entry = silc_dlist_get(list)) != SILC_LIST_END) {
377     silc_attribute_payload_free(entry);
378     silc_dlist_del(list, entry);
379   }
380
381   silc_dlist_uninit(list);
382 }
383
384 /* Return attribute type */
385
386 SilcAttribute silc_attribute_get_attribute(SilcAttributePayload payload)
387 {
388   return payload->attribute;
389 }
390
391 /* Return attribute flags */
392
393 SilcAttributeFlags silc_attribute_get_flags(SilcAttributePayload payload)
394 {
395   return payload->flags;
396 }
397
398 /* Return attribute data from the payload */
399
400 const unsigned char *silc_attribute_get_data(SilcAttributePayload payload,
401                                              SilcUInt32 *data_len)
402 {
403   if (data_len)
404     *data_len = payload->data_len;
405   return (const unsigned char *)payload->data;
406 }
407
408 /* Construct digital signature verification data */
409
410 unsigned char *silc_attribute_get_verify_data(SilcDList attrs,
411                                               bool server_verification,
412                                               SilcUInt32 *data_len)
413 {
414   SilcAttributePayload attr;
415   SilcBufferStruct buffer;
416   unsigned char *data = NULL;
417   SilcUInt32 len = 0;
418
419   silc_dlist_start(attrs);
420   while ((attr = silc_dlist_get(attrs)) != SILC_LIST_END) {
421     switch (attr->attribute) {
422     case SILC_ATTRIBUTE_SERVER_DIGITAL_SIGNATURE:
423       /* Server signature is never part of the verification data */
424       break;
425
426     case SILC_ATTRIBUTE_USER_DIGITAL_SIGNATURE:
427       /* For user signature verification this is not part of the data */
428       if (!server_verification)
429         break;
430
431       /* Fallback, for server signature verification, user digital signature
432          is part of verification data. */
433
434     default:
435       /* All other data is part of the verification data */
436       data = silc_realloc(data, sizeof(*data) * (4 + attr->data_len + len));
437       if (!data)
438         return NULL;
439       silc_buffer_set(&buffer, data + len, 4 + attr->data_len);
440       silc_buffer_format(&buffer, 
441                          SILC_STR_UI_CHAR(attr->attribute),
442                          SILC_STR_UI_CHAR(attr->flags),
443                          SILC_STR_UI_SHORT(attr->data_len),
444                          SILC_STR_UI_XNSTRING(attr->data, attr->data_len),
445                          SILC_STR_END);
446       len += 4 + attr->data_len;
447       break;
448     }
449   }
450
451   if (data_len)
452     *data_len = len;
453
454   return data;
455 }
456
457 /* Return parsed attribute object */
458
459 bool silc_attribute_get_object(SilcAttributePayload payload,
460                                void *object, SilcUInt32 object_size)
461 {
462   SilcUInt16 len;
463   bool ret = FALSE;
464
465   if (!object || payload->flags & SILC_ATTRIBUTE_FLAG_INVALID)
466     return FALSE;
467
468   switch (payload->attribute) {
469   case SILC_ATTRIBUTE_USER_INFO:
470     {
471       SilcVCard vcard = object;
472       if (object_size != sizeof(*vcard))
473         break;
474       if (!silc_vcard_decode(payload->data, payload->data_len, vcard))
475         break;
476       ret = TRUE;
477     }
478     break;
479
480   case SILC_ATTRIBUTE_SERVICE:
481     {
482       SilcAttributeObjService *service = object;
483       if (object_size != sizeof(*service))
484         break;
485       if (payload->data_len < 7)
486         break;
487       SILC_GET32_MSB(service->port, payload->data);
488       SILC_GET16_MSB(len, payload->data + 4);
489       if (payload->data_len < 7 + len)
490         break;
491       memcpy(service->address, payload->data + 6,
492              (len < sizeof(service->address) - 1 ? len :
493               sizeof(service->address) - 1));
494       service->status = payload->data[6 + len] ? TRUE : FALSE;
495       ret = TRUE;
496     }
497     break;
498
499   case SILC_ATTRIBUTE_STATUS_MOOD:
500   case SILC_ATTRIBUTE_PREFERRED_CONTACT:
501     {
502       SilcUInt32 *mask = (SilcUInt32 *)object;
503       if (object_size != sizeof(SilcUInt32))
504         break;
505       if (payload->data_len < 4)
506         break;
507       SILC_GET32_MSB(*mask, payload->data);
508       ret = TRUE;
509     }
510     break;
511
512   case SILC_ATTRIBUTE_STATUS_FREETEXT:
513   case SILC_ATTRIBUTE_PREFERRED_LANGUAGE:
514   case SILC_ATTRIBUTE_TIMEZONE:
515     {
516       char *string = object;
517       if (payload->data_len < 2)
518         break;
519       SILC_GET16_MSB(len, payload->data);
520       if (payload->data_len < 2 + len)
521         break;
522       if (object_size < len)
523         break;
524       memcpy(string, payload->data + 2, len);
525       ret = TRUE;
526     }
527     break;
528
529   case SILC_ATTRIBUTE_STATUS_MESSAGE:
530   case SILC_ATTRIBUTE_EXTENSION:
531     {
532       SilcAttributeObjMime *mime = object;
533       if (object_size != sizeof(*mime))
534         break;
535       mime->mime = (const unsigned char *)payload->data;
536       mime->mime_len = payload->data_len;
537       ret = TRUE;
538     }
539     break;
540
541   case SILC_ATTRIBUTE_GEOLOCATION:
542     {
543       SilcAttributeObjGeo *geo = object;
544       SilcBufferStruct buffer;
545       int res;
546       if (object_size != sizeof(*geo))
547         break;
548       silc_buffer_set(&buffer, (unsigned char *)payload->data,
549                       payload->data_len);
550       res = silc_buffer_unformat(&buffer,
551                                  SILC_STR_UI16_STRING_ALLOC(&geo->longitude),
552                                  SILC_STR_UI16_STRING_ALLOC(&geo->latitude),
553                                  SILC_STR_UI16_STRING_ALLOC(&geo->altitude),
554                                  SILC_STR_UI16_STRING_ALLOC(&geo->accuracy),
555                                  SILC_STR_END);
556       if (res == -1)
557         break;
558       ret = TRUE;
559     }
560     break;
561
562   case SILC_ATTRIBUTE_DEVICE_INFO:
563     {
564       SilcAttributeObjDevice *dev = object;
565       SilcBufferStruct buffer;
566       SilcUInt32 type;
567       int res;
568       if (object_size != sizeof(*dev))
569         break;
570       silc_buffer_set(&buffer, (unsigned char *)payload->data,
571                       payload->data_len);
572       res =
573         silc_buffer_unformat(&buffer,
574                              SILC_STR_UI_INT(&type),
575                              SILC_STR_UI16_STRING_ALLOC(&dev->manufacturer),
576                              SILC_STR_UI16_STRING_ALLOC(&dev->version),
577                              SILC_STR_UI16_STRING_ALLOC(&dev->model),
578                              SILC_STR_UI16_STRING_ALLOC(&dev->language),
579                              SILC_STR_END);
580       if (res == -1)
581         break;
582       dev->type = type;
583       ret = TRUE;
584     }
585     break;
586
587   case SILC_ATTRIBUTE_USER_PUBLIC_KEY:
588   case SILC_ATTRIBUTE_SERVER_PUBLIC_KEY:
589     {
590       SilcAttributeObjPk *pk = object;
591       SilcBufferStruct buffer;
592       int res;
593       if (object_size != sizeof(*pk))
594         break;
595       silc_buffer_set(&buffer, (unsigned char *)payload->data,
596                       payload->data_len);
597       res =
598         silc_buffer_unformat(&buffer,
599                              SILC_STR_UI16_NSTRING_ALLOC(&pk->type, &len),
600                              SILC_STR_END);
601       if (res == -1)
602         break;
603       pk->data = silc_memdup(payload->data + 2 + len,
604                              payload->data_len - 2 - len);
605       pk->data_len = payload->data_len - 2 - len;
606       ret = TRUE;
607     }
608     break;
609
610   case SILC_ATTRIBUTE_USER_DIGITAL_SIGNATURE:
611   case SILC_ATTRIBUTE_SERVER_DIGITAL_SIGNATURE:
612     {
613       SilcAttributeObjPk *pk = object;
614       if (object_size != sizeof(*pk))
615         break;
616       pk->type = NULL;
617       pk->data = silc_memdup(payload->data, payload->data_len);
618       pk->data_len = payload->data_len;
619       ret = TRUE;
620     }
621     break;
622
623   default:
624     break;
625   }
626
627   return ret;
628 }