src/message_api.c
author vb
Tue, 03 Mar 2015 11:02:26 +0100
changeset 83 c5d82f11689a
parent 82 d0ad5655688a
child 85 6fd10806c186
permissions -rw-r--r--
...
vb@37
     1
#include "message_api.h"
vb@37
     2
#include "keymanagement.h"
vb@62
     3
#include "mime.h"
vb@37
     4
vb@37
     5
#include <assert.h>
vb@37
     6
#include <string.h>
vb@39
     7
#include <stdlib.h>
vb@39
     8
vb@76
     9
#define NOT_IMPLEMENTED assert(0); return PEP_UNKNOWN_ERROR;
vb@37
    10
vb@83
    11
static char * combine_short_and_long(const char *shortmsg, const char *longmsg)
vb@62
    12
{
vb@62
    13
    char * ptext;
vb@81
    14
vb@83
    15
    assert(shortmsg);
vb@83
    16
    assert(strcmp(shortmsg, "pEp") != 0);
vb@62
    17
vb@83
    18
    if (longmsg == NULL)
vb@63
    19
        longmsg = "";
vb@63
    20
vb@83
    21
    ptext = calloc(1, strlen(shortmsg) + strlen(longmsg) + 12);
vb@62
    22
    if (ptext == NULL)
vb@62
    23
        return NULL;
vb@62
    24
vb@62
    25
    strcpy(ptext, "subject: ");
vb@83
    26
    strcat(ptext, shortmsg);
vb@62
    27
    strcat(ptext, "\n\n");
vb@63
    28
    strcat(ptext, longmsg);
vb@62
    29
vb@62
    30
    return ptext;
vb@62
    31
}
vb@44
    32
vb@82
    33
static int seperate_short_and_long(const char *src, char **shortmsg, char **longmsg)
vb@82
    34
{
vb@82
    35
    char *_shortmsg = NULL;
vb@82
    36
    char *_longmsg = NULL;
vb@82
    37
vb@82
    38
    assert(src);
vb@82
    39
    assert(shortmsg);
vb@82
    40
    assert(longmsg);
vb@82
    41
vb@82
    42
    *shortmsg = NULL;
vb@82
    43
    *longmsg = NULL;
vb@82
    44
vb@82
    45
    if (strncmp(src, "subject: ", 9) == 0) {
vb@82
    46
        char *line_end = strchr(src, '\n');
vb@82
    47
        
vb@82
    48
        if (line_end == NULL) {
vb@82
    49
            _shortmsg = strdup(src + 9);
vb@82
    50
            if (_shortmsg == NULL)
vb@82
    51
                goto enomem;
vb@82
    52
            // _longmsg = NULL;
vb@82
    53
        }
vb@82
    54
        else {
vb@82
    55
            size_t n = line_end - src;
vb@82
    56
            if (*(line_end - 1) == '\r')
vb@82
    57
                _shortmsg = strndup(src + 9, n - 1);
vb@82
    58
            else
vb@82
    59
                _shortmsg = strndup(src + 9, n);
vb@82
    60
            if (_shortmsg == NULL)
vb@82
    61
                goto enomem;
vb@82
    62
            _longmsg = strdup(src + n);
vb@82
    63
            if (_longmsg == NULL)
vb@82
    64
                goto enomem;
vb@82
    65
        }
vb@82
    66
    }
vb@82
    67
    else {
vb@82
    68
        _shortmsg = strdup("pEp");
vb@82
    69
        if (_shortmsg == NULL)
vb@82
    70
            goto enomem;
vb@82
    71
        _longmsg = strdup(src);
vb@82
    72
        if (_longmsg == NULL)
vb@82
    73
            goto enomem;
vb@82
    74
    }
vb@82
    75
    
vb@82
    76
    *shortmsg = _shortmsg;
vb@82
    77
    *longmsg = _longmsg;
vb@82
    78
vb@82
    79
    return 0;
vb@82
    80
vb@82
    81
enomem:
vb@82
    82
    free(_shortmsg);
vb@82
    83
    free(_longmsg);
vb@82
    84
vb@82
    85
    return -1;
vb@82
    86
}
vb@82
    87
vb@81
    88
