File: getidblock.c

package info (click to toggle)
libpst 0.5.3-1
  • links: PTS
  • area: main
  • in suites: lenny, squeeze
  • size: 516 kB
  • ctags: 714
  • sloc: ansic: 7,115; makefile: 98; sh: 23
file content (139 lines) | stat: -rw-r--r-- 3,477 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#include <stdio.h>
#include <string.h>

#ifndef __GNUC__
# include "XGetopt.h"
#endif

#ifndef _WIN32
# include <unistd.h>
#endif

#include "define.h"
#include "libpst.h"

static void usage();

int main(int argc, char ** argv) {
  // pass the id number to display on the command line
  char *fname, *sid;
  pst_file pstfile;
  unsigned int id;
  int decrypt = 0, process = 0, binary = 0, c;
  unsigned char *buf = NULL;
  size_t readSize;
  pst_item *item;
  pst_desc_ll* ptr;

  DEBUG_INIT("getidblock.log");
  DEBUG_REGISTER_CLOSE();
  DEBUG_ENT("main");

  while ((c = getopt(argc, argv, "bdp")) != -1) {
    switch (c) {
    case 'b':
      // enable binary output
      binary = 1;
      break;
    case 'd':
      //enable decrypt
      decrypt = 1;
      break;
    case 'p':
      // enable procesing of block
      process = 1;
      break;
    default:
      usage();
      exit(EXIT_FAILURE);
    }
  }

  if (optind+1 >= argc) {
    // no more items on the cmd
    usage();
    exit(EXIT_FAILURE);
  }
  fname = argv[optind];
  sid = argv[optind+1];
  id = (unsigned int)strtol(sid, NULL, 0);

  DEBUG_MAIN(("Opening file\n"));
  memset(&pstfile, 0, sizeof(pstfile));
  if (pst_open(&pstfile, fname, "r")!=0) {
    DIE(("Error opening file\n"));
  }
  
  DEBUG_MAIN(("Loading Index\n"));
  if (pst_load_index(&pstfile) != 0) {
    DIE(("Error loading file index\n"));
  }

  //  if ((ptr = _pst_getID(&pstfile, id)) == NULL) {
  //    DIE(("id not found [%#x]\n", id));
  //  }

  DEBUG_MAIN(("Loading block\n"));

  if ((readSize = _pst_ff_getIDblock(&pstfile, id, &buf)) <= 0 || buf == NULL) {
    //  if ((readSize = _pst_read_block_size(&pstfile, ptr->offset, ptr->size, &buf, 1, 1)) < ptr->size) {
    DIE(("Error loading block\n"));
  }
  if (binary==0) printf("Block %#x, size %#x[%i]\n",id, (unsigned int)readSize, (int) readSize);

  if (decrypt!=0)
    if (_pst_decrypt(buf, readSize, (int)pstfile.encryption) != 0) {
      DIE(("Error decrypting block\n"));
    }

  DEBUG_MAIN(("Printing block... [id %#x, size %#x]\n", id, readSize));
  if (binary==0) {
    _pst_debug_hexdump(stdout, buf, readSize, 0x10);
  } else {
    if (fwrite(buf, 1, readSize, stdout) != 0) {
      DIE(("Error occured during writing of buf to stdout\n"));
    }
  }
  free(buf);

  if (process!=0) {
    DEBUG_MAIN(("Parsing block...\n"));
    ptr = pstfile.d_head;
    while(ptr != NULL) {
      if (ptr->list_index != NULL && ptr->list_index->id == id)
	break;
      if (ptr->desc != NULL && ptr->desc->id == id)
	break;
      ptr = pst_getNextDptr(ptr);
    }
    if (ptr == NULL) {
      ptr = (pst_desc_ll*)xmalloc(sizeof(pst_desc_ll));
      ptr->desc = _pst_getID(&pstfile, id);
      ptr->list_index = NULL;
    }
    if (ptr != NULL) {
      if ((item = _pst_parse_item(&pstfile, ptr)) != NULL)
	_pst_freeItem(item);
    } else {
      DEBUG_MAIN(("item not found with this ID\n"));
      printf("Cannot find the owning Record of this ID. Cannot parse\n");
    }
  }
  
  if(pst_close(&pstfile)!=0) {
    DIE(("pst_close failed\n"));
  }

  DEBUG_RET();
  return 0;
}

void usage() {
  printf("usage: getidblock [options] filename id\n");
  printf("\tfilename - name of the file to access\n");
  printf("\tid - ID of the block to fetch - can begin with 0x for hex\n");
  printf("\toptions\n");
  printf("\t\t-d\tDecrypt the block before printing\n");
  printf("\t\t-p\tProcess the block before finishing.\n");
  printf("\t\t\tView the debug log for information\n");
}