Handle uncompressed messages.

[cascardo/libreceita.git] / rnetclient.c

diff --git a/rnetclient.c b/rnetclient.c
index e1a3038..a9c75cd 100644 (file)
--- a/rnetclient.c
+++ b/rnetclient.c
@@ -74,10 +74,10 @@ 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;
@@ -194,20 +194,92 @@ static int rnet_recv(gnutls_session_t session, struct rnet_message **message)
        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);
+       if (buffer[0] == 0x01) {
+               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);
+       } else {
+               len = (buffer[1] << 8 | buffer[2]);
+               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;
@@ -216,6 +288,7 @@ int main(int argc, char **argv)
        struct rnet_message *message = NULL;
        gnutls_session_t session;
        int finish = 0;
+       char *cpf;
        
        if (argc < 2) {
                usage();
@@ -227,6 +300,8 @@ int main(int argc, char **argv)
                exit(1);
        }
 
+       cpf = rnet_decfile_get_header_field(decfile, "cpf");
+
        gnutls_global_init();
 
        session_new(&session);
@@ -255,16 +330,21 @@ int main(int argc, char **argv)
                fprintf(stderr, "error when receiving response\n");
                goto out;
        }
-       write(1, message->buffer, message->len);
        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 2:
        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;
        }
@@ -282,16 +362,15 @@ int main(int argc, char **argv)
                fprintf(stderr, "error when receiving response\n");
                goto out;
        }
-       write(1, message->buffer, message->len);
        switch (message->buffer[0]) {
        case 3: /* error */
-               finish = 1;
+               handle_response_error(message);
                break;
        case 2:
        case 4:
        case 5:
        case 1:
-               finish = 1;
+               handle_response_text_and_file(cpf, message);
                break;
        }