Merge pull request iovisor#889 from totally/preprocess_4.10

Fixes bio{snoop,top} on 4.10
linuxzjs · Jan 9, 2017 · a043d5c · a043d5c
2 parents e14519e + bc43a29
commit a043d5c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/tools/biosnoop.py b/tools/biosnoop.py
@@ -106,14 +106,12 @@
  * test, and maintenance burden.
  */
 #ifdef REQ_WRITE
-if (req->cmd_flags & REQ_WRITE) {
+ data.rwflag = !!(req->cmd_flags & REQ_WRITE);
+#elif defined(REQ_OP_SHIFT)
+ data.rwflag = !!((req->cmd_flags >> REQ_OP_SHIFT) == REQ_OP_WRITE);
 #else
-if ((req->cmd_flags >> REQ_OP_SHIFT) == REQ_OP_WRITE) {
+ data.rwflag = !!((req->cmd_flags & REQ_OP_MASK) == REQ_OP_WRITE);
 #endif
- data.rwflag = 1;
- } else {
- data.rwflag = 0;
- }
 
  events.perf_submit(ctx, &data, sizeof(data));
  start.delete(&req);

diff --git a/tools/biotop.py b/tools/biotop.py
@@ -137,8 +137,10 @@ def signal_ignore(signal, frame):
  */
 #ifdef REQ_WRITE
  info.rwflag = !!(req->cmd_flags & REQ_WRITE);
-#else
+#elif defined(REQ_OP_SHIFT)
  info.rwflag = !!((req->cmd_flags >> REQ_OP_SHIFT) == REQ_OP_WRITE);
+#else
+ info.rwflag = !!((req->cmd_flags & REQ_OP_MASK) == REQ_OP_WRITE);
 #endif
 
  whop = whobyreq.lookup(&req);