Files
archived-mbuild/mbuild/scanner.py
Mark Charney 7a97205bc5 scanner: file open idiom
Change-Id: Ic7e574227de72fd8068acb8af83b345b5b2dbf97
(cherry picked from commit 3ee19239e31f468eb380303885abef671baaac3e)
2017-06-12 14:45:48 -04:00

114 lines
3.4 KiB
Python
Executable File

# -*- python -*-
# Mark Charney
#BEGIN_LEGAL
#
#Copyright (c) 2016 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#END_LEGAL
"""Function for header include scanning"""
from __future__ import print_function
import re
import os
import sys
from . import base
from . import util
class mbuild_header_record_t:
"""Stores information about headers that we find"""
def __init__(self, fn, found=True):
self.file_name = fn
self.system = False
self.found = found
def __str__(self):
s = ''
s = self.file_name
if self.system:
s += ' system'
if not self.found:
s += ' not-found'
return s
def mbuild_compute_path(hname, search_path):
"""Return the full path of the header hname, if found and None
otherwise. Search the path in order and see if we find the file"""
for p in search_path:
tname = util.join(p,hname)
tname = os.path.realpath(tname)
#mbuild_base.msgb("TESTING", tname)
if os.path.exists(tname):
return tname
return None
# FIXME: ignoring system headers for now.
mbuild_include_pattern = re.compile(r'^[ \t]*#[ \t]*include[ \t]+"(?P<hdr>[^"]+)"')
mbuild_nasm_include_pattern = re.compile(r'^[ \t]*%include[ \t]+"(?P<hdr>[^"]+)"')
is_py2 = sys.version[0] == '2'
def _open_errors(fn):
if is_py2:
return open(fn, 'r')
else:
return open(fn, 'r', errors='ignore')
def mbuild_scan(fn, search_path):
"""Given a file name fn, and a list of search paths, scan for
headers in fn and return a list of mbuild_header_record_t's. The
header records indicate if the file is a system include based on
<> symbols or if the file was missing. If the file cannot be
found, we assume it is in the assumed_directory."""
global mbuild_include_pattern
global mbuild_nasm_include_pattern
all_names = []
if not os.path.exists(fn):
return all_names
source_path = os.path.dirname(fn)
if source_path == '':
source_path = '.'
aug_search_path = [source_path] + search_path
with _open_errors(fn) as f:
for line in f:
#print (line)
hgroup = mbuild_include_pattern.match(line)
if not hgroup:
hgroup = mbuild_nasm_include_pattern.match(line)
if hgroup:
hname = hgroup.group('hdr')
#print (hname)
full_name = mbuild_compute_path(hname, aug_search_path)
if full_name:
hr = mbuild_header_record_t(full_name)
else:
hr = mbuild_header_record_t(hname, found=False)
all_names.append(hr)
return all_names
def _test_scan():
paths = ["/home/mjcharne/proj/learn/" ]
all_headers = mbuild_scan("/home/mjcharne/proj/learn/foo.cpp", paths)
for hr in all_headers:
print (hr)
if __name__ == '__main__':
_test_scan()