static message * clone_to_empty_message(const message * src)
vb@80
    89
{
vb@80
    90
    pEp_identity *from = NULL;
vb@80
    91
    identity_list *to = NULL;
vb@81
    92
vb@80
    93
    message * msg = NULL;
vb@80
    94
vb@81
    95
    assert(src);
vb@81
    96
    assert(src->from);
vb@81
    97
    assert(src->to);
vb@81
    98
vb@80
    99
    from = identity_dup(src->from);
vb@80
   100
    if (from == NULL)
vb@80
   101
        goto enomem;
vb@80
   102
vb@80
   103
    from->me = true;
vb@80
   104
vb@80
   105
    to = identity_list_dup(src->to);
vb@80
   106
    if (to == NULL)
vb@80
   107
        goto enomem;
vb@80
   108
vb@80
   109
    msg = new_message(src->dir, from, to, NULL);
vb@80
   110
    if (msg == NULL)
vb@80
   111
        goto enomem;
vb@80
   112
vb@82
   113
    msg->dir = src->dir;
vb@82
   114
vb@81
   115
    if (src->cc) {
vb@81
   116
        msg->cc = identity_list_dup(src->cc);
vb@81
   117
        if (msg->cc == NULL)
vb@81
   118
            goto enomem;
vb@81
   119
    }
vb@81
   120
vb@81
   121
    if (src->bcc) {
vb@81
   122
        msg->bcc = identity_list_dup(src->bcc);
vb@81
   123
        if (msg->bcc == NULL)
vb@81
   124
            goto enomem;
vb@81
   125
    }
vb@81
   126
vb@81
   127
    if (src->reply_to) {
vb@81
   128
        msg->reply_to = identity_dup(src->reply_to);
vb@81
   129
        if (msg->reply_to == NULL)
vb@81
   130
            goto enomem;
vb@81
   131
    }
vb@81
   132
vb@81
   133
    msg->sent = src->sent;
vb@81
   134
    msg->recv = src->recv;
vb@81
   135
vb@80
   136
    return msg;
vb@80
   137
vb@80
   138
enomem:
vb@81
   139
    if (msg) {
vb@81
   140
        free_message(msg);
vb@81
   141
    }
vb@81
   142
    else {
vb@81
   143
        free_identity(from);
vb@81
   144
        free_identity_list(to);
vb@81
   145
    }
vb@81
   146
vb@80
   147
    return NULL;
vb@80
   148
}
vb@80
   149
vb@48
   150
DYNAMIC_API PEP_STATUS encrypt_message(
vb@37
   151
        PEP_SESSION session,
vb@37
   152
        const message *src,
vb@37
   153
        stringlist_t * extra,
vb@38
   154
        message **dst,
vb@81
   155
        PEP_enc_format enc_format
vb@37
   156
    )
