Rizin
unix-like reverse engineering framework and cli tools
blast.c File Reference
#include <stddef.h>
#include <setjmp.h>
#include "blast.h"

Go to the source code of this file.

Classes

struct  state
 
struct  huffman
 

Macros

#define local   static /* for local function definitions */
 
#define MAXBITS   13 /* maximum code length */
 
#define MAXWIN   4096 /* maximum window size */
 

Functions

int bits (struct state *s, int need)
 
int decode (struct state *s, struct huffman *h)
 
int construct (struct huffman *h, const unsigned char *rep, int n)
 
int decomp (struct state *s)
 
int blast (blast_in infun, void *inhow, blast_out outfun, void *outhow, unsigned *left, unsigned char **in)
 

Macro Definition Documentation

◆ local

#define local   static /* for local function definitions */

Definition at line 36 of file blast.c.

◆ MAXBITS

#define MAXBITS   13 /* maximum code length */

Definition at line 37 of file blast.c.

◆ MAXWIN

#define MAXWIN   4096 /* maximum window size */

Definition at line 38 of file blast.c.

Function Documentation

◆ bits()

int bits ( struct state s,
int  need 
)

Definition at line 72 of file blast.c.

73 {
74  int val; /* bit accumulator */
75 
76  /* load at least need bits into val */
77  val = s->bitbuf;
78  while (s->bitcnt < need) {
79  if (s->left == 0) {
80  s->left = s->infun(s->inhow, &(s->in));
81  if (s->left == 0) longjmp(s->env, 1); /* out of input */
82  }
83  val |= (int)(*(s->in)++) << s->bitcnt; /* load eight bits */
84  s->left--;
85  s->bitcnt += 8;
86  }
87 
88  /* drop need bits and update buffer, always zero to seven bits left */
89  s->bitbuf = val >> need;
90  s->bitcnt -= need;
91 
92  /* return need bits, zeroing the bits above that */
93  return val & ((1 << need) - 1);
94 }
ut16 val
Definition: armass64_const.h:6
static RzSocket * s
Definition: rtr.c:28
static int
Definition: sfsocketcall.h:114

References int, s, and val.

