Initial revision
[linux-2.6.git] / scripts / rt-tester / rt-tester.py
1 #!/usr/bin/env python
2 #
3 # rt-mutex tester
4 #
5 # (C) 2006 Thomas Gleixner <tglx@linutronix.de>
6 #
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License version 2 as
9 # published by the Free Software Foundation.
10 #
11 import os
12 import sys
13 import getopt
14 import shutil
15 import string
16
17 # Globals
18 quiet = 0
19 test = 0
20 comments = 0
21
22 sysfsprefix = "/sys/devices/system/rttest/rttest"
23 statusfile = "/status"
24 commandfile = "/command"
25
26 # Command opcodes
27 cmd_opcodes = {
28     "schedother"    : "1",
29     "schedfifo"     : "2",
30     "lock"          : "3",
31     "locknowait"    : "4",
32     "lockint"       : "5",
33     "lockintnowait" : "6",
34     "lockcont"      : "7",
35     "unlock"        : "8",
36     "lockbkl"       : "9",
37     "unlockbkl"     : "10",
38     "signal"        : "11",
39     "resetevent"    : "98",
40     "reset"         : "99",
41     }
42
43 test_opcodes = {
44     "prioeq"        : ["P" , "eq" , None],
45     "priolt"        : ["P" , "lt" , None],
46     "priogt"        : ["P" , "gt" , None],
47     "nprioeq"       : ["N" , "eq" , None],
48     "npriolt"       : ["N" , "lt" , None],
49     "npriogt"       : ["N" , "gt" , None],
50     "unlocked"      : ["M" , "eq" , 0],
51     "trylock"       : ["M" , "eq" , 1],
52     "blocked"       : ["M" , "eq" , 2],
53     "blockedwake"   : ["M" , "eq" , 3],
54     "locked"        : ["M" , "eq" , 4],
55     "opcodeeq"      : ["O" , "eq" , None],
56     "opcodelt"      : ["O" , "lt" , None],
57     "opcodegt"      : ["O" , "gt" , None],
58     "eventeq"       : ["E" , "eq" , None],
59     "eventlt"       : ["E" , "lt" , None],
60     "eventgt"       : ["E" , "gt" , None],
61     }
62
63 # Print usage information
64 def usage():
65     print "rt-tester.py <-c -h -q -t> <testfile>"
66     print " -c    display comments after first command"
67     print " -h    help"
68     print " -q    quiet mode"
69     print " -t    test mode (syntax check)"
70     print " testfile: read test specification from testfile"
71     print " otherwise from stdin"
72     return
73
74 # Print progress when not in quiet mode
75 def progress(str):
76     if not quiet:
77         print str
78
79 # Analyse a status value
80 def analyse(val, top, arg):
81
82     intval = int(val)
83
84     if top[0] == "M":
85         intval = intval / (10 ** int(arg))
86         intval = intval % 10
87         argval = top[2]
88     elif top[0] == "O":
89         argval = int(cmd_opcodes.get(arg, arg))
90     else:
91         argval = int(arg)
92
93     # progress("%d %s %d" %(intval, top[1], argval))
94
95     if top[1] == "eq" and intval == argval:
96         return 1
97     if top[1] == "lt" and intval < argval:
98         return 1
99     if top[1] == "gt" and intval > argval:
100         return 1
101     return 0
102
103 # Parse the commandline
104 try:
105     (options, arguments) = getopt.getopt(sys.argv[1:],'chqt')
106 except getopt.GetoptError, ex:
107     usage()
108     sys.exit(1)
109
110 # Parse commandline options
111 for option, value in options:
112     if option == "-c":
113         comments = 1
114     elif option == "-q":
115         quiet = 1
116     elif option == "-t":
117         test = 1
118     elif option == '-h':
119         usage()
120         sys.exit(0)
121
122 # Select the input source
123 if arguments:
124     try:
125         fd = open(arguments[0])
126     except Exception,ex:
127         sys.stderr.write("File not found %s\n" %(arguments[0]))
128         sys.exit(1)
129 else:
130     fd = sys.stdin
131
132 linenr = 0
133
134 # Read the test patterns
135 while 1:
136
137     linenr = linenr + 1
138     line = fd.readline()
139     if not len(line):
140         break
141
142     line = line.strip()
143     parts = line.split(":")
144
145     if not parts or len(parts) < 1:
146         continue
147
148     if len(parts[0]) == 0:
149         continue
150
151     if parts[0].startswith("#"):
152         if comments > 1:
153             progress(line)
154         continue
155
156     if comments == 1:
157         comments = 2
158
159     progress(line)
160
161     cmd = parts[0].strip().lower()
162     opc = parts[1].strip().lower()
163     tid = parts[2].strip()
164     dat = parts[3].strip()
165
166     try:
167         # Test or wait for a status value
168         if cmd == "t" or cmd == "w":
169             testop = test_opcodes[opc]
170
171             fname = "%s%s%s" %(sysfsprefix, tid, statusfile)
172             if test:
173                 print fname
174                 continue
175
176             while 1:
177                 query = 1
178                 fsta = open(fname, 'r')
179                 status = fsta.readline().strip()
180                 fsta.close()
181                 stat = status.split(",")
182                 for s in stat:
183                     s = s.strip()
184                     if s.startswith(testop[0]):
185                         # Seperate status value
186                         val = s[2:].strip()
187                         query = analyse(val, testop, dat)
188                         break
189                 if query or cmd == "t":
190                     break
191
192             progress("   " + status)
193
194             if not query:
195                 sys.stderr.write("Test failed in line %d\n" %(linenr))
196                 sys.exit(1)
197
198         # Issue a command to the tester
199         elif cmd == "c":
200             cmdnr = cmd_opcodes[opc]
201             # Build command string and sys filename
202             cmdstr = "%s:%s" %(cmdnr, dat)
203             fname = "%s%s%s" %(sysfsprefix, tid, commandfile)
204             if test:
205                 print fname
206                 continue
207             fcmd = open(fname, 'w')
208             fcmd.write(cmdstr)
209             fcmd.close()
210
211     except Exception,ex:
212         sys.stderr.write(str(ex))
213         sys.stderr.write("\nSyntax error in line %d\n" %(linenr))
214         if not test:
215             fd.close()
216             sys.exit(1)
217
218 # Normal exit pass
219 print "Pass"
220 sys.exit(0)
221
222