if parts[0] not in ["r", "p", "t", "u"]:
self._bail("probe type must be 'p', 'r', 't', or 'u'" +
" but got '%s'" % parts[0])
- if re.match(r"\w+\(.*\)", parts[2]) is None:
+ if re.match(r"\S+\(.*\)", parts[2]) is None:
self._bail(("function signature '%s' has an invalid " +
"format") % parts[2])
self._bail("no exprs specified")
self.exprs = exprs.split(',')
+ def _make_valid_identifier(self, ident):
+ return re.sub(r'[^A-Za-z0-9_]', '_', ident)
+
def __init__(self, tool, type, specifier):
self.usdt_ctx = None
self.streq_functions = ""
self.tp_event = self.function
elif self.probe_type == "u":
self.library = parts[1]
- self.probe_func_name = "%s_probe%d" % \
- (self.function, Probe.next_probe_index)
+ self.probe_func_name = self._make_valid_identifier(
+ "%s_probe%d" % \
+ (self.function, Probe.next_probe_index))
self._enable_usdt_probe()
else:
self.library = parts[1]
self.entry_probe_required = self.probe_type == "r" and \
(any(map(check, self.exprs)) or check(self.filter))
- self.probe_func_name = "%s_probe%d" % \
- (self.function, Probe.next_probe_index)
- self.probe_hash_name = "%s_hash%d" % \
- (self.function, Probe.next_probe_index)
+ self.probe_func_name = self._make_valid_identifier(
+ "%s_probe%d" % \
+ (self.function, Probe.next_probe_index))
+ self.probe_hash_name = self._make_valid_identifier(
+ "%s_hash%d" % \
+ (self.function, Probe.next_probe_index))
Probe.next_probe_index += 1
def _enable_usdt_probe(self):
self.probe_num = Probe.probe_count
self.probe_name = "probe_%s_%d" % \
(self._display_function(), self.probe_num)
- if self.probe_name.find(".") > 0: # for golang
- self.probe_name = self.probe_name.replace(".", "_DOT_")
+ self.probe_name = re.sub(r'[^A-Za-z0-9_]', '_', self.probe_name)
def __str__(self):
return "%s:%s:%s FLT=%s ACT=%s/%s" % (self.probe_type,