diff options
author | Frank Ch. Eigler <fche@elastic.org> | 2010-02-23 20:38:56 -0500 |
---|---|---|
committer | Frank Ch. Eigler <fche@elastic.org> | 2010-02-23 20:38:56 -0500 |
commit | c7005ee1ba902de01a8863661a2031bfd45d3d40 (patch) | |
tree | e8a7bfa55c9afcfac71b5aafa5b54256f356c81a /tapset-procfs.cxx | |
parent | 30263a7389d5c2712536b74656193708bbc64d49 (diff) | |
parent | 10328bcf86d49e5de83c5b22890204b5874740a9 (diff) | |
download | systemtap-steved-c7005ee1ba902de01a8863661a2031bfd45d3d40.tar.gz systemtap-steved-c7005ee1ba902de01a8863661a2031bfd45d3d40.tar.xz systemtap-steved-c7005ee1ba902de01a8863661a2031bfd45d3d40.zip |
Merge branch 'master' of ssh://sources.redhat.com/git/systemtap
* 'master' of ssh://sources.redhat.com/git/systemtap:
PR10719 part 1: Partial constant folding
Simplify null_statement construction
Fixed PR 11269 by properly handling mmap syscall 'fd' argument.
Removed rvalue operator check.
PR 10690 (partial fix). Handle '.=' operator in procfs probes.
Conflicts:
tapsets.h
Diffstat (limited to 'tapset-procfs.cxx')
-rw-r--r-- | tapset-procfs.cxx | 57 |
1 files changed, 43 insertions, 14 deletions
diff --git a/tapset-procfs.cxx b/tapset-procfs.cxx index c4eb54f3..1926582b 100644 --- a/tapset-procfs.cxx +++ b/tapset-procfs.cxx @@ -74,9 +74,7 @@ public: struct procfs_var_expanding_visitor: public var_expanding_visitor { procfs_var_expanding_visitor(systemtap_session& s, const string& pn, - string path, bool write_probe): - sess (s), probe_name (pn), path (path), write_probe (write_probe), - target_symbol_seen (false) {} + string path, bool write_probe); systemtap_session& sess; string probe_name; @@ -342,6 +340,18 @@ procfs_derived_probe_group::emit_module_exit (systemtap_session& s) } +procfs_var_expanding_visitor::procfs_var_expanding_visitor (systemtap_session& s, + const string& pn, + string path, + bool write_probe): + sess (s), probe_name (pn), path (path), write_probe (write_probe), + target_symbol_seen (false) +{ + // procfs probes can also handle '.='. + valid_ops.insert (".="); +} + + void procfs_var_expanding_visitor::visit_target_symbol (target_symbol* e) { @@ -371,21 +381,40 @@ procfs_var_expanding_visitor::visit_target_symbol (target_symbol* e) embeddedcode *ec = new embeddedcode; ec->tok = e->tok; - string fname = (string(lvalue ? "_procfs_value_set" : "_procfs_value_get")); + string fname; string locvalue = "CONTEXT->data"; if (! lvalue) - ec->code = string(" struct _stp_procfs_data *data = (struct _stp_procfs_data *)(") + locvalue + string("); /* pure */\n") + { + fname = "_procfs_value_get"; + ec->code = string(" struct _stp_procfs_data *data = (struct _stp_procfs_data *)(") + locvalue + string("); /* pure */\n") - + string(" _stp_copy_from_user(THIS->__retvalue, data->buffer, data->count);\n") - + string(" THIS->__retvalue[data->count] = '\\0';\n"); - else - ec->code = string("int bytes = 0;\n") - + string(" struct _stp_procfs_data *data = (struct _stp_procfs_data *)(") + locvalue + string(");\n") - + string(" bytes = strnlen(THIS->value, MAXSTRINGLEN - 1);\n") - + string(" memcpy((void *)data->buffer, THIS->value, bytes);\n") - + string(" data->buffer[bytes] = '\\0';\n") - + string(" data->count = bytes;\n"); + + string(" _stp_copy_from_user(THIS->__retvalue, data->buffer, data->count);\n") + + string(" THIS->__retvalue[data->count] = '\\0';\n"); + } + else // lvalue + { + if (*op == "=") + { + fname = "_procfs_value_set"; + ec->code = string("struct _stp_procfs_data *data = (struct _stp_procfs_data *)(") + locvalue + string(");\n") + + string(" strlcpy(data->buffer, THIS->value, MAXSTRINGLEN);\n") + + string(" data->count = strlen(data->buffer);\n"); + } + else if (*op == ".=") + { + fname = "_procfs_value_append"; + ec->code = string("struct _stp_procfs_data *data = (struct _stp_procfs_data *)(") + locvalue + string(");\n") + + string(" strlcat(data->buffer, THIS->value, MAXSTRINGLEN);\n") + + string(" data->count = strlen(data->buffer);\n"); + } + else + { + throw semantic_error ("Only the following assign operators are" + " implemented on procfs read target variables:" + " '=', '.='", e->tok); + } + } fdecl->name = fname; fdecl->body = ec; |