2002-01-30 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->out_of_core)
139     return;
140   if (!ctx->result.import)
141     {
142       ctx->result.import = xtrycalloc (1, sizeof *ctx->result.import);
143       if (!ctx->result.import)
144         {
145           ctx->out_of_core = 1;
146           return;
147         }
148     }
149
150   switch (code)
151     {
152     case STATUS_EOF:
153       if (ctx->result.import->xmlinfo)
154         {
155           append_xml_impinfo (&ctx->result.import->xmlinfo, code, NULL);
156           _gpgme_set_op_info (ctx, ctx->result.import->xmlinfo);
157           ctx->result.import->xmlinfo = NULL;
158         }
159       break;
160
161     case STATUS_IMPORTED:
162     case STATUS_IMPORT_RES:
163       append_xml_impinfo (&ctx->result.import->xmlinfo, code, args);
164       break;
165
166     default:
167       break;
168     }
169 }
170
171
172 GpgmeError
173 gpgme_op_import_start (GpgmeCtx ctx, GpgmeData keydata)
174 {
175   int err = 0;
176
177   fail_on_pending_request (ctx);
178   ctx->pending = 1;
179
180   _gpgme_engine_release (ctx->engine);
181   ctx->engine = NULL;
182   err = _gpgme_engine_new (ctx->use_cms ? GPGME_PROTOCOL_CMS
183                            : GPGME_PROTOCOL_OpenPGP, &ctx->engine);
184   if (err)
185     goto leave;
186
187   /* Check the supplied data */
188   if (gpgme_data_get_type (keydata) == GPGME_DATA_TYPE_NONE)
189     {
190       err = mk_error (No_Data);
191       goto leave;
192     }
193   _gpgme_data_set_mode (keydata, GPGME_DATA_MODE_OUT);
194
195   _gpgme_engine_set_status_handler (ctx->engine, import_status_handler, ctx);
196   _gpgme_engine_set_verbosity (ctx->engine, ctx->verbosity);
197
198   _gpgme_engine_op_import (ctx->engine, keydata);
199
200   if (!err)
201     err = _gpgme_engine_start (ctx->engine, ctx);
202
203  leave:
204   if (err)
205     {
206       ctx->pending = 0;
207       _gpgme_engine_release (ctx->engine);
208       ctx->engine = NULL;
209     }
210   return err;
211 }
212
213
214 /**
215  * gpgme_op_import:
216  * @c: Context 
217  * @keydata: Data object
218  * 
219  * Import all key material from @keydata into the key database.
220  * 
221  * Return value: o on success or an error code.
222  **/
223 GpgmeError
224 gpgme_op_import (GpgmeCtx ctx, GpgmeData keydata)
225 {
226   GpgmeError err = gpgme_op_import_start (ctx, keydata);
227   if (!err)
228     gpgme_wait (ctx, 1);
229   return err;
230 }