Factor out the main() function.

git-svn-id: https://llvm.org/svn/llvm-project/compiler-rt/trunk@161046 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 16 insertions, 9 deletions

diff --git a/lib/asan/scripts/asan_symbolize.py b/lib/asan/scripts/asan_symbolize.py
index e4897d0c..346dba12 100755
--- a/lib/asan/scripts/asan_symbolize.py
+++ b/lib/asan/scripts/asan_symbolize.py
@@ -17,6 +17,7 @@ pipes = {}
 filetypes = {}
 DEBUG=False
 
+
 def fix_filename(file_name):
   for path_to_cut in sys.argv[1:]:
     file_name = re.sub(".*" + path_to_cut, "", file_name)
@@ -116,12 +117,18 @@ def symbolize_atos(line):
   else:
     print line.rstrip()
 
-system = os.uname()[0]
-if system in ['Linux', 'Darwin']:
-  for line in sys.stdin:
-    if system == 'Linux':
-      symbolize_addr2line(line)
-    elif system == 'Darwin':
-      symbolize_atos(line)
-else:
-  print 'Unknown system: ', system
+
+def main():
+  system = os.uname()[0]
+  if system in ['Linux', 'Darwin']:
+    for line in sys.stdin:
+      if system == 'Linux':
+        symbolize_addr2line(line)
+      elif system == 'Darwin':
+        symbolize_atos(line)
+  else:
+    print 'Unknown system: ', system
+
+
+if __name__ == '__main__':
+  main()