vb@37
   157
{
vb@37
   158
    PEP_STATUS status = PEP_STATUS_OK;
vb@63
   159
    message * msg = NULL;
vb@63
   160
    stringlist_t * keys = NULL;
vb@37
   161
vb@37
   162
    assert(session);
vb@37
   163
    assert(src);
vb@37
   164
    assert(dst);
vb@81
   165
    assert(enc_format >= PEP_enc_pieces);
vb@81
   166
vb@37
   167
    *dst = NULL;
vb@67
   168
vb@81
   169
    if (src->enc_format >= PEP_enc_pieces) {
vb@81
   170
        if (src->enc_format == enc_format) {
vb@81
   171
            msg = message_dup(src);
vb@81
   172
            if (msg == NULL)
vb@81
   173
                goto enomem;
vb@81
   174
            *dst = msg;
vb@81
   175
            return PEP_STATUS_OK;
vb@81
   176
        }
vb@81
   177
        else {
vb@81
   178
            // TODO: we don't re-encrypt yet
vb@81
   179
            NOT_IMPLEMENTED
vb@81
   180
        }
vb@67
   181
    }
vb@37
   182
vb@81
   183
    msg = clone_to_empty_message(src);
vb@80
   184
    if (msg == NULL)
vb@63
   185
        goto enomem;
vb@40
   186
vb@48
   187
    msg->enc_format = PEP_enc_pieces;
vb@37
   188
vb@80
   189
    status = myself(session, src->from);
vb@63
   190
    if (status != PEP_STATUS_OK)
vb@63
   191
        goto pep_error;
vb@37
   192
vb@80
   193
    keys = new_stringlist(src->from->fpr);
vb@63
   194
    if (keys == NULL)
vb@63
   195
        goto enomem;
vb@37
   196
vb@39
   197
    stringlist_t *_k = keys;
vb@39
   198
vb@39
   199
    if (extra) {
vb@39
   200
        _k = stringlist_append(_k, extra);
vb@63
   201
        if (_k == NULL)
vb@63
   202
            goto enomem;
vb@37
   203
    }
vb@39
   204
vb@39
   205
    bool dest_keys_found = false;
vb@37
   206
    identity_list * _il;
vb@80
   207
    for (_il = msg->to; _il && _il->ident; _il = _il->next) {
vb@63
   208
        PEP_STATUS status = update_identity(session, _il->ident);
vb@63
   209
        if (status != PEP_STATUS_OK)
vb@63
   210
            goto pep_error;
vb@63
   211
vb@37
   212
        if (_il->ident->fpr) {
vb@39
   213
            dest_keys_found = true;
vb@39
   214
            _k = stringlist_add(_k, _il->ident->fpr);
vb@63
   215
            if (_k == NULL)
vb@63
   216
                goto enomem;
vb@37
   217
        }
vb@37
   218
        else
vb@37
   219
            status = PEP_KEY_NOT_FOUND;
vb@37
   220
    }
vb@37
   221
vb@39
   222
    if (dest_keys_found) {
vb@38
   223
        char *ptext;
vb@37
   224
        char *ctext = NULL;
vb@37
   225
        size_t csize = 0;
vb@37
   226
vb@81
   227
        switch (enc_format) {
vb@44
   228
        case PEP_enc_MIME_multipart: {
vb@62
   229
            bool free_ptext = false;
vb@64
   230
vb@48
   231
            msg->enc_format = PEP_enc_MIME_multipart;
vb@37
   232
vb@63
   233
            if (src->shortmsg && strcmp(src->shortmsg, "pEp") != 0) {
vb@83
   234
                ptext = combine_short_and_long(src->shortmsg, src->longmsg);
vb@63
   235
                if (ptext == NULL)
vb@63
   236
                    goto enomem;
vb@62
   237
                free_ptext = true;
vb@62
   238
            }
vb@62
   239
            else if (src->longmsg) {
vb@62
   240
                ptext = src->longmsg;
vb@62
   241
            }
vb@62
   242
vb@81
   243
            if (src->enc_format == PEP_enc_none) {
vb@67
   244
                char *_ptext = ptext;
vb@67
   245
                status = mime_encode_text(_ptext, src->longmsg_formatted,
vb@67
   246
                        src->attachments, &ptext);
vb@67
   247
                assert(status == PEP_STATUS_OK);
vb@67
   248
                if (free_ptext)
vb@67
   249
                    free(_ptext);
vb@67
   250
                assert(ptext);
vb@67
   251
                if (ptext == NULL)
vb@67
   252
                    goto pep_error;
vb@67
   253
                free_ptext = true;
vb@67
   254
            }
vb@81
   255
            else if (src->enc_format == PEP_enc_none_MIME) {
vb@67
   256
                assert(src->longmsg);
vb@67
   257
                if (src->longmsg == NULL) {
vb@67
   258
                    status = PEP_ILLEGAL_VALUE;
vb@67
   259
                    goto pep_error;
vb@67
   260
                }
vb@67
   261
                ptext = src->longmsg;
vb@67
   262
            }
vb@67
   263
vb@67
   264
            status = encrypt_and_sign(session, keys, ptext, strlen(ptext),
vb@67
   265
                    &ctext, &csize);
vb@62
   266
            if (free_ptext)
vb@62
   267
                free(ptext);
vb@62
   268
            if (ctext) {
vb@62
   269
                msg->longmsg = strdup(ctext);
vb@64
   270
                if (msg->longmsg == NULL)
vb@63
   271
                    goto enomem;
vb@63
   272
            }
vb@63
   273
            else {
vb@63
   274
                goto pep_error;
vb@62
   275
            }
vb@62
   276
        }
vb@63
   277
        break;
vb@62
   278
vb@62
   279
        case PEP_enc_pieces:
vb@64
   280
            msg->enc_format = PEP_enc_pieces;
vb@64
   281
vb@67
   282
            // TODO: decoding MIME
vb@81
   283
            if (src->enc_format == PEP_enc_none_MIME) {
vb@67
   284
                NOT_IMPLEMENTED
vb@67
   285
            }
vb@67
   286
vb@63
   287
            if (src->shortmsg && strcmp(src->shortmsg, "pEp") != 0) {
vb@83
   288
                ptext = combine_short_and_long(src->shortmsg, src->longmsg);
vb@63
   289
                if (ptext == NULL)
vb@63
   290
                    goto enomem;
vb@63
   291
vb@39
   292
                status = encrypt_and_sign(session, keys, ptext, strlen(ptext),
vb@39
   293
                        &ctext, &csize);
vb@40
   294
                free(ptext);
vb@38
   295
                if (ctext) {
vb@40
   296
                    msg->longmsg = strdup(ctext);
vb@64
   297
                    if (msg->longmsg == NULL)
vb@63
   298
                        goto enomem;
vb@38
   299
                }
vb@38
   300
                else {
vb@63
   301
                    goto pep_error;
vb@38
   302
                }
vb@38
   303
            }
vb@38
   304
            else if (src->longmsg) {
vb@38
   305
                ptext = src->longmsg;
vb@39
   306
                status = encrypt_and_sign(session, keys, ptext, strlen(ptext),
vb@39
   307
                        &ctext, &csize);
vb@38
   308
                if (ctext) {
vb@40
   309
                    msg->longmsg = strdup(ctext);
vb@64
   310
                    if (msg->longmsg == NULL)
vb@63
   311
                        goto enomem;
vb@38
   312
                }
vb@38
   313
                else {
vb@63
   314
                    goto pep_error;
vb@38
   315
                }
vb@38
   316
            }
vb@63
   317
vb@63
   318
            if (msg->longmsg_formatted) {
vb@38
   319
                ptext = src->longmsg_formatted;
vb@39
   320
                status = encrypt_and_sign(session, keys, ptext, strlen(ptext),
vb@39
   321
                        &ctext, &csize);
vb@38
   322
                if (ctext) {
vb@40
   323
                    msg->longmsg_formatted = strdup(ctext);
vb@63
   324
                    if (msg->longmsg_formatted == NULL)
vb@63
   325
                        goto enomem;
vb@63
   326
                }
vb@63
   327
                else {
vb@63
   328
                    goto pep_error;
vb@63
   329
                }
vb@63
   330
            }
vb@63
   331
vb@63
   332
            if (src->attachments) {
vb@63
   333
                bloblist_t *_s;
vb@63
   334
                bloblist_t *_d = new_bloblist(NULL, 0, NULL, NULL);
vb@63
   335
                if (_d == NULL)
vb@63
   336
                    goto enomem;
vb@63
   337
vb@63
   338
                msg->attachments = _d;
vb@63
   339
                for (_s = src->attachments; _s && _s->data; _s = _s->next) {
vb@63
   340
                    int psize = _s->size;
vb@63
   341
                    ptext = _s->data;
vb@63
   342
                    status = encrypt_and_sign(session, keys, ptext, psize,
vb@63
   343
                            &ctext, &csize);
vb@63
   344
                    if (ctext) {
vb@63
   345
                        char * _c = strdup(ctext);
vb@63
   346
                        if (_c == NULL)
vb@63
   347
                            goto enomem;
vb@63
   348
vb@63
   349
                        _d = bloblist_add(_d, _c, csize, _s->mime_type,
vb@63
   350
                                _s->file_name);
vb@63
   351
                        if (_d == NULL)
vb@63
   352
                            goto enomem;
vb@63
   353
                    }
vb@63
   354
                    else {
vb@63
   355
                        goto pep_error;
vb@40
   356
                    }
vb@38
   357
                }
vb@38
   358
            }
vb@38
   359
            break;
vb@38
   360
vb@81
   361
        case PEP_enc_PEP:
vb@81
   362
            // TODO: implement
vb@81
   363
            NOT_IMPLEMENTED
vb@81
   364
vb@38
   365
        default:
vb@38
   366
            assert(0);
vb@63
   367
            status = PEP_ILLEGAL_VALUE;
vb@63
   368
            goto pep_error;
vb@37
   369
        }
vb@37
   370
    }
vb@37
   371
vb@37
   372
    free_stringlist(keys);
vb@63
   373
vb@64
   374
    if (msg->shortmsg == NULL)
vb@64
   375
        msg->shortmsg = strdup("pEp");
vb@64
   376
vb@63
   377
    *dst = msg;
vb@63
   378
    return PEP_STATUS_OK;
vb@63
   379
vb@63
   380
enomem:
vb@63
   381
    status = PEP_OUT_OF_MEMORY;
vb@63
   382
vb@63
   383
pep_error:
vb@63
   384
    free_stringlist(keys);
vb@63
   385
    free_message(msg);
vb@63
   386
vb@37
   387
    return status;
vb@37
   388
}
vb@37
   389
