openharmony/startup_init
1
0
Fork 0
mirror of https://gitee.com/openharmony/startup_init synced 2024-11-23 16:20:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Description:Code optimization

Browse Source 
Feature or Bugfix:Bugfix
Binary Source:No

Signed-off-by: lwx1281857 <linnanmu@h-partners.com>
This commit is contained in:
lwx1281857 2024-07-18 20:20:33 +08:00
parent 61061796b3
commit fe9d533280
1 changed files with 37 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
services/modules/seccomp/scripts/generate_code_from_policy.py
View File

@ -426,6 +426,43 @@ class GenBpfPolicy:
bpf_policy.append(BPF_JEQ.format(syscall_nr, 0, cur_size))
return bpf_policy
@staticmethod
def check_arg_str(arg_atom):
arg_str = arg_atom[0:3]
if arg_str != 'arg':
raise ValidateError('format ERROR, {} is not equal to arg'.format(arg_atom))
arg_id = int(arg_atom[3])
if arg_id not in range(6):
raise ValidateError('arg num out of the scope 0~5')
return arg_id, True
@staticmethod
def check_operation_str(operation_atom):
operation_str = operation_atom
if operation_str not in operation:
operation_str = operation_atom[0]
if operation_str not in operation:
raise ValidateError('operation not in [<, <=, !=, ==, >, >=, &]')
return operation_str, True
#gen bpf (argn & mask) == value
@staticmethod
def gen_mask_equal_bpf(arg_id, mask, value, cur_size):
bpf_policy = []
#high 4 bytes
bpf_policy.append(BPF_LOAD.format(20 + arg_id * 8))
bpf_policy.append(BPF_AND.format('((uint64_t)' + mask + ') >> 32'))
bpf_policy.append(BPF_JEQ.format('((uint64_t)' + value + ') >> 32', 0, cur_size + 4))
#low 4 bytes
bpf_policy.append(BPF_LOAD.format(16 + arg_id * 8))
bpf_policy.append(BPF_AND.format(mask))
bpf_policy.append(BPF_JEQ.format(value, cur_size, cur_size + 1))
return bpf_policy
def update_arch(self, arch):
self.arch = arch
self.syscall_nr_range = []
@ -640,43 +677,6 @@ class GenBpfPolicy:
return bpf_policy
@staticmethod
def check_arg_str(arg_atom):
arg_str = arg_atom[0:3]
if arg_str != 'arg':
raise ValidateError('format ERROR, {} is not equal to arg'.format(arg_atom))
arg_id = int(arg_atom[3])
if arg_id not in range(6):
raise ValidateError('arg num out of the scope 0~5')
return arg_id, True
@staticmethod
def check_operation_str(operation_atom):
operation_str = operation_atom
if operation_str not in operation:
operation_str = operation_atom[0]
if operation_str not in operation:
raise ValidateError('operation not in [<, <=, !=, ==, >, >=, &]')
return operation_str, True
#gen bpf (argn & mask) == value
@staticmethod
def gen_mask_equal_bpf(arg_id, mask, value, cur_size):
bpf_policy = []
#high 4 bytes
bpf_policy.append(BPF_LOAD.format(20 + arg_id * 8))
bpf_policy.append(BPF_AND.format('((uint64_t)' + mask + ') >> 32'))
bpf_policy.append(BPF_JEQ.format('((uint64_t)' + value + ') >> 32', 0, cur_size + 4))
#low 4 bytes
bpf_policy.append(BPF_LOAD.format(16 + arg_id * 8))
bpf_policy.append(BPF_AND.format(mask))
bpf_policy.append(BPF_JEQ.format(value, cur_size, cur_size + 1))
return bpf_policy
def compile_mask_equal_atom(self, atom, cur_size):
bpf_policy = []
left_brace_pos = atom.find('(')