mirror of
https://github.com/mii443/qemu.git
synced 2025-09-01 14:49:23 +00:00
94dfc0f343
This started as a simple script that scanned for regular expressions, but became more and more complex when exceptions to the rules were found. I don't know if this should be maintained in the QEMU source tree long term (maybe it can be reused for other code transformations that Coccinelle can't handle). In either case, this is included as part of the patch series to document how exactly the automated code transformations in the next patches were done. Signed-off-by: Eduardo Habkost <ehabkost@redhat.com> Message-Id: <20200831210740.126168-7-ehabkost@redhat.com> Signed-off-by: Eduardo Habkost <ehabkost@redhat.com>
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
# Copyright (C) 2020 Red Hat Inc.
|
|
#
|
|
# Authors:
|
|
# Eduardo Habkost <ehabkost@redhat.com>
|
|
#
|
|
# This work is licensed under the terms of the GNU GPL, version 2. See
|
|
# the COPYING file in the top-level directory.
|
|
from tempfile import NamedTemporaryFile
|
|
from .patching import FileInfo, FileMatch, Patch, FileList
|
|
from .regexps import *
|
|
|
|
class BasicPattern(FileMatch):
|
|
regexp = '[abc]{3}'
|
|
|
|
@property
|
|
def name(self):
|
|
return self.group(0)
|
|
|
|
def replacement(self) -> str:
|
|
# replace match with the middle character repeated 5 times
|
|
return self.group(0)[1].upper()*5
|
|
|
|
def test_pattern_patching():
|
|
of = NamedTemporaryFile('wt')
|
|
of.writelines(['one line\n',
|
|
'this pattern will be patched: defbbahij\n',
|
|
'third line\n',
|
|
'another pattern: jihaabfed'])
|
|
of.flush()
|
|
|
|
files = FileList()
|
|
f = FileInfo(files, of.name)
|
|
f.load()
|
|
f.scan_for_matches()
|
|
matches = f.matches_of_type(BasicPattern)
|
|
assert len(matches) == 2
|
|
p2 = matches[1]
|
|
|
|
# manually add patch, to see if .append() works:
|
|
f.patches.append(p2.append('XXX'))
|
|
|
|
# apply all patches:
|
|
f.gen_patches()
|
|
patched = f.get_patched_content()
|
|
assert patched == ('one line\n'+
|
|
'this pattern will be patched: defBBBBBhij\n'+
|
|
'third line\n'+
|
|
'another pattern: jihAAAAAXXXfed')
|
|
|
|
class Function(FileMatch):
|
|
regexp = S(r'BEGIN\s+', NAMED('name', RE_IDENTIFIER), r'\n',
|
|
r'(.*\n)*?END\n')
|
|
|
|
class Statement(FileMatch):
|
|
regexp = S(r'^\s*', NAMED('name', RE_IDENTIFIER), r'\(\)\n')
|
|
|
|
def test_container_match():
|
|
of = NamedTemporaryFile('wt')
|
|
of.writelines(['statement1()\n',
|
|
'statement2()\n',
|
|
'BEGIN function1\n',
|
|
' statement3()\n',
|
|
' statement4()\n',
|
|
'END\n',
|
|
'BEGIN function2\n',
|
|
' statement5()\n',
|
|
' statement6()\n',
|
|
'END\n',
|
|
'statement7()\n'])
|
|
of.flush()
|
|
|
|
files = FileList()
|
|
f = FileInfo(files, of.name)
|
|
f.load()
|
|
assert len(f.matches_of_type(Function)) == 2
|
|
print(' '.join(m.name for m in f.matches_of_type(Statement)))
|
|
assert len(f.matches_of_type(Statement)) == 7
|
|
|
|
f1 = f.find_match(Function, 'function1')
|
|
f2 = f.find_match(Function, 'function2')
|
|
st1 = f.find_match(Statement, 'statement1')
|
|
st2 = f.find_match(Statement, 'statement2')
|
|
st3 = f.find_match(Statement, 'statement3')
|
|
st4 = f.find_match(Statement, 'statement4')
|
|
st5 = f.find_match(Statement, 'statement5')
|
|
st6 = f.find_match(Statement, 'statement6')
|
|
st7 = f.find_match(Statement, 'statement7')
|
|
|
|
assert not f1.contains(st1)
|
|
assert not f1.contains(st2)
|
|
assert not f1.contains(st2)
|
|
assert f1.contains(st3)
|
|
assert f1.contains(st4)
|
|
assert not f1.contains(st5)
|
|
assert not f1.contains(st6)
|
|
assert not f1.contains(st7)
|
|
|
|
assert not f2.contains(st1)
|
|
assert not f2.contains(st2)
|
|
assert not f2.contains(st2)
|
|
assert not f2.contains(st3)
|
|
assert not f2.contains(st4)
|
|
assert f2.contains(st5)
|
|
assert f2.contains(st6)
|
|
assert not f2.contains(st7)
|