Referenced by __print_stack_cb(), add_sub(), addpoint(), address_bits(), adjust_unsigned(), anop_esil(), assemble(), autocmplt_bits_plugin(), bfd_get_bits(), bfm(), bic(), binary(), bitfield(), bits_hint_acc_cb(), DotZLib.Info::bitSize(), bitwise(), bl(), branch(), cb_asmarch(), cb_asmbits(), cbz(), cls(), clz(), cmd_analysis_trampoline(), cmp(), core_bin_file_print(), countLeadingZeros(), create(), cset(), csinc(), decode_add(), decode_arith(), decode_call(), decode_cmpeq(), decode_cmpgte(), decode_cmplte(), decode_cmpugte(), decode_cmpulte(), decode_div(), decode_divu(), decode_jmp(), decode_mul(), decode_mulu(), decode_not(), decode_sub(), decomp(), DEFINE_HANDLE_TS_FCN_AND_SYMBOL(), deflatePending(), deflatePrime(), ds_print_data_type(), entries(), esil_mem_addeq_n(), esil_mem_andeq_n(), esil_mem_deceq_n(), esil_mem_diveq_n(), esil_mem_inceq_n(), esil_mem_lsleq_n(), esil_mem_lsreq_n(), esil_mem_modeq_n(), esil_mem_muleq_n(), esil_mem_oreq_n(), esil_mem_subeq_n(), esil_mem_xoreq_n(), esil_peek_n(), esil_poke_n(), examine(), expand_1st_key(), expand_2nd_key(), expand_subkey(), extr(), extract_binobj(), filter_reg_items(), fixedtables(), gdbr_get_reg_profile(), gdbr_set_architecture(), gen_bitlen(), gen_codes(), genmask(), get_bits(), get_category_t(), get_dwarf_reg_name(), get_symbols(), get_symbols_list(), getvalue(), gzip_normalize(), handle_arm_hint(), handleHints(), has_bits(), inflate(), inflate_fast(), inflate_table(), inflate_table9(), inflateBack(), inflateBack9(), inflatePrime(), info(), insert_offset(), is_arm_thumb_hack(), ldr(), libps_decode(), load_effect(), load_omf_symb(), lunpipe(), madd(), main(), mov(), movk(), movn(), mrs(), msr(), mul(), mvn(), newEntry(), number_of_bits(), omf_count_symb(), op_fillval(), opmov(), parse_struct_node(), parse_union_node(), parseDragons(), patch_capstone_bugs(), print_assembly_output(), print_bits_hint_cb(), process_constructors(), rasm_asm(), rasm_disasm(), rbit(), rc_direct_price(), reloc_target_size(), rotateAsmBits(), rz_analysis_data(), rz_analysis_function_setbits_handler(), rz_analysis_hint_set_bits(), rz_analysis_hint_set_bits_handler(), rz_analysis_hint_set_newbits(), rz_analysis_set_bits(), rz_analysis_set_triplet(), rz_arm_cs_analysis_op_64_esil(), rz_asm_set_arch(), rz_asm_set_bits(), rz_asm_setup(), rz_asn1_stringify_oid(), rz_bin_arch_options_init(), rz_bin_elf_fix_arm_thumb_object(), rz_bin_file_find_by_arch_bits(), rz_bin_object_find_by_arch_bits(), rz_bin_p9_get_arch(), rz_bin_pe_get_bits(), rz_bin_select(), rz_bin_select_object(), rz_bin_use_arch(), rz_bp_get_bytes(), rz_bp_size(), rz_bp_size_at(), rz_bp_traptrace_add(), rz_bv_hash(), rz_cmd_help(), rz_core_analysis_cc_init(), rz_core_analysis_hasrefs_to_depth(), rz_core_analysis_hint_print(), rz_core_analysis_sigdb_apply(), rz_core_analysis_sigdb_print(), rz_core_analysis_type_init(), rz_core_arch_bits_at(), rz_core_arg_get(), rz_core_asm_plugin_print(), rz_core_bin_apply_all_info(), rz_core_bin_info_print(), rz_core_bin_load(), rz_core_bin_set_arch_bits(), rz_core_bin_update_arch_bits(), rz_core_cmd_subst_i(), rz_core_egg_setup(), rz_core_file_reopen_debug(), rz_core_file_reopen_remote_debug(), rz_core_hack_arm(), rz_core_rtr_gdb_run(), rz_core_seek_arch_bits(), rz_core_set_asm_configs(), rz_coresym_cache_element_new(), rz_debug_bochs_reg_profile(), rz_debug_gdb_attach(), rz_debug_gdb_reg_profile(), rz_debug_gdb_reg_write(), rz_debug_qnx_attach(), rz_debug_qnx_reg_profile(), rz_debug_qnx_reg_write(), rz_debug_set_arch(), rz_egg_Cfile_parser(), rz_egg_Cfile_set_cEnv(), rz_egg_patch_num(), rz_egg_setup(), rz_main_rz_asm(), rz_main_rz_bin(), rz_main_rz_gg(), rz_mem_copybits(), rz_mem_copybits_delta(), rz_open_arch_bits_handler(), rz_project_migrate_v4_v5(), rz_read_be_double(), rz_read_be_float(), rz_read_le_double(), rz_read_le_float(), rz_read_me_double(), rz_read_me_float(), rz_str_utf16_to_utf8(), rz_syscall_setup(), rz_test_load_asm_test_file(), rz_test_run_asm_test(), rz_type_db_init(), rz_type_db_set_bits(), rz_write_be_double(), rz_write_be_float(), rz_write_le_double(), rz_write_le_float(), rz_write_me_double(), rz_write_me_float(), saturate(), saturate_signed(), saturate_signed_to_range(), saturate_unsigned(), sbfx(), sdiv(), set_default_type(), set_src_dst(), set_tmp_bits(), setab(), shift(), sign_ext(), sign_extend(), smaddl(), smulh(), smull(), str(), string_scan_range_cfstring(), sxt(), syscall_reload_needed(), sysregs_reload_needed(), tbz(), tr_static_init(), tst(), typeString(), udiv(), update_asmbits_options(), update_bits_range(), update_reg_list_regbits(), utf16be_to_utf8_impl(), winkd_get_profile(), and xtr_metadata_match().

