Skip to content

Commit ca81d42

Browse files
committed
Final adjustments
1 parent f9783b8 commit ca81d42

File tree

1 file changed

+38
-37
lines changed

1 file changed

+38
-37
lines changed

qiling/os/posix/const_mapping.py

Lines changed: 38 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@ def _constant_mapping(bits: int, d_map: Mapping[str, int], ret: MutableSequence[
1919
return b_map[bits]
2020

2121
for val, sym in b_map.items():
22-
if val & bits != 0:
22+
if val & bits:
2323
bits ^= val
2424
ret.append(sym)
2525

26-
if bits != 0:
27-
ret.append(str(bits))
26+
if bits:
27+
ret.append(f'{bits:#x}')
2828

2929
return " | ".join(ret)
3030

@@ -225,12 +225,12 @@ def open_flags_mapping(flags, arch):
225225
arm64_open_flags = arm_open_flags
226226

227227
mapping_dict = {
228-
QL_ARCH.X86: linux_open_flags,
229-
QL_ARCH.X8664: linux_open_flags,
230-
QL_ARCH.ARM: arm_open_flags,
231-
QL_ARCH.ARM64: arm64_open_flags,
232-
QL_ARCH.MIPS: mips_open_flags,
233-
QL_OS.MACOS: mac_open_flags,
228+
QL_ARCH.X86: linux_open_flags,
229+
QL_ARCH.X8664: linux_open_flags,
230+
QL_ARCH.ARM: arm_open_flags,
231+
QL_ARCH.ARM64: arm64_open_flags,
232+
QL_ARCH.MIPS: mips_open_flags,
233+
QL_OS.MACOS: mac_open_flags,
234234
}[arch]
235235

236236
ret = ["O_RDONLY"]
@@ -279,43 +279,44 @@ def mmap_prot_mapping(prots):
279279

280280
def socket_type_mapping(t, arch):
281281
socket_type_map = {
282-
QL_ARCH.X86: linux_socket_types,
283-
QL_ARCH.X8664: linux_socket_types,
284-
QL_ARCH.ARM: arm_socket_types,
285-
QL_ARCH.ARM_THUMB: arm_socket_types,
286-
QL_ARCH.ARM64: arm_socket_types,
287-
QL_ARCH.MIPS: mips_socket_types,
288-
QL_OS.MACOS: linux_socket_types,
282+
QL_ARCH.X86: linux_socket_types,
283+
QL_ARCH.X8664: linux_socket_types,
284+
QL_ARCH.ARM: arm_socket_types,
285+
QL_ARCH.ARM_THUMB: arm_socket_types,
286+
QL_ARCH.ARM64: arm_socket_types,
287+
QL_ARCH.MIPS: mips_socket_types,
288+
QL_OS.MACOS: linux_socket_types,
289289
}[arch]
290290

291291
# https://code.woboq.org/linux/linux/net/socket.c.html#1363
292292
t &= SOCK_TYPE_MASK
293+
293294
return _constant_mapping(t, socket_type_map, single_mapping=True)
294295

295296

296297
def socket_domain_mapping(p, arch):
297298
socket_domain_map = {
298-
QL_ARCH.X86: linux_socket_domain,
299-
QL_ARCH.X8664: linux_socket_domain,
300-
QL_ARCH.ARM: arm_socket_domain,
301-
QL_ARCH.ARM_THUMB: arm_socket_domain,
302-
QL_ARCH.ARM64: arm_socket_domain,
303-
QL_ARCH.MIPS: mips_socket_domain,
304-
QL_OS.MACOS: macos_socket_domain,
299+
QL_ARCH.X86: linux_socket_domain,
300+
QL_ARCH.X8664: linux_socket_domain,
301+
QL_ARCH.ARM: arm_socket_domain,
302+
QL_ARCH.ARM_THUMB: arm_socket_domain,
303+
QL_ARCH.ARM64: arm_socket_domain,
304+
QL_ARCH.MIPS: mips_socket_domain,
305+
QL_OS.MACOS: macos_socket_domain,
305306
}[arch]
306307

307308
return _constant_mapping(p, socket_domain_map, single_mapping=True)
308309

309310

310311
def socket_level_mapping(t, arch):
311312
socket_level_map = {
312-
QL_ARCH.X86: linux_socket_level,
313-
QL_ARCH.X8664: linux_socket_level,
314-
QL_ARCH.ARM: arm_socket_level,
315-
QL_ARCH.ARM_THUMB: arm_socket_level,
316-
QL_ARCH.ARM64: arm_socket_level,
317-
QL_ARCH.MIPS: mips_socket_level,
318-
QL_OS.MACOS: linux_socket_level,
313+
QL_ARCH.X86: linux_socket_level,
314+
QL_ARCH.X8664: linux_socket_level,
315+
QL_ARCH.ARM: arm_socket_level,
316+
QL_ARCH.ARM_THUMB: arm_socket_level,
317+
QL_ARCH.ARM64: arm_socket_level,
318+
QL_ARCH.MIPS: mips_socket_level,
319+
QL_OS.MACOS: linux_socket_level,
319320
}[arch]
320321

321322
return _constant_mapping(t, socket_level_map, single_mapping=True)
@@ -337,13 +338,13 @@ def socket_ip_option_mapping(t, arch):
337338

338339
def socket_option_mapping(t, arch):
339340
socket_option_map = {
340-
QL_ARCH.X86: linux_socket_options,
341-
QL_ARCH.X8664: linux_socket_options,
342-
QL_ARCH.ARM: arm_socket_options,
343-
QL_ARCH.ARM_THUMB: arm_socket_options,
344-
QL_ARCH.ARM64: arm_socket_options,
345-
QL_ARCH.MIPS: mips_socket_options,
346-
QL_OS.MACOS: linux_socket_options,
341+
QL_ARCH.X86: linux_socket_options,
342+
QL_ARCH.X8664: linux_socket_options,
343+
QL_ARCH.ARM: arm_socket_options,
344+
QL_ARCH.ARM_THUMB: arm_socket_options,
345+
QL_ARCH.ARM64: arm_socket_options,
346+
QL_ARCH.MIPS: mips_socket_options,
347+
QL_OS.MACOS: linux_socket_options,
347348
}[arch]
348349

349350
return _constant_mapping(t, socket_option_map, single_mapping=True)

0 commit comments

Comments
 (0)