2 * Copyright (C) 2012-2013 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 #include <sys/socket.h>
25 #include <netinet/in.h>
26 #include <arpa/inet.h>
28 #include <gnutls/gnutls.h>
31 #include "rnet_message.h"
32 #include "rnet_encode.h"
34 static size_t chars2len (unsigned char buf[2]) {
35 return (buf[0] << 8 | buf[1]);
38 static void * get_creds(char *certfile)
40 static gnutls_certificate_credentials_t cred;
41 gnutls_certificate_allocate_credentials(&cred);
42 gnutls_certificate_set_x509_trust_file(cred, certfile,
47 static void session_new(gnutls_session_t *session)
50 cred = get_creds("cert.pem");
51 gnutls_init(session, GNUTLS_CLIENT);
52 gnutls_set_default_priority(*session);
53 gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE, cred);
56 static int deflateRecord(char *buffer, size_t len, char **out, size_t *olen, int header)
60 zstrm.zalloc = Z_NULL;
62 zstrm.opaque = Z_NULL;
63 if ((r = deflateInit(&zstrm, Z_DEFAULT_COMPRESSION)) != Z_OK)
65 *out = malloc(len * 2 + 36);
70 zstrm.next_in = buffer;
72 zstrm.next_out = *out + 6;
73 zstrm.avail_out = len * 2 + 30;
74 while ((r = deflate(&zstrm, Z_FINISH)) != Z_STREAM_END &&
76 if ((r = deflate(&zstrm, Z_FINISH)) != Z_STREAM_END) {
81 *olen = zstrm.total_out + 6;
83 (*out)[1] = (zstrm.total_out >> 8);
84 (*out)[2] = (zstrm.total_out & 0xff);
85 (*out)[3] = (len >> 8);
86 (*out)[4] = (len & 0xff);
87 (*out)[5] = header ? 0x01 : 0x0;
92 static int inflateRecord(char *buffer, size_t len, char **out, size_t *olen)
96 zstrm.zalloc = Z_NULL;
98 zstrm.opaque = Z_NULL;
99 if ((r = inflateInit(&zstrm)) != Z_OK)
101 *olen = chars2len(buffer+3);
102 *out = malloc(*olen);
107 zstrm.next_in = buffer + 6;
108 zstrm.avail_in = len - 6;
109 zstrm.next_out = *out;
110 zstrm.avail_out = *olen;
111 while ((r = inflate(&zstrm, Z_FINISH)) != Z_STREAM_END &&
112 zstrm.avail_out > 0);
113 if ((r = inflate(&zstrm, Z_FINISH)) != Z_STREAM_END) {
122 #define RNET_ADDRESS "receitanet.receita.fazenda.gov.br"
124 static int connect_rnet(int *c)
126 struct addrinfo *addresses;
127 struct addrinfo *addr;
128 struct addrinfo hint;
129 struct sockaddr_in saddr;
133 memset(&hint, 0, sizeof(hint));
134 hint.ai_family = AF_UNSPEC;
135 hint.ai_socktype = SOCK_STREAM;
136 hint.ai_protocol = IPPROTO_TCP;
137 hint.ai_flags = AI_ADDRCONFIG;
138 r = getaddrinfo(RNET_ADDRESS, "3456", &hint, &addresses);
142 for (addr = addresses; addr != NULL; addr = addr->ai_next) {
143 fd = socket(addr->ai_family, addr->ai_socktype,
146 if (!(r = connect(fd, addr->ai_addr,
152 freeaddrinfo(addresses);
159 static int handshake(int c)
165 write(c, "00000000000000", 14);
166 r = read(c, buffer, 1);
167 if (r != 1 && buffer[0] != 'E')
169 r = read(c, buffer, 14);
175 static void usage(void)
177 fprintf(stderr, "rnetclient [filename]\n");
181 static int rnet_send(gnutls_session_t session, char *buffer, size_t len, int header)
185 deflateRecord(buffer, len, &out, &olen, header);
186 gnutls_record_send(session, out, olen);
191 static int rnet_recv(gnutls_session_t session, struct rnet_message **message)
198 rnet_message_expand(message, 6);
199 buffer = (*message)->buffer;
200 r = gnutls_record_recv(session, buffer, 6);
201 if (buffer[0] == 0x01) {
202 len = chars2len(buffer+1);
203 rnet_message_expand(message, len);
204 buffer = (*message)->buffer + 6;
205 r = gnutls_record_recv(session, buffer, len);
206 inflateRecord(buffer - 6, len + 6, &out, &olen);
207 rnet_message_del(*message);
209 rnet_message_expand(message, olen);
210 memcpy((*message)->buffer, out, olen);
211 (*message)->len = olen;
214 len = chars2len(buffer+1);
215 rnet_message_expand(message, len - 1);
216 buffer = (*message)->buffer + 6;
217 r = gnutls_record_recv(session, buffer, len - 1);
218 (*message)->len = len + 4;
219 rnet_message_strip(*message, 4);
224 static void save_rec_file(char *cpf, char *buffer, int len)
232 home = getenv("HOME");
234 tmpdir = getenv("TMPDIR");
239 fnlen = strlen(home) + strlen(cpf) + 13;
240 filename = malloc(fnlen);
241 snprintf(filename, fnlen, "%s/%s.REC.XXXXXX", home, cpf);
243 fd = mkstemp(filename);
245 fprintf(stderr, "Could not create receipt file: %s\n",
249 r = write(fd, buffer, len);
251 fprintf(stderr, "Could not write to receipt file%s%s\n",
253 r < 0 ? strerror(errno) : "");
256 fprintf(stderr, "Wrote the receipt to %s.\n", filename);
263 static void handle_response_text_and_file(char *cpf, struct rnet_message *message)
267 if (!rnet_message_parse(message, "texto", &value, &vlen))
268 fprintf(stderr, "%.*s\n", vlen, value);
269 if (!rnet_message_parse(message, "arquivo", &value, &vlen))
270 save_rec_file(cpf, value, vlen);
273 static void handle_response_already_found(char *cpf, struct rnet_message *message)
275 handle_response_text_and_file(cpf, message);
278 static void handle_response_error(struct rnet_message *message)
282 if (!rnet_message_parse(message, "texto", &value, &vlen))
283 fprintf(stderr, "%.*s\n", vlen, value);
284 fprintf(stderr, "Error transmiting DEC file.\n");
287 int main(int argc, char **argv)
291 struct rnet_decfile *decfile;
292 struct rnet_message *message = NULL;
293 gnutls_session_t session;
301 decfile = rnet_decfile_open(argv[1]);
303 fprintf(stderr, "could not parse %s: %s\n", argv[1], strerror(errno));
307 cpf = rnet_decfile_get_header_field(decfile, "cpf");
309 gnutls_global_init();
311 session_new(&session);
312 r = connect_rnet(&c);
314 fprintf(stderr, "error connecting to server: %s\n",
315 r == EAI_SYSTEM ? strerror(errno) : gai_strerror(r));
318 gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t)(intptr_t) c);
323 if ((r = gnutls_handshake(session)) < 0)
324 fprintf(stderr, "error in handshake: %s\n",
327 rnet_encode(decfile, &message);
328 rnet_send(session, message->buffer, message->len, 1);
329 rnet_message_del(message);
332 r = rnet_recv(session, &message);
333 if (r || !message || message->len == 0) {
334 fprintf(stderr, "error when receiving response\n");
337 switch (message->buffer[0]) {
338 case 1: /* go ahead */
339 handle_response_text_and_file(cpf, message);
342 handle_response_error(message);
346 handle_response_already_found(cpf, message);
351 handle_response_text_and_file(cpf, message);
355 rnet_message_del(message);
360 message = rnet_decfile_get_file(decfile);
361 rnet_send(session, message->buffer, message->len, 0);
364 r = rnet_recv(session, &message);
365 if (r || !message || message->len == 0) {
366 fprintf(stderr, "error when receiving response\n");
369 switch (message->buffer[0]) {
371 handle_response_error(message);
377 handle_response_text_and_file(cpf, message);
382 gnutls_bye(session, GNUTLS_SHUT_RDWR);
384 rnet_decfile_close(decfile);
385 gnutls_global_deinit();