2002-02-02 Marcus Brinkmann <marcus@g10code.de>
[gpgme.git] / gpgme / import.c
1 /* import.c -  encrypt functions
2  *      Copyright (C) 2000 Werner Koch (dd9jn)
3  *      Copyright (C) 2001, 2002 g10 Code GmbH
4  *
5  * This file is part of GPGME.
6  *
7  * GPGME is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * GPGME is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
20  */
21
22 #include <config.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <assert.h>
27
28 #include "util.h"
29 #include "context.h"
30 #include "ops.h"
31
32
33 struct import_result_s
34 {
35   GpgmeData xmlinfo;
36 };
37
38
39 void
40 _gpgme_release_import_result (ImportResult result)
41 {
42   if (!result)
43     return;
44   gpgme_data_release (result->xmlinfo);
45   xfree (result);
46 }
47
48
49 /* Parse the args and append the information to the XML structure in
50    the data buffer.  With args of NULL the xml structure is
51    closed.  */
52 static void
53 append_xml_impinfo (GpgmeData *rdh, GpgStatusCode code, char *args)
54 {
55 #define MAX_IMPORTED_FIELDS 14
56   static char *imported_fields[MAX_IMPORTED_FIELDS]
57     = { "keyid", "username", 0 };
58   static char *import_res_fields[MAX_IMPORTED_FIELDS]
59     = { "count", "no_user_id", "imported", "imported_rsa",
60         "unchanged", "n_uids", "n_subk", "n_sigs", "s_sigsn_revoc",
61         "sec_read", "sec_imported", "sec_dups", "skipped_new", 0 };
62   char *field[MAX_IMPORTED_FIELDS];
63   char **field_name = 0;
64   GpgmeData dh;
65   int i;
66
67   /* Verify that we can use the args.  */
68   if (code != STATUS_EOF)
69     {
70       if (!args)
71         return;
72
73       if (code == STATUS_IMPORTED)
74         field_name = imported_fields;
75       else if (code == STATUS_IMPORT_RES)
76         field_name = import_res_fields;
77       else
78         return;
79
80       for (i = 0; field_name[i]; i++)
81         {
82           field[i] = args;
83           if (field_name[i + 1])
84             {
85               args = strchr (args, ' ');
86               if (!args)
87                 return;  /* Invalid line.  */
88               *args++ = '\0';
89             }
90         }
91     }
92
93   /* Initialize the data buffer if necessary.  */
94   if (!*rdh)
95     {
96       if (gpgme_data_new (rdh))
97         return; /* FIXME: We are ignoring out-of-core.  */
98       dh = *rdh;
99       _gpgme_data_append_string (dh, "<GnupgOperationInfo>\n");
100     }
101   else
102     dh = *rdh;
103     
104   if (code == STATUS_EOF)
105     {
106       /* Just close the XML containter.  */
107       _gpgme_data_append_string (dh, "</GnupgOperationInfo>\n");
108     }
109   else
110     {
111       if (code == STATUS_IMPORTED)
112         _gpgme_data_append_string (dh, "  <import>\n");
113       else if (code == STATUS_IMPORT_RES)
114         _gpgme_data_append_string (dh, "  <importResult>\n");
115
116       for (i = 0; field_name[i]; i++)
117         {
118           _gpgme_data_append_string (dh, "    <");
119           _gpgme_data_append_string (dh, field_name[i]);
120           _gpgme_data_append_string (dh, ">");
121           _gpgme_data_append_string (dh, field[i]);
122           _gpgme_data_append_string (dh, "</");
123           _gpgme_data_append_string (dh, field_name[i]);
124           _gpgme_data_append_string (dh, ">\n");
125         }
126
127       if (code == STATUS_IMPORTED)
128         _gpgme_data_append_string (dh, "  </import>\n");
129       else if (code == STATUS_IMPORT_RES)
130         _gpgme_data_append_string (dh, "  </importResult>\n");
131     }
132 }
133
134
135 static void
136 import_status_handler (GpgmeCtx ctx, GpgStatusCode code, char *args)
137 {
138   if (ctx->error)
139     return;
140   test_and_allocate_result (ctx, import);
141
142   switch (code)
143     {
144     case STATUS_EOF:
145       if (ctx->result.import->xmlinfo)
146         {
147           append_xml_impinfo (&ctx->result.import->xmlinfo, code, NULL);
148           _gpgme_set_op_info (ctx, ctx->result.import->xmlinfo);
149           ctx->result.import->xmlinfo = NULL;
150         }
151       /* XXX Calculate error value.  */
152       break;
153
154     case STATUS_IMPORTED:
155     case STATUS_IMPORT_RES:
156       append_xml_impinfo (&ctx->result.import->xmlinfo, code, args);
157       break;
158
159     default:
160       break;
161     }
162 }
163
164
165 GpgmeError
166 gpgme_op_import_start (GpgmeCtx ctx, GpgmeData keydata)
167 {
168   int err = 0;
169
170   fail_on_pending_request (ctx);
171   ctx->pending = 1;
172
173   _gpgme_engine_release (ctx->engine);
174   ctx->engine = NULL;
175   err = _gpgme_engine_new (ctx->use_cms ? GPGME_PROTOCOL_CMS
176                            : GPGME_PROTOCOL_OpenPGP, &ctx->engine);
177   if (err)
178     goto leave;
179
180   /* Check the supplied data */
181   if (gpgme_data_get_type (keydata) == GPGME_DATA_TYPE_NONE)
182     {
183       err = mk_error (No_Data);
184       goto leave;
185     }
186   _gpgme_data_set_mode (keydata, GPGME_DATA_MODE_OUT);
187
188   _gpgme_engine_set_status_handler (ctx->engine, import_status_handler, ctx);
189   _gpgme_engine_set_verbosity (ctx->engine, ctx->verbosity);
190
191   _gpgme_engine_op_import (ctx->engine, keydata);
192
193   if (!err)
194     err = _gpgme_engine_start (ctx->engine, ctx);
195
196  leave:
197   if (err)
198     {
199       ctx->pending = 0;
200       _gpgme_engine_release (ctx->engine);
201       ctx->engine = NULL;
202     }
203   return err;
204 }
205
206
207 /**
208  * gpgme_op_import:
209  * @c: Context 
210  * @keydata: Data object
211  * 
212  * Import all key material from @keydata into the key database.
213  * 
214  * Return value: o on success or an error code.
215  **/
216 GpgmeError
217 gpgme_op_import (GpgmeCtx ctx, GpgmeData keydata)
218 {
219   GpgmeError err = gpgme_op_import_start (ctx, keydata);
220   if (!err)
221     {
222       gpgme_wait (ctx, 1);
223       err = ctx->error;
224     }
225   return err;
226 }