X-Git-Url: https://git.ozlabs.org/?p=bitfield;a=blobdiff_plain;f=bitfield;h=1a03ddb96419c9f2f77ea08830a21c6d25665d14;hp=3651677f86c29ab8a90d9a67df14fa086ffa4503;hb=369073ac299df29a428871938851e56c8acebe91;hpb=27f2cb1942b6391d3081016260e0d29b0ffa74f6;ds=inline

diff --git a/bitfield b/bitfield
index 3651677..1a03ddb 100644
--- a/bitfield
+++ b/bitfield
@@ -275,19 +275,19 @@ def main():
 		print "No such register '%s'" % reg_id
 		return 1
 
-	r = regs[reg_id]
-	print "decoding as %s" % r.name
+	reg = regs[reg_id]
+	print "decoding as %s" % reg.name
 
 	if args:
-		values = args
+		value_iter = args.__iter__()
 	else:
-		try:
-			values = sys.stdin.readlines()
-		except KeyboardInterrupt, e:
-			return
-
-	for value in values:
-		decode_value(r, value.strip(), options)
+		value_iter = iter(sys.stdin.readline, '')
+		
+	try:
+		for value in value_iter:
+			decode_value(reg, value.strip(), options)
+	except KeyboardInterrupt, e:
+		pass
 
 	return 0