1#!/usr/bin/python2.4 2# 3# 4# Copyright 2009, The Android Open Source Project 5# 6# Licensed under the Apache License, Version 2.0 (the "License"); 7# you may not use this file except in compliance with the License. 8# You may obtain a copy of the License at 9# 10# http://www.apache.org/licenses/LICENSE-2.0 11# 12# Unless required by applicable law or agreed to in writing, software 13# distributed under the License is distributed on an "AS IS" BASIS, 14# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15# See the License for the specific language governing permissions and 16# limitations under the License. 17 18"""In memory representation of Android.mk file. 19 20Specifications for Android.mk can be found at 21development/ndk/docs/ANDROID-MK.txt 22""" 23 24import os 25import re 26from sets import Set 27 28import logger 29 30class AndroidMK(object): 31 """In memory representation of Android.mk file.""" 32 33 _RE_INCLUDE = re.compile(r'include\s+\$\((.+)\)') 34 _RE_VARIABLE_REF = re.compile(r'\$\((.+)\)') 35 _VAR_DELIMITER = ":=" 36 FILENAME = "Android.mk" 37 CERTIFICATE = "LOCAL_CERTIFICATE" 38 PACKAGE_NAME = "LOCAL_PACKAGE_NAME" 39 40 def __init__(self): 41 self._includes = Set() # variables included in makefile 42 self._variables = {} # variables defined in makefile 43 self._has_gtestlib = False 44 45 def _ProcessMKLine(self, line): 46 """Add a variable definition or include. 47 48 Ignores unrecognized lines. 49 50 Args: 51 line: line of text from makefile 52 """ 53 m = self._RE_INCLUDE.match(line) 54 if m: 55 self._includes.add(m.group(1)) 56 else: 57 parts = line.split(self._VAR_DELIMITER) 58 if len(parts) > 1: 59 self._variables[parts[0].strip()] = parts[1].strip() 60 # hack, look for explicit mention of libgtest_main 61 if line.find('libgtest_main') != -1: 62 self._has_gtestlib = True 63 64 def GetVariable(self, identifier): 65 """Retrieve makefile variable. 66 67 Args: 68 identifier: name of variable to retrieve 69 Returns: 70 value of specified identifier, None if identifier not found in makefile 71 """ 72 # use dict.get(x) rather than dict[x] to avoid KeyError exception, 73 # so None is returned if identifier not found 74 return self._variables.get(identifier, None) 75 76 def GetExpandedVariable(self, identifier): 77 """Retrieve makefile variable. 78 79 If variable value refers to another variable, recursively expand it to 80 find its literal value 81 82 Args: 83 identifier: name of variable to retrieve 84 Returns: 85 value of specified identifier, None if identifier not found in makefile 86 """ 87 # use dict.get(x) rather than dict[x] to avoid KeyError exception, 88 # so None is returned if identifier not found 89 return self.__RecursiveGetVariable(identifier, Set()) 90 91 def __RecursiveGetVariable(self, identifier, visited_variables): 92 variable_value = self.GetVariable(identifier) 93 if not variable_value: 94 return None 95 if variable_value in visited_variables: 96 raise RuntimeError('recursive loop found for makefile variable %s' 97 % variable_value) 98 m = self._RE_VARIABLE_REF.match(variable_value) 99 if m: 100 logger.SilentLog('Found variable ref %s for identifier %s' 101 % (variable_value, identifier)) 102 variable_ref = m.group(1) 103 visited_variables.add(variable_ref) 104 return self.__RecursiveGetVariable(variable_ref, visited_variables) 105 else: 106 return variable_value 107 108 def HasInclude(self, identifier): 109 """Check variable is included in makefile. 110 111 Args: 112 identifer: name of variable to check 113 Returns: 114 True if identifer is included in makefile, otherwise False 115 """ 116 return identifier in self._includes 117 118 def IncludesMakefilesUnder(self): 119 """Check if makefile has a 'include makefiles under here' rule""" 120 return self.HasInclude('call all-makefiles-under,$(LOCAL_PATH)') 121 122 def HasJavaLibrary(self, library_name): 123 """Check if library is specified as a local java library in makefile. 124 125 Args: 126 library_name: name of library to check 127 Returns: 128 True if library_name is included in makefile, otherwise False 129 """ 130 java_lib_string = self.GetExpandedVariable('LOCAL_JAVA_LIBRARIES') 131 if java_lib_string: 132 java_libs = java_lib_string.split(' ') 133 return library_name in java_libs 134 return False 135 136 def HasGTest(self): 137 """Check if makefile includes rule to build a native gtest. 138 139 Returns: 140 True if rule to build native test is in makefile, otherwise False 141 """ 142 return self._has_gtestlib or self.HasInclude('BUILD_NATIVE_TEST') 143 144 def _ParseMK(self, mk_path): 145 """Parse Android.mk at the specified path. 146 147 Args: 148 mk_path: path to Android.mk 149 Raises: 150 IOError: Android.mk cannot be found at given path, or cannot be opened 151 for reading 152 """ 153 mk = open(mk_path) 154 for line in mk: 155 self._ProcessMKLine(line) 156 mk.close() 157 158 159def CreateAndroidMK(path, filename=AndroidMK.FILENAME): 160 """Factory method for creating a AndroidMK. 161 162 Args: 163 path: the directory of the make file 164 filename: the filename of the makefile 165 166 Return: 167 the AndroidMK or None if there was no file present 168 """ 169 mk_path = os.path.join(path, filename) 170 if os.path.isfile(mk_path): 171 mk = AndroidMK() 172 mk._ParseMK(mk_path) 173 return mk 174 else: 175 return None 176