X-Git-Url: http://git.cascardo.eti.br/?a=blobdiff_plain;f=rnetclient.c;h=a58b62b43f61cb78083d124c1886e9e71792d6b2;hb=c09b5340844f0e8ea2613e5ad19adc811195dfce;hp=b5a8052a2966aa950b842b73de10bd3499590b44;hpb=6587900951afaa59075fcced40ba22fded88f1fe;p=cascardo%2Flibreceita.git diff --git a/rnetclient.c b/rnetclient.c index b5a8052..a58b62b 100644 --- a/rnetclient.c +++ b/rnetclient.c @@ -27,6 +27,13 @@ #include #include #include +#include "decfile.h" +#include "rnet_message.h" +#include "rnet_encode.h" + +static size_t chars2len (unsigned char buf[2]) { + return (buf[0] << 8 | buf[1]); +} static void * get_creds(char *certfile) { @@ -46,7 +53,7 @@ static void session_new(gnutls_session_t *session) gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE, cred); } -static int deflateRecord(char *buffer, size_t len, char **out, size_t *olen) +static int deflateRecord(char *buffer, size_t len, char **out, size_t *olen, int header) { z_stream zstrm; int r; @@ -71,13 +78,13 @@ static int deflateRecord(char *buffer, size_t len, char **out, size_t *olen) free(*out); return -1; } - *olen = zstrm.avail_out + 6; + *olen = zstrm.total_out + 6; (*out)[0] = 0x1; - (*out)[1] = (zstrm.avail_out >> 8); - (*out)[2] = (zstrm.avail_out & 0xff); + (*out)[1] = (zstrm.total_out >> 8); + (*out)[2] = (zstrm.total_out & 0xff); (*out)[3] = (len >> 8); (*out)[4] = (len & 0xff); - (*out)[5] = 0x1; + (*out)[5] = header ? 0x01 : 0x0; deflateEnd(&zstrm); return 0; } @@ -91,7 +98,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 = chars2len(buffer+3); *out = malloc(*olen); if (!out) { inflateEnd(&zstrm); @@ -165,15 +172,142 @@ static int handshake(int c) 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, int header) +{ + char *out; + size_t olen; + deflateRecord(buffer, len, &out, &olen, header); + 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); + if (buffer[0] == 0x01) { + len = chars2len(buffer+1); + 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); + } else { + len = chars2len(buffer+1); + rnet_message_expand(message, len - 1); + buffer = (*message)->buffer + 6; + r = gnutls_record_recv(session, buffer, len - 1); + (*message)->len = len + 4; + rnet_message_strip(*message, 4); + } + return 0; +} + +static void save_rec_file(char *cpf, char *buffer, int len) +{ + int fd; + char *filename; + char *home, *tmpdir; + mode_t mask; + size_t fnlen; + int r; + home = getenv("HOME"); + if (!home) { + tmpdir = getenv("TMPDIR"); + if (!tmpdir) + tmpdir = "/tmp"; + home = tmpdir; + } + fnlen = strlen(home) + strlen(cpf) + 13; + filename = malloc(fnlen); + snprintf(filename, fnlen, "%s/%s.REC.XXXXXX", home, cpf); + mask = umask(0177); + fd = mkstemp(filename); + if (fd < 0) { + fprintf(stderr, "Could not create receipt file: %s\n", + strerror(errno)); + goto out; + } + r = write(fd, buffer, len); + if (r != len) { + fprintf(stderr, "Could not write to receipt file%s%s\n", + r < 0 ? ": " : ".", + r < 0 ? strerror(errno) : ""); + goto out; + } + fprintf(stderr, "Wrote the receipt to %s.\n", filename); +out: + close(fd); + free(filename); + umask(mask); +} + +static void handle_response_text_and_file(char *cpf, struct rnet_message *message) +{ + char *value; + int vlen; + if (!rnet_message_parse(message, "texto", &value, &vlen)) + fprintf(stderr, "%.*s\n", vlen, value); + if (!rnet_message_parse(message, "arquivo", &value, &vlen)) + save_rec_file(cpf, value, vlen); +} + +static void handle_response_already_found(char *cpf, struct rnet_message *message) +{ + handle_response_text_and_file(cpf, message); +} + +static void handle_response_error(struct rnet_message *message) +{ + char *value; + int vlen; + if (!rnet_message_parse(message, "texto", &value, &vlen)) + fprintf(stderr, "%.*s\n", vlen, value); + fprintf(stderr, "Error transmiting DEC file.\n"); +} + int main(int argc, char **argv) { int c; int r; - char buffer[2048]; - char *out; - size_t olen; + struct rnet_decfile *decfile; + struct rnet_message *message = NULL; gnutls_session_t session; + int finish = 0; + char *cpf; + + 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); + } + + cpf = rnet_decfile_get_header_field(decfile, "cpf"); + gnutls_global_init(); + session_new(&session); r = connect_rnet(&c); if (r) { @@ -181,7 +315,7 @@ int main(int argc, char **argv) r == EAI_SYSTEM ? strerror(errno) : gai_strerror(r)); exit(1); } - gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) c); + gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t)(intptr_t) c); r = handshake(c); if (r < 0) { exit(1); @@ -189,15 +323,66 @@ int main(int argc, char **argv) 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, 1); + 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; + } + switch (message->buffer[0]) { + case 1: /* go ahead */ + handle_response_text_and_file(cpf, message); + break; + case 3: /* error */ + handle_response_error(message); + finish = 1; + break; + case 4: + handle_response_already_found(cpf, message); + finish = 1; + break; + case 2: + case 5: + handle_response_text_and_file(cpf, message); + finish = 1; + break; + } + rnet_message_del(message); + + if (finish) + goto out; + + message = rnet_decfile_get_file(decfile); + rnet_send(session, message->buffer, message->len, 0); + + message = NULL; + r = rnet_recv(session, &message); + if (r || !message || message->len == 0) { + fprintf(stderr, "error when receiving response\n"); + goto out; + } + switch (message->buffer[0]) { + case 3: /* error */ + handle_response_error(message); + break; + case 2: + case 4: + case 5: + case 1: + handle_response_text_and_file(cpf, message); + break; + } + +out: + gnutls_bye(session, GNUTLS_SHUT_RDWR); close(c); + rnet_decfile_close(decfile); gnutls_global_deinit(); + return 0; }