◆ blast()

int blast ( blast_in  infun,
void *  inhow,
blast_out  outfun,
void *  outhow,
unsigned left,
unsigned char **  in 
)

Definition at line 383 of file blast.c.

385 {
386  struct state s; /* input/output state */
387  int err; /* return value */
388 
389  /* initialize input state */
390  s.infun = infun;
391  s.inhow = inhow;
392  if (left != NULL && *left) {
393  s.left = *left;
394  s.in = *in;
395  }
396  else
397  s.left = 0;
398  s.bitbuf = 0;
399  s.bitcnt = 0;
400 
401  /* initialize output state */
402  s.outfun = outfun;
403  s.outhow = outhow;
404  s.next = 0;
405  s.first = 1;
406 
407  /* return if bits() or decode() tries to read past available input */
408  if (setjmp(s.env) != 0) /* if came back here via longjmp(), */
409  err = 2; /* then skip decomp(), return error */
410  else
411  err = decomp(&s); /* decompress */
412 
413  /* return unused input */
414  if (left != NULL)
415  *left = s.left;
416  if (in != NULL)
417  *in = s.left ? s.in : NULL;
418 
419  /* write any leftover output and update the error code if needed */
420  if (err != 1 && s.next && s.outfun(s.outhow, s.out, s.next) && err == 0)
421  err = 1;
422  return err;
423 }
static bool err
Definition: armass.c:435
int decomp(struct state *s)
Definition: blast.c:282
const lzma_allocator const uint8_t * in
Definition: block.h:527
#define NULL
Definition: cris-opc.c:27
Definition: dis.h:43
unsigned left
Definition: blast.c:46
blast_out outfun
Definition: blast.c:54
blast_in infun
Definition: blast.c:43
void * outhow
Definition: blast.c:55
void * inhow
Definition: blast.c:44

References decomp(), err, in, state::infun, state::inhow, state::left, NULL, state::outfun, state::outhow, and s.

◆ construct()

int construct ( struct huffman h,
const unsigned char *  rep,
int  n 
)

Definition at line 191 of file blast.c.

192 {
193  int symbol; /* current symbol when stepping through length[] */
194  int len; /* current length when stepping through h->count[] */
195  int left; /* number of possible codes left of current length */
196  short offs[MAXBITS+1]; /* offsets in symbol table for each length */
197  short length[256]; /* code lengths */
198 
199  /* convert compact repeat counts into symbol bit length list */
200  symbol = 0;
201  do {
202  len = *rep++;
203  left = (len >> 4) + 1;
204  len &= 15;
205  do {
206  length[symbol++] = len;
207  } while (--left);
208  } while (--n);
209  n = symbol;
210 
211  /* count number of codes of each length */
212  for (len = 0; len <= MAXBITS; len++)
213  h->count[len] = 0;
214  for (symbol = 0; symbol < n; symbol++)
215  (h->count[length[symbol]])++; /* assumes lengths are within bounds */
216  if (h->count[0] == n) /* no codes! */
217  return 0; /* complete, but decode() will fail */
218 
219  /* check for an over-subscribed or incomplete set of lengths */
220  left = 1; /* one possible code of zero length */
221  for (len = 1; len <= MAXBITS; len++) {
222  left <<= 1; /* one more bit, double codes left */
223  left -= h->count[len]; /* deduct count from possible codes */
224  if (left < 0) return left; /* over-subscribed--return negative */
225  } /* left > 0 means incomplete */
226 
227  /* generate offsets into symbol table for each length for sorting */
228  offs[1] = 0;
229  for (len = 1; len < MAXBITS; len++)
230  offs[len + 1] = offs[len] + h->count[len];
231 
232  /*
233  * put symbols in table sorted by length, by symbol order within each
234  * length
235  */
236  for (symbol = 0; symbol < n; symbol++)
237  if (length[symbol] != 0)
238  h->symbol[offs[length[symbol]]++] = symbol;
239 
240  /* return zero for complete set, positive for incomplete set */
241  return left;
242 }
size_t len
Definition: 6502dis.c:15
#define MAXBITS
Definition: blast.c:37
static static sync static getppid static getegid const char static filename char static len const char char static bufsiz static mask static vfork const void static prot static getpgrp const char static swapflags static arg static fd static protocol static who struct sockaddr static addrlen static backlog struct timeval struct timezone static tz const struct iovec static count static mode const void const struct sockaddr static tolen const char static pathname void static offset struct stat static buf void long static basep static whence static length const void static len static semflg const void static shmflg const struct timespec struct timespec static rem const char static group const void length
Definition: sflib.h:133
int n
Definition: mipsasm.c:19
#define h(i)
Definition: sha256.c:48

References h, len, length, MAXBITS, and n.

Referenced by decomp().

◆ decode()

int decode ( struct state s,
struct huffman h 
)

Definition at line 129 of file blast.c.

130 {
131  int len; /* current number of bits in code */
132  int code; /* len bits being decoded */
133  int first; /* first code of length len */
134  int count; /* number of codes of length len */
135  int index; /* index of first code of length len in symbol table */
136  int bitbuf; /* bits from stream */
137  int left; /* bits left in next or left to process */
138  short *next; /* next number of codes */
139 
140  bitbuf = s->bitbuf;
141  left = s->bitcnt;
142  code = first = index = 0;
143  len = 1;
144  next = h->count + 1;
145  while (1) {
146  while (left--) {
147  code |= (bitbuf & 1) ^ 1; /* invert code */
148  bitbuf >>= 1;
149  count = *next++;
150  if (code < first + count) { /* if length len, return symbol */
151  s->bitbuf = bitbuf;
152  s->bitcnt = (s->bitcnt - len) & 7;
153  return h->symbol[index + (code - first)];
154  }
155  index += count; /* else update for next length */
156  first += count;
157  first <<= 1;
158  code <<= 1;
159  len++;
160  }
161  left = (MAXBITS+1) - len;
162  if (left == 0) break;
163  if (s->left == 0) {
164  s->left = s->infun(s->inhow, &(s->in));
165  if (s->left == 0) longjmp(s->env, 1); /* out of input */
166  }
167  bitbuf = *(s->in)++;
168  s->left--;
169  if (left > 8) left = 8;
170  }
171  return -9; /* ran out of codes */
172 }
static static sync static getppid static getegid const char static filename char static len const char char static bufsiz static mask static vfork const void static prot static getpgrp const char static swapflags static arg static fd static protocol static who struct sockaddr static addrlen static backlog struct timeval struct timezone static tz const struct iovec static count static mode const void const struct sockaddr static tolen const char static pathname void count
Definition: sflib.h:98
const char * code
Definition: pal.c:98
Definition: inftree9.h:24

References code, count, h, len, MAXBITS, and s.

Referenced by decomp().

◆ decomp()

int decomp ( struct state s)

Definition at line 282 of file blast.c.

283 {
284  int lit; /* true if literals are coded */
285  int dict; /* log2(dictionary size) - 6 */
286  int symbol; /* decoded symbol, extra bits for distance */
287  int len; /* length for copy */
288  unsigned dist; /* distance for copy */
289  int copy; /* copy counter */
290  unsigned char *from, *to; /* copy pointers */
291  static int virgin = 1; /* build tables once */
292  static short litcnt[MAXBITS+1], litsym[256]; /* litcode memory */
293  static short lencnt[MAXBITS+1], lensym[16]; /* lencode memory */
294  static short distcnt[MAXBITS+1], distsym[64]; /* distcode memory */
295  static struct huffman litcode = {litcnt, litsym}; /* length code */
296  static struct huffman lencode = {lencnt, lensym}; /* length code */
297  static struct huffman distcode = {distcnt, distsym};/* distance code */
298  /* bit lengths of literal codes */
299  static const unsigned char litlen[] = {
300  11, 124, 8, 7, 28, 7, 188, 13, 76, 4, 10, 8, 12, 10, 12, 10, 8, 23, 8,
301  9, 7, 6, 7, 8, 7, 6, 55, 8, 23, 24, 12, 11, 7, 9, 11, 12, 6, 7, 22, 5,
302  7, 24, 6, 11, 9, 6, 7, 22, 7, 11, 38, 7, 9, 8, 25, 11, 8, 11, 9, 12,
303  8, 12, 5, 38, 5, 38, 5, 11, 7, 5, 6, 21, 6, 10, 53, 8, 7, 24, 10, 27,
304  44, 253, 253, 253, 252, 252, 252, 13, 12, 45, 12, 45, 12, 61, 12, 45,
305  44, 173};
306  /* bit lengths of length codes 0..15 */
307  static const unsigned char lenlen[] = {2, 35, 36, 53, 38, 23};
308  /* bit lengths of distance codes 0..63 */
309  static const unsigned char distlen[] = {2, 20, 53, 230, 247, 151, 248};
310  static const short base[16] = { /* base for length codes */
311  3, 2, 4, 5, 6, 7, 8, 9, 10, 12, 16, 24, 40, 72, 136, 264};
312  static const char extra[16] = { /* extra bits for length codes */
313  0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8};
314 
315  /* set up decoding tables (once--might not be thread-safe) */
316  if (virgin) {
317  construct(&litcode, litlen, sizeof(litlen));
318  construct(&lencode, lenlen, sizeof(lenlen));
319  construct(&distcode, distlen, sizeof(distlen));
320  virgin = 0;
321  }
322 
323  /* read header */
324  lit = bits(s, 8);
325  if (lit > 1) return -1;
326  dict = bits(s, 8);
327  if (dict < 4 || dict > 6) return -2;
328 
329  /* decode literals and length/distance pairs */
330  do {
331  if (bits(s, 1)) {
332  /* get length */
333  symbol = decode(s, &lencode);
334  len = base[symbol] + bits(s, extra[symbol]);
335  if (len == 519) break; /* end code */
336 
337  /* get distance */
338  symbol = len == 2 ? 2 : dict;
339  dist = decode(s, &distcode) << symbol;
340  dist += bits(s, symbol);
341  dist++;
342  if (s->first && dist > s->next)
343  return -3; /* distance too far back */
344 
345  /* copy length bytes from distance bytes back */
346  do {
347  to = s->out + s->next;
348  from = to - dist;
349  copy = MAXWIN;
350  if (s->next < dist) {
351  from += copy;
352  copy = dist;
353  }
354  copy -= s->next;
355  if (copy > len) copy = len;
356  len -= copy;
357  s->next += copy;
358  do {
359  *to++ = *from++;
360  } while (--copy);
361  if (s->next == MAXWIN) {
362  if (s->outfun(s->outhow, s->out, s->next)) return 1;
363  s->next = 0;
364  s->first = 0;
365  }
366  } while (len != 0);
367  }
368  else {
369  /* get literal and write it */
370  symbol = lit ? decode(s, &litcode) : bits(s, 8);
371  s->out[s->next++] = symbol;
372  if (s->next == MAXWIN) {
373  if (s->outfun(s->outhow, s->out, s->next)) return 1;
374  s->next = 0;
375  s->first = 0;
376  }
377  }
378  } while (1);
379  return 0;
380 }
#define MAXWIN
Definition: blast.c:38
int construct(struct huffman *h, const unsigned char *rep, int n)
Definition: blast.c:191
int decode(struct state *s, struct huffman *h)
Definition: blast.c:129
int bits(struct state *s, int need)
Definition: blast.c:72
static struct sockaddr static addrlen static backlog const void static flags void struct sockaddr from
Definition: sfsocketcall.h:123
static struct sockaddr static addrlen static backlog const void static flags void struct sockaddr socklen_t static fromlen const void const struct sockaddr to
Definition: sfsocketcall.h:125
Definition: blast.c:103
short * symbol
Definition: blast.c:105

References bits(), construct(), decode(), from, len, MAXBITS, MAXWIN, s, huffman::symbol, and to.

Referenced by blast().