vb@48
   390
DYNAMIC_API PEP_STATUS decrypt_message(
vb@37
   391
        PEP_SESSION session,
vb@37
   392
        const message *src,
vb@81
   393
        message **dst,
vb@81
   394
        PEP_enc_format enc_format
vb@37
   395
    )
vb@37
   396
{
vb@37
   397
    PEP_STATUS status = PEP_STATUS_OK;
vb@73
   398
    message *msg = NULL;
vb@37
   399
vb@74
   400
    assert(session);
vb@74
   401
    assert(src);
vb@81
   402
    assert(src->dir == PEP_dir_incoming);
vb@74
   403
    assert(dst);
vb@81
   404
    assert(enc_format < PEP_enc_pieces);
vb@73
   405
vb@74
   406
    *dst = NULL;
vb@81
   407
 
vb@81
   408
    if (src->enc_format < PEP_enc_pieces) {
vb@81
   409
        if (enc_format == src->enc_format) {
vb@81
   410
            msg = message_dup(src);
vb@81
   411
            if (msg == NULL)
vb@81
   412
                goto enomem;
vb@81
   413
            *dst = msg;
vb@81
   414
            return PEP_STATUS_OK;
vb@81
   415
        }
vb@81
   416
        else {
vb@81
   417
            // TODO: we don't re-encode yet
vb@81
   418
            NOT_IMPLEMENTED
vb@81
   419
        }
vb@81
   420
    }
vb@81
   421
vb@81
   422
    msg = clone_to_empty_message(src);
vb@81
   423
    if (msg == NULL)
vb@81
   424
        goto enomem;
vb@81
   425
vb@81
   426
    switch (enc_format) {
vb@81
   427
        case PEP_enc_none:
vb@82
   428
            if (src->enc_format == PEP_enc_PEP) {
vb@82
   429
                // TODO: implement
vb@82
   430
                NOT_IMPLEMENTED
vb@82
   431
            }
vb@82
   432
vb@81
   433
            break;
vb@81
   434
vb@81
   435
        case PEP_enc_none_MIME:
vb@82
   436
            if (src->enc_format == PEP_enc_PEP) {
vb@82
   437
                // TODO: implement
vb@82
   438
                NOT_IMPLEMENTED
vb@82
   439
            }
vb@82
   440
vb@82
   441
            char *ctext = src->longmsg;
vb@82
   442
            size_t csize = strlen(src->longmsg);
vb@82
   443
            char *ptext;
vb@82
   444
            size_t psize;
vb@82
   445
            stringlist_t *keylist;
vb@82
   446
vb@82
   447
            status = decrypt_and_verify(session, ctext, csize, &ptext, &psize,
vb@82
   448
                    &keylist);
vb@82
   449
            if (ptext == NULL)
vb@82
   450
                goto pep_error;
vb@82
   451
vb@82
   452
            if (src->enc_format == PEP_enc_MIME_multipart) {
vb@82
   453
                if (src->shortmsg == NULL || strcmp(src->shortmsg, "pEp") == 0)
vb@82
   454
                {
vb@82
   455
                    char * shortmsg;
vb@82
   456
                    char * longmsg;
vb@82
   457
vb@82
   458
                    int r = seperate_short_and_long(ptext, &shortmsg,
vb@82
   459
                            &longmsg);
vb@82
   460
                    free(ptext);
vb@82
   461
                    if (r == -1)
vb@82
   462
                        goto enomem;
vb@82
   463
vb@82
   464
                    msg->shortmsg = shortmsg;
vb@82
   465
                    msg->longmsg = longmsg;
vb@82
   466
                }
vb@82
   467
                else {
vb@82
   468
                    msg->shortmsg = strdup(src->shortmsg);
vb@82
   469
                    if (msg->shortmsg == NULL)
vb@82
   470
                        goto enomem;
vb@82
   471
                    msg->longmsg = ptext;
vb@82
   472
                }
vb@82
   473
            }
vb@82
   474
            else {
vb@82
   475
                
vb@82
   476
            }
vb@81
   477
            break;
vb@81
   478
vb@81
   479
        default:
vb@81
   480
            assert(0);
vb@81
   481
            status = PEP_ILLEGAL_VALUE;
vb@81
   482
            goto pep_error;
vb@81
   483
    }
vb@74
   484
vb@74
   485
    *dst = msg;
vb@74
   486
    return PEP_STATUS_OK;
vb@73
   487
vb@73
   488
enomem:
vb@73
   489
    status = PEP_OUT_OF_MEMORY;
vb@73
   490
vb@73
   491
pep_error:
vb@73
   492
    free_message(msg);
vb@39
   493
vb@37
   494
    return status;
vb@37
   495
}
vb@37
   496