Rizin
unix-like reverse engineering framework and cli tools
parse_z80_pseudo.c
Go to the documentation of this file.
1 // SPDX-FileCopyrightText: 2021 deroad <wargio@libero.it>
2 // SPDX-License-Identifier: LGPL-3.0-only
3 
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <string.h>
7 
8 #include <rz_lib.h>
9 #include <rz_util.h>
10 #include <rz_flag.h>
11 #include <rz_analysis.h>
12 #include <rz_parse.h>
13 
14 #include "parse_common.c"
15 
16 static RzList *z80_tokenize(const char *assembly, size_t length);
17 
18 // https://wikiti.brandonw.net/index.php?title=Z80_Instruction_Set
19 static const RzPseudoGrammar z80_lexicon[] = {
20  RZ_PSEUDO_DEFINE_GRAMMAR("adc", "1 += 2 + cf"),
21  RZ_PSEUDO_DEFINE_GRAMMAR("add", "1 += 2"),
22  RZ_PSEUDO_DEFINE_GRAMMAR("and", "1 &= 2"),
23  RZ_PSEUDO_DEFINE_GRAMMAR("bit", "tmp = 2 & (#1 << 1)"),
24  RZ_PSEUDO_DEFINE_GRAMMAR("call", "call 1"),
25  // RZ_PSEUDO_DEFINE_GRAMMAR("ccf", "1 = 1 ? 2"),
26  // RZ_PSEUDO_DEFINE_GRAMMAR("cp", "1 = 1 ? 2"),
27  // RZ_PSEUDO_DEFINE_GRAMMAR("cpd", "1 = 1 ? 2"),
28  // RZ_PSEUDO_DEFINE_GRAMMAR("cpdr", "1 = 1 ? 2"),
29  // RZ_PSEUDO_DEFINE_GRAMMAR("cpi", "1 = 1 ? 2"),
30  // RZ_PSEUDO_DEFINE_GRAMMAR("cpir", "1 = 1 ? 2"),
31  RZ_PSEUDO_DEFINE_GRAMMAR("cpl", "1 = ~1"),
32  // RZ_PSEUDO_DEFINE_GRAMMAR("daa", "1 = 1 ? 2"),
33  RZ_PSEUDO_DEFINE_GRAMMAR("dec", "1--"),
34  // RZ_PSEUDO_DEFINE_GRAMMAR("di", "1 = 1 ? 2"),
35  // RZ_PSEUDO_DEFINE_GRAMMAR("djnz", "1 = 1 ? 2"),
36  // RZ_PSEUDO_DEFINE_GRAMMAR("ei", "1 = 1 ? 2"),
37  RZ_PSEUDO_DEFINE_GRAMMAR("ex", "swap(1, 2)"),
38  // RZ_PSEUDO_DEFINE_GRAMMAR("exx", "1 = 1 ? 2"),
39  // RZ_PSEUDO_DEFINE_GRAMMAR("halt", "1 = 1 ? 2"),
40  // RZ_PSEUDO_DEFINE_GRAMMAR("im", "1 = 1 ? 2"),
41  // RZ_PSEUDO_DEFINE_GRAMMAR("in", "1 = 1 ? 2"),
42  RZ_PSEUDO_DEFINE_GRAMMAR("in", "1 = [2]"),
43  RZ_PSEUDO_DEFINE_GRAMMAR("inc", "1++"),
44  // RZ_PSEUDO_DEFINE_GRAMMAR("ind", "1 = 1 ? 2"),
45  // RZ_PSEUDO_DEFINE_GRAMMAR("indr", "1 = 1 ? 2"),
46  // RZ_PSEUDO_DEFINE_GRAMMAR("ini", "1 = 1 ? 2"),
47  // RZ_PSEUDO_DEFINE_GRAMMAR("inir", "1 = 1 ? 2"),
48  RZ_PSEUDO_DEFINE_GRAMMAR("jp", "goto 1"),
49  RZ_PSEUDO_DEFINE_GRAMMAR("jp", "goto [1]"),
50  RZ_PSEUDO_DEFINE_GRAMMAR("jr", "goto +1"),
51  RZ_PSEUDO_DEFINE_GRAMMAR("ld", "1 = 2"),
52  RZ_PSEUDO_DEFINE_GRAMMAR("ldd", "1 = 2--"),
53  // RZ_PSEUDO_DEFINE_GRAMMAR("lddr", "1 = 1 ? 2"),
54  // RZ_PSEUDO_DEFINE_GRAMMAR("ldi", "1 = 1 ? 2"),
55  // RZ_PSEUDO_DEFINE_GRAMMAR("ldir", "1 = 1 ? 2"),
56  // RZ_PSEUDO_DEFINE_GRAMMAR("mulub", "1 = 1 ? 2"),
57  // RZ_PSEUDO_DEFINE_GRAMMAR("muluw", "1 = 1 ? 2"),
58  RZ_PSEUDO_DEFINE_GRAMMAR("neg", "1 = -1"),
59  RZ_PSEUDO_DEFINE_GRAMMAR("nop", ""),
60  RZ_PSEUDO_DEFINE_GRAMMAR("or", "1 |= 2"),
61  // RZ_PSEUDO_DEFINE_GRAMMAR("otdr", "1 = 1 ? 2"),
62  // RZ_PSEUDO_DEFINE_GRAMMAR("otir", "1 = 1 ? 2"),
63  RZ_PSEUDO_DEFINE_GRAMMAR("out", "1 = 2"),
64  // RZ_PSEUDO_DEFINE_GRAMMAR("outd", "1 = 1 ? 2"),
65  // RZ_PSEUDO_DEFINE_GRAMMAR("outi", "1 = 1 ? 2"),
66  RZ_PSEUDO_DEFINE_GRAMMAR("pop", "pop 1"),
67  RZ_PSEUDO_DEFINE_GRAMMAR("push", "push 1"),
68  // RZ_PSEUDO_DEFINE_GRAMMAR("res", "1 = 1 ? 2"),
69  RZ_PSEUDO_DEFINE_GRAMMAR("ret", "return"),
70  RZ_PSEUDO_DEFINE_GRAMMAR("reti", "return"),
71  RZ_PSEUDO_DEFINE_GRAMMAR("retn", "return"),
72  // RZ_PSEUDO_DEFINE_GRAMMAR("rl", "1 = 1 ? 2"),
73  // RZ_PSEUDO_DEFINE_GRAMMAR("rla", "1 = 1 ? 2"),
74  // RZ_PSEUDO_DEFINE_GRAMMAR("rlc", "1 = 1 ? 2"),
75  // RZ_PSEUDO_DEFINE_GRAMMAR("rlca", "1 = 1 ? 2"),
76  // RZ_PSEUDO_DEFINE_GRAMMAR("rld", "1 = 1 ? 2"),
77  RZ_PSEUDO_DEFINE_GRAMMAR("rr", "1 <<= 2"),
78  // RZ_PSEUDO_DEFINE_GRAMMAR("rra", "1 = 1 ? 2"),
79  // RZ_PSEUDO_DEFINE_GRAMMAR("rrc", "1 = 1 ? 2"),
80  // RZ_PSEUDO_DEFINE_GRAMMAR("rrca", "1 = 1 ? 2"),
81  // RZ_PSEUDO_DEFINE_GRAMMAR("rrd", "1 = 1 ? 2"),
82  RZ_PSEUDO_DEFINE_GRAMMAR("rst", "call 1"),
83  RZ_PSEUDO_DEFINE_GRAMMAR("sbc", "1 -= 2"),
84  // RZ_PSEUDO_DEFINE_GRAMMAR("scf", "1 = 1 ? 2"),
85  // RZ_PSEUDO_DEFINE_GRAMMAR("set", "1 = 1 ? 2"),
86  RZ_PSEUDO_DEFINE_GRAMMAR("sla", "1 <<= 2"),
87  RZ_PSEUDO_DEFINE_GRAMMAR("sra", "1 >>= 2"),
88  RZ_PSEUDO_DEFINE_GRAMMAR("srl", "1 >>= 2"),
89  RZ_PSEUDO_DEFINE_GRAMMAR("sub", "1 -= 2"),
90  RZ_PSEUDO_DEFINE_GRAMMAR("xor", "1 ^= 2"),
91 };
92 
93 static const RzPseudoConfig z80_config = RZ_PSEUDO_DEFINE_CONFIG_ONLY_LEXICON(z80_lexicon, 3, z80_tokenize);
94 
95 RzList *z80_tokenize(const char *assembly, size_t length) {
96  size_t i, p;
97  char *buf = NULL;
98  const char *comma_replace = NULL;
99  bool keep = false;
100  RzList *tokens = NULL;
101 
102  buf = rz_str_ndup(assembly, length);
103  if (!buf) {
104  return NULL;
105  }
106 
107  for (i = 0, p = 0; p < length; ++i, ++p) {
108  if (buf[p] == ',') {
109  if (!keep) {
110  p++;
111  } else if (buf[p + 1] == ' ') {
112  buf[i] = buf[p];
113  p++;
114  continue;
115  }
116  } else if (buf[p] == '(') {
117  keep = true;
118  comma_replace = ", ";
119  } else if (buf[p] == ')') {
120  keep = false;
121  }
122  if (p > i) {
123  buf[i] = buf[p];
124  }
125  }
126  buf[i] = 0;
127 
128  tokens = rz_str_split_duplist(buf, " ", true);
129  free(buf);
130  if (!tokens) {
131  return NULL;
132  }
133 
134  if (!strcmp((char *)rz_list_first(tokens), "call") && rz_list_length(tokens) == 3) {
135  void *arg1 = rz_list_get_n(tokens, 1);
136  void *arg2 = rz_list_get_n(tokens, 2);
137  rz_list_set_n(tokens, 1, arg2);
138  rz_list_set_n(tokens, 2, arg1);
139  }
140 
141  if (comma_replace) {
142  RzListIter *it;
143  rz_list_foreach (tokens, it, buf) {
144  it->data = rz_str_replace(buf, ",", comma_replace, 1);
145  }
146  }
147 
148  return tokens;
149 }
150 
151 static bool parse(RzParse *parse, const char *assembly, RzStrBuf *sb) {
152  return rz_pseudo_convert(&z80_config, assembly, sb);
153 }
154 
155 RzParsePlugin rz_parse_plugin_z80_pseudo = {
156  .name = "z80.pseudo",
157  .desc = "z80 pseudo syntax",
158  .init = NULL,
159  .fini = NULL,
160  .parse = parse,
161 };
162 
163 #ifndef RZ_PLUGIN_INCORE
164 RZ_API RzLibStruct rizin_plugin = {
165  .type = RZ_LIB_TYPE_PARSE,
166  .data = &rz_parse_plugin_z80_pseudo,
167  .version = RZ_VERSION
168 };
169 #endif
i
lzma_index ** i
Definition: index.h:629
sb
static SblHeader sb
Definition: bin_mbn.c:26
RZ_API
#define RZ_API
Definition: core_plugin_example.c:36
NULL
#define NULL
Definition: cris-opc.c:27
length
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
free
RZ_API void Ht_() free(HtName_(Ht) *ht)
Definition: ht_inc.c:130
buf
voidpf void * buf
Definition: ioapi.h:138
p
void * p
Definition: libc.cpp:67
rz_list_set_n
RZ_API ut32 rz_list_set_n(RZ_NONNULL RzList *list, ut32 n, void *p)
Sets the N-th element of the list.
Definition: list.c:552
rz_list_first
RZ_API RZ_BORROW void * rz_list_first(RZ_NONNULL const RzList *list)
Returns the first element of the list.
Definition: list.c:77
rz_list_get_n
RZ_API RZ_BORROW void * rz_list_get_n(RZ_NONNULL const RzList *list, ut32 n)
Returns the N-th element of the list.
Definition: list.c:574
rz_list_length
RZ_API ut32 rz_list_length(RZ_NONNULL const RzList *list)
Returns the length of the list.
Definition: list.c:109
parse_common.c
rz_pseudo_convert
static bool rz_pseudo_convert(const RzPseudoConfig *config, const char *assembly, RzStrBuf *sb)
Definition: parse_common.c:103
RZ_PSEUDO_DEFINE_GRAMMAR
#define RZ_PSEUDO_DEFINE_GRAMMAR(x, y)
Definition: parse_common.c:58
RZ_PSEUDO_DEFINE_CONFIG_ONLY_LEXICON
#define RZ_PSEUDO_DEFINE_CONFIG_ONLY_LEXICON(l, m, t)
Definition: parse_common.c:91
rizin_plugin
RZ_API RzLibStruct rizin_plugin
Definition: parse_z80_pseudo.c:164
z80_tokenize
static RzList * z80_tokenize(const char *assembly, size_t length)
Definition: parse_z80_pseudo.c:95
rz_parse_plugin_z80_pseudo
RzParsePlugin rz_parse_plugin_z80_pseudo
Definition: parse_z80_pseudo.c:155
parse
static bool parse(RzParse *parse, const char *assembly, RzStrBuf *sb)
Definition: parse_z80_pseudo.c:151
z80_config
static const RzPseudoConfig z80_config
Definition: parse_z80_pseudo.c:93
z80_lexicon
static const RzPseudoGrammar z80_lexicon[]
Definition: parse_z80_pseudo.c:19
rz_analysis.h
rz_flag.h
rz_lib.h
RZ_LIB_TYPE_PARSE
@ RZ_LIB_TYPE_PARSE
Definition: rz_lib.h:74
rz_parse.h
rz_str_ndup
RZ_API char * rz_str_ndup(RZ_NULLABLE const char *ptr, int len)
Create new copy of string ptr limited to size len.
Definition: str.c:1006
rz_str_split_duplist
RZ_API RzList * rz_str_split_duplist(const char *str, const char *c, bool trim)
Split the string str according to the substring c and returns a RzList with the result.
Definition: str.c:3464
rz_str_replace
RZ_API char * rz_str_replace(char *str, const char *key, const char *val, int g)
Definition: str.c:1110
rz_util.h
RZ_VERSION
#define RZ_VERSION
Definition: rz_version.h:8
parse
Definition: regcomp.c:57
rz_lib_struct_t
Definition: rz_lib.h:57
rz_lib_struct_t::type
int type
Definition: rz_lib.h:58
rz_list_iter_t
Definition: rz_list.h:13
rz_list_iter_t::data
void * data
Definition: rz_list.h:14
rz_list_t
Definition: rz_list.h:18
rz_parse_plugin_t
Definition: rz_parse.h:44
rz_parse_plugin_t::name
char * name
Definition: rz_parse.h:45
rz_parse_t
Definition: rz_parse.h:20