1 /*
2 * Copyright 2014-2022 The GmSSL Project. All Rights Reserved.
3 *
4 * Licensed under the Apache License, Version 2.0 (the License); you may
5 * not use this file except in compliance with the License.
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 */
9
10
11 #include <stdio.h>
12 #include <string.h>
13 #include <stdlib.h>
14 #include <gmssl/pem.h>
15 #include <gmssl/error.h>
16
17
pem_write(FILE * fp,const char * name,const uint8_t * data,size_t datalen)18 int pem_write(FILE *fp, const char *name, const uint8_t *data, size_t datalen)
19 {
20 int ret = 0;
21 BASE64_CTX ctx;
22 uint8_t b64[datalen * 2];
23 int len;
24
25 base64_encode_init(&ctx);
26 base64_encode_update(&ctx, data, (int)datalen, b64, &len);
27 base64_encode_finish(&ctx, b64 + len, &len);
28
29 ret += fprintf(fp, "-----BEGIN %s-----\n", name);
30 ret += fprintf(fp, "%s", (char *)b64);
31 ret += fprintf(fp, "-----END %s-----\n", name);
32 //return ret;
33 return 1;
34 }
35
pem_read(FILE * fp,const char * name,uint8_t * data,size_t * datalen,size_t maxlen)36 int pem_read(FILE *fp, const char *name, uint8_t *data, size_t *datalen, size_t maxlen)
37 {
38 char line[80];
39 char begin_line[80];
40 char end_line[80];
41 int len;
42 BASE64_CTX ctx;
43
44 snprintf(begin_line, sizeof(begin_line), "-----BEGIN %s-----\n", name);
45 snprintf(end_line, sizeof(end_line), "-----END %s-----\n", name);
46
47 if (feof(fp)) {
48 return 0;
49 }
50
51 if (!fgets(line, sizeof(line), fp)) {
52 if (feof(fp))
53 return 0;
54 else {
55 error_print();
56 return -1;
57 }
58 }
59
60 if (strcmp(line, begin_line) != 0) {
61 // FIXME: 这里是不是应该容忍一些错误呢？
62
63 fprintf(stderr, "%s %d: %s\n", __FILE__, __LINE__, line);
64 fprintf(stderr, "%s %d: %s\n", __FILE__, __LINE__, begin_line);
65
66 error_print();
67 return -1;
68 }
69
70 *datalen = 0;
71
72 base64_decode_init(&ctx);
73
74 for (;;) {
75 if (!fgets(line, sizeof(line), fp)) {
76 error_print();
77 return -1;
78 }
79 if (strcmp(line, end_line) == 0) {
80 break;
81 }
82
83 base64_decode_update(&ctx, (uint8_t *)line, strlen(line), data, &len);
84 data += len;
85 *datalen += len;
86 }
87
88 base64_decode_finish(&ctx, data, &len);
89 *datalen += len;
90 return 1;
91 }
92