decfile.c

   1 /*
   2  *  Copyright (C) 2012-2013  Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
   3  *
   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.
   8  *
   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.
  13  *
  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.
  17  */
  18
  19 #define _GNU_SOURCE
  20 #include "decfile.h"
  21 #include <string.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <errno.h>
  25 #include <unistd.h>
  26 #include "pmhash.h"
  27 #include "rnet_message.h"
  28
  29 #ifndef MAX
  30 #define MAX(a,b) (a >= b) ? a : b
  31 #endif
  32
  33 struct rnet_decfile {
  34         char *filename;
  35         FILE *file;
  36         char **lines;
  37         int lines_len;
  38         struct pmhash *header;
  39         struct rnet_message *message;
  40 };
  41
  42 /*
  43  * line should be an allocated buffer given to append_line
  44  * this means, free(line) will be called when decfile is released
  45  */
  46 static int append_line(struct rnet_decfile *decfile, char *line)
  47 {
  48         size_t len;
  49         char **old_lines;
  50         decfile->lines_len += 1;
  51         len = sizeof(*decfile->lines) * decfile->lines_len;
  52         old_lines = decfile->lines;
  53         decfile->lines = realloc(decfile->lines, len);
  54         if (!decfile->lines) {
  55                 decfile->lines = old_lines;
  56                 goto out;
  57         }
  58         decfile->lines[decfile->lines_len - 1] = line;
  59         return 0;
  60 out:
  61         decfile->lines_len -= 1;
  62         return -1;
  63 }
  64
  65 static void decfile_release_lines(struct rnet_decfile *decfile)
  66 {
  67         int i;
  68         for (i = 0; i < decfile->lines_len; i++)
  69                 free(decfile->lines[i]);
  70         free(decfile->lines);
  71         decfile->lines = NULL;
  72 }
  73
  74 static char * get_header(struct rnet_decfile *decfile);
  75 static int parse_header(struct pmhash *hash, char *buffer);
  76 static int decfile_parse_file(struct rnet_decfile *decfile);
  77
  78 static int decfile_parse_header(struct rnet_decfile *decfile)
  79 {
  80         char *buffer = get_header(decfile);
  81         if (!buffer || strlen(buffer) != 765)
  82                 return 1;
  83         return parse_header(decfile->header, buffer);
  84 }
  85
  86 static int decfile_parse(struct rnet_decfile *decfile)
  87 {
  88         char *buffer = NULL;
  89         size_t len = 0;
  90         int r;
  91         while ((r = getline(&buffer, &len, decfile->file)) > 0) {
  92                 r = append_line(decfile, buffer);
  93                 if (r) {
  94                         free(buffer);
  95                         goto out;
  96                 }
  97                 buffer = NULL;
  98                 len = 0;
  99         }
 100         if (!decfile_parse_header(decfile) && !decfile_parse_file(decfile))
 101                 return 0;
 102 out:
 103         decfile_release_lines(decfile);
 104         return -1;
 105 }
 106
 107 struct rnet_decfile * rnet_decfile_open(char *filename)
 108 {
 109         struct rnet_decfile *decfile;
 110         decfile = malloc(sizeof(*decfile));
 111         if (!decfile)
 112                 return NULL;
 113         decfile->header = pmhash_new();
 114         if (!decfile->header)
 115                 goto out_header;
 116         decfile->message = rnet_message_new();
 117         if (!decfile->message)
 118                 goto out_message;
 119         decfile->filename = strdup(filename);
 120         if (!decfile->filename)
 121                 goto out_filename;
 122         decfile->file = fopen(filename, "r");
 123         if (!decfile->file)
 124                 goto out_file;
 125         decfile->lines_len = 0;
 126         decfile->lines = NULL;
 127         if (decfile_parse(decfile))
 128                 goto out_parse;
 129         return decfile;
 130 out_parse:
 131         fclose(decfile->file);
 132 out_file:
 133         free(decfile->filename);
 134 out_filename:
 135         rnet_message_del(decfile->message);
 136 out_message:
 137         pmhash_del(decfile->header);
 138 out_header:
 139         free(decfile);
 140         return NULL;
 141 }
 142
 143 void rnet_decfile_close(struct rnet_decfile *decfile)
 144 {
 145         decfile_release_lines(decfile);
 146         fclose(decfile->file);
 147         free(decfile->filename);
 148         free(decfile);
 149 }
 150
 151 static char * get_header(struct rnet_decfile *decfile)
 152 {
 153         int i;
 154         for (i = 0; i < decfile->lines_len; i++) {
 155                 if (!strncmp(decfile->lines[i], "IRPF", 4)) {
 156                         return decfile->lines[i];
 157                 }
 158         }
 159         return NULL;
 160 }
 161
 162 static int parse_header(struct pmhash *hash, char *buffer)
 163 {
 164         char *p = buffer;
 165         char *key;
 166         char *val;
 167
 168 #define parse(field, sz) \
 169         val = malloc(sz + 1); \
 170         if (!val) \
 171                 goto out_val; \
 172         val[sz] = 0; \
 173         memcpy(val, p, sz); \
 174         p += sz; \
 175         key = strdup(field); \
 176         if (!key) \
 177                 goto out_key; \
 178         if (pmhash_add(&hash, key, val)) \
 179                 goto out_add;
 180
 181         parse("sistema", 8);
 182         parse("exerc", 4);
 183         parse("ano", 4);
 184         parse("codigo_recnet", 4);
 185         parse("in_ret", 1);
 186         parse("cpf", 11);
 187         parse("filler", 3);
 188         parse("tipo_ni", 1);
 189         parse("nr_versao", 3);
 190         parse("nome", 60);
 191         parse("uf", 2);
 192         parse("hash", 10);
 193         parse("in_cert", 1);
 194         parse("dt_nasc", 8);
 195         parse("in_comp", 1);
 196         parse("in_res", 1);
 197         parse("in_gerada", 1);
 198         parse("nr_recibo_anterior", 10);
 199         parse("in_pgd", 1);
 200         parse("so", 14);
 201         parse("versao_so", 7);
 202         parse("jvm", 9);
 203         parse("nr_recibo", 10);
 204         parse("municipio", 4);
 205         parse("conjuge", 11);
 206         parse("obrig", 1);
 207         parse("impdevido", 13);
 208         parse("nr_recibo", 10);
 209         parse("in_seg", 1);
 210         parse("imppago", 2);
 211         parse("impant", 1);
 212         parse("mudend", 1);
 213         parse("cep", 8);
 214         parse("debito", 1);
 215         parse("banco", 3);
 216         parse("agencia", 4);
 217         parse("filler", 1);
 218         parse("data_julgado", 8);
 219         parse("imppagar", 13);
 220         parse("tribfonte", 1);
 221         parse("cpfrra", 11);
 222         parse("trib_rra", 1);
 223         parse("cpf_rra2", 11);
 224         parse("trib_3rra", 1);
 225         parse("cpf_rra3", 11);
 226         parse("vr_doacao", 13);
 227         parse("cnpj1", 14);
 228         parse("cnpj2", 14);
 229         parse("cnpj3", 14);
 230         parse("cnpj4", 14);
 231         parse("cpf_dep1", 11);
 232         parse("dnas_dep1", 8);
 233         parse("cpf_dep2", 11);
 234         parse("dnas_dep2", 8);
 235         parse("cpf_dep3", 11);
 236         parse("dnas_dep3", 8);
 237         parse("cpf_dep4", 11);
 238         parse("dnas_dep4", 8);
 239         parse("cpf_dep5", 11);
 240         parse("dnas_dep5", 8);
 241         parse("cpf_dep6", 11);
 242         parse("dnas_dep6", 8);
 243         parse("cnpj_med1", 14);
 244         parse("cnpj_med2", 14);
 245         parse("cpf_alim", 11);
 246         parse("cpf_invent", 11);
 247         parse("municipio", 40);
 248         parse("contribuinte", 60);
 249         parse("cpf_empregada", 11);
 250         parse("hashcode", 12);
 251         parse("data_nao_residente", 8);
 252         parse("cpf_procurador", 11);
 253         parse("obrigatoriedade", 3);
 254         parse("rendtrib", 13);
 255         parse("cnpj_prev", 14);
 256         parse("cnpj_prev2", 14);
 257         parse("vr_totisentos", 13);
 258         parse("vr_totexclusivo", 13);
 259         parse("vr_totpagamentos", 13);
 260         parse("versaotestpgd", 3);
 261         parse("controle", 10);
 262
 263         return 0;
 264 out_add:
 265         free(key);
 266 out_key:
 267         free(val);
 268 out_val:
 269         return -1;
 270 }
 271
 272 char *rnet_decfile_get_header_field(struct rnet_decfile *decfile, char *field)
 273 {
 274         return pmhash_get(decfile->header, field);
 275 }
 276
 277 /* returns true if register is declaration and not a receipt */
 278 static int decfile_reg_is_dec(char *line)
 279 {
 280         if (line[0] >= '0' && line[0] <= '9' &&
 281             line[1] >= '0' && line[1] <= '9')
 282                 return 1;
 283         if (!strncmp(line, "IRPF    ", 8))
 284                 return 1;
 285         if (!strncmp(line, "T9", 2))
 286                 return 1;
 287         return 0;
 288 }
 289
 290 /* strip a register from its control number and append it to message */
 291 static int append_stripped_reg_ctrl(struct rnet_message **message, char *line)
 292 {
 293         size_t len;
 294         struct rnet_message *msg = *message;
 295         if (!decfile_reg_is_dec(line))
 296                 return 0;
 297         len = strlen(line);
 298         if (len < 12)
 299                 return -EINVAL;
 300         if (msg->alen - msg->len < len) {
 301                 if (rnet_message_expand(message, MAX(msg->len, len)))
 302                         return -ENOMEM;
 303                 msg = *message;
 304         }
 305         memcpy(&msg->buffer[msg->len], line, len - 12);
 306         msg->buffer[msg->len + len - 12] = '\r';
 307         msg->buffer[msg->len + len - 11] = '\n';
 308         msg->len += len - 10;
 309         return 0;
 310 }
 311
 312 static int decfile_parse_file(struct rnet_decfile *decfile)
 313 {
 314         int i;
 315         int r;
 316         for (i = 0; i < decfile->lines_len; i++) {
 317                 r = append_stripped_reg_ctrl(&decfile->message,
 318                                                 decfile->lines[i]);
 319                 if (r)
 320                         return r;
 321         }
 322         return 0;
 323 }
 324
 325 struct rnet_message * rnet_decfile_get_file(struct rnet_decfile *decfile)
 326 {
 327         return decfile->message;
 328 }