slither-format: clean code

pull/238/head
Josselin 6 years ago
parent eaf1c99ffe
commit 5b0a788766
  1. 2
      utils/slither_format/README.md
  2. 32
      utils/slither_format/slither_format.py

@ -29,7 +29,7 @@ Slither detectors highlight names, context and source-mapping of code constructs
Run Slither-format on a single file:
```
$ python3 -m slither-format ./utils/slither_format/tests/test_data/constant.sol
$ slither-format ./utils/slither_format/tests/test_data/constant.sol
```
This produces `constant.sol.format` file which has all the feature replacements.

@ -65,10 +65,14 @@ def sort_and_flag_overlapping_patches(patches):
patches[file][j+1] = patches[file][j]
patches[file][j] = temp
# Overlap check
if ((int(patches[file][j]['start']) >= int(patches[file][j+1]['start']) and
int(patches[file][j]['start']) <= int(patches[file][j+1]['end'])) or
(int(patches[file][j+1]['start']) >= int(patches[file][j]['start']) and
int(patches[file][j+1]['start']) <= int(patches[file][j]['end']))):
current = patches[file][j]
current_start = int(current['start'])
current_end = int(current['end'])
next = patches[file][j+1]
next_start = int(next['start'])
next_end = int(next['start'])
if ((current_start >= next_start and current_start <= next_end) or
(next_start >= current_start and next_start <= current_end)):
patches[file][j]['overlaps'] = "Yes"
patches[file][j+1]['overlaps'] = "Yes"
@ -92,22 +96,26 @@ def prune_overlapping_patches(args, patches):
def generate_patch_files(slither, patches):
for file in patches:
_in_file = file
if patches[file]:
in_file_str = slither.source_code[patches[file][0]['file']].encode('utf-8')
current_patches = patches[file]
if current_patches:
in_file_str = slither.source_code[current_patches[0]['file']].encode('utf-8')
out_file_str = ""
for i in range(len(patches[file])):
for i in range(len(current_patches)):
if i != 0:
out_file_str += in_file_str[int(patches[file][i-1]['end']):int(patches[file][i]['start'])].decode('utf-8')
out_file_str += in_file_str[int(current_patches[i-1]['end']):int(current_patches[i]['start'])].decode('utf-8')
else:
out_file_str += in_file_str[:int(patches[file][i]['start'])].decode('utf-8')
out_file_str += patches[file][i]['new_string']
if (i == (len(patches[file]) - 1)):
out_file_str += in_file_str[int(patches[file][i]['end']):].decode('utf-8')
out_file_str += in_file_str[:int(current_patches[i]['start'])].decode('utf-8')
out_file_str += current_patches[i]['new_string']
if (i == (len(current_patches) - 1)):
out_file_str += in_file_str[int(current_patches[i]['end']):].decode('utf-8')
logger.info(f'Output new file in {_in_file+".format"}')
out_file = open(_in_file+".format",'w')
out_file.write(out_file_str)
out_file.close()
patch_file_name = _in_file + ".format.patch"
outFD = open(patch_file_name,"w")
logger.info(f'Output new file in {patch_file_name}')
p1 = subprocess.Popen(['diff', '-u', _in_file, _in_file+".format"], stdout=outFD)
p1.wait()
outFD.close()

Loading…
Cancel
Save