X-Git-Url: http://git.cascardo.info/?p=cascardo%2Flibreceita.git;a=blobdiff_plain;f=rnetclient.c;h=e1a30387120c2172cc0d63aacdba9823c086dddc;hp=0e13bb831c6cfed4a70af2d03e6fdced7ef587c6;hb=52012f62f845abd12b3e9fddb8aefc95e4b015ed;hpb=b377084739f475d7162ff2bdb3550cca848ef488 diff --git a/rnetclient.c b/rnetclient.c index 0e13bb8..e1a3038 100644 --- a/rnetclient.c +++ b/rnetclient.c @@ -1,5 +1,5 @@ /* - * Copyright (C) 2012 Thadeu Lima de Souza Cascardo + * Copyright (C) 2012-2013 Thadeu Lima de Souza Cascardo * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -24,8 +24,12 @@ #include #include #include +#include #include #include +#include "decfile.h" +#include "rnet_message.h" +#include "rnet_encode.h" static void * get_creds(char *certfile) { @@ -90,7 +94,7 @@ static int inflateRecord(char *buffer, size_t len, char **out, size_t *olen) zstrm.opaque = Z_NULL; if ((r = inflateInit(&zstrm)) != Z_OK) return -1; - *olen = (buffer[3] << 8 & buffer[4]); + *olen = (buffer[3] << 8 | buffer[4]); *out = malloc(*olen); if (!out) { inflateEnd(&zstrm); @@ -111,49 +115,191 @@ static int inflateRecord(char *buffer, size_t len, char **out, size_t *olen) return 0; } -int main(int argc, char **argv) +#define RNET_ADDRESS "receitanet.receita.fazenda.gov.br" + +static int connect_rnet(int *c) { + struct addrinfo *addresses; + struct addrinfo *addr; + struct addrinfo hint; struct sockaddr_in saddr; - int c; int r; - char buffer[1024]; + int fd = *c = -1; + int i; + memset(&hint, 0, sizeof(hint)); + hint.ai_family = AF_UNSPEC; + hint.ai_socktype = SOCK_STREAM; + hint.ai_protocol = IPPROTO_TCP; + hint.ai_flags = AI_ADDRCONFIG; + r = getaddrinfo(RNET_ADDRESS, "3456", &hint, &addresses); + if (r) { + return r; + } + for (addr = addresses; addr != NULL; addr = addr->ai_next) { + fd = socket(addr->ai_family, addr->ai_socktype, + addr->ai_protocol); + if (fd >= 0) + if (!(r = connect(fd, addr->ai_addr, + addr->ai_addrlen))) + break; + close(fd); + fd = -1; + } + freeaddrinfo(addresses); + *c = fd; + if (fd == -1) + return EAI_SYSTEM; + return 0; +} + +static int handshake(int c) +{ + char buffer[16]; + int r; + buffer[0] = 1; + write(c, buffer, 1); + write(c, "00000000000000", 14); + r = read(c, buffer, 1); + if (r != 1 && buffer[0] != 'E') + return -1; + r = read(c, buffer, 14); + if (r != 14) + return -1; + return 0; +} + +static void usage(void) +{ + fprintf(stderr, "rnetclient [filename]\n"); + exit(1); +} + +static int rnet_send(gnutls_session_t session, char *buffer, size_t len) +{ + char *out; + size_t olen; + deflateRecord(buffer, len, &out, &olen); + gnutls_record_send(session, out, olen); + free(out); + return 0; +} + +static int rnet_recv(gnutls_session_t session, struct rnet_message **message) +{ char *out; size_t olen; + int r; + char *buffer; + size_t len; + rnet_message_expand(message, 6); + buffer = (*message)->buffer; + r = gnutls_record_recv(session, buffer, 6); + len = (buffer[1] << 8 | buffer[2]); + rnet_message_expand(message, len); + buffer = (*message)->buffer + 6; + r = gnutls_record_recv(session, buffer, len); + inflateRecord(buffer - 6, len + 6, &out, &olen); + rnet_message_del(*message); + *message = NULL; + rnet_message_expand(message, olen); + memcpy((*message)->buffer, out, olen); + (*message)->len = olen; + free(out); + return 0; +} + +int main(int argc, char **argv) +{ + int c; + int r; + struct rnet_decfile *decfile; + struct rnet_message *message = NULL; gnutls_session_t session; + int finish = 0; + + if (argc < 2) { + usage(); + } + + decfile = rnet_decfile_open(argv[1]); + if (!decfile) { + fprintf(stderr, "could not parse %s: %s\n", argv[1], strerror(errno)); + exit(1); + } + gnutls_global_init(); + session_new(&session); - c = socket(PF_INET, SOCK_STREAM, 0); - saddr.sin_family = AF_INET; - saddr.sin_port = htons(3456); - saddr.sin_addr.s_addr = inet_addr("161.148.185.140"); - r = connect(c, (struct sockaddr *) &saddr, sizeof(saddr)); - if (r < 0) { + r = connect_rnet(&c); + if (r) { fprintf(stderr, "error connecting to server: %s\n", - strerror(errno)); + r == EAI_SYSTEM ? strerror(errno) : gai_strerror(r)); exit(1); } gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) c); - buffer[0] = 1; - write(c, buffer, 1); - write(c, "00000000000000", 14); - r = read(c, buffer, 1); - if (r != 1 && buffer[0] != 'E') - exit(1); - r = read(c, buffer, 14); - if (r != 14) + r = handshake(c); + if (r < 0) { exit(1); + } if ((r = gnutls_handshake(session)) < 0) fprintf(stderr, "error in handshake: %s\n", gnutls_strerror(r)); - else - fprintf(stderr, "handshake ok\n"); - r = read(0, buffer, sizeof(buffer)); - deflateRecord(buffer, r, &out, &olen); - gnutls_record_send(session, out, olen); - free(out); - while ((r = gnutls_record_recv(session, buffer, sizeof(buffer))) > 0) - write(1, buffer, r); + + rnet_encode(decfile, &message); + rnet_send(session, message->buffer, message->len); + rnet_message_del(message); + + message = NULL; + r = rnet_recv(session, &message); + if (r || !message || message->len == 0) { + fprintf(stderr, "error when receiving response\n"); + goto out; + } + write(1, message->buffer, message->len); + switch (message->buffer[0]) { + case 1: /* go ahead */ + break; + case 3: /* error */ + finish = 1; + break; + case 2: + case 4: + case 5: + finish = 1; + break; + } + rnet_message_del(message); + + if (finish) + goto out; + + message = rnet_decfile_get_file(decfile); + rnet_send(session, message->buffer, message->len); + + message = NULL; + r = rnet_recv(session, &message); + if (r || !message || message->len == 0) { + fprintf(stderr, "error when receiving response\n"); + goto out; + } + write(1, message->buffer, message->len); + switch (message->buffer[0]) { + case 3: /* error */ + finish = 1; + break; + case 2: + case 4: + case 5: + case 1: + finish = 1; + break; + } + +out: + gnutls_bye(session, GNUTLS_SHUT_RDWR); close(c); + rnet_decfile_close(decfile); gnutls_global_deinit(); + return 0; }