lib/base/ioprio.cpp

   1 #include <lib/base/ioprio.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <errno.h>
   5 #include <getopt.h>
   6 #include <unistd.h>
   7 #include <sys/ptrace.h>
   8 #include <asm/unistd.h>
   9
  10 #include <lib/base/eerror.h>
  11
  12 extern "C" int sys_ioprio_set(int, int, int);
  13 extern "C" int sys_ioprio_get(int, int);
  14
  15 #ifndef __NR_ioprio_set
  16         #if defined(__i386__)
  17                 #define __NR_ioprio_set         289
  18                 #define __NR_ioprio_get         290
  19         #elif defined(__ppc__) || defined(__powerpc__)
  20                 #define __NR_ioprio_set         273
  21                 #define __NR_ioprio_get         274
  22         #elif defined(__x86_64__)
  23                 #define __NR_ioprio_set         251
  24                 #define __NR_ioprio_get         252
  25         #elif defined(__ia64__)
  26                 #define __NR_ioprio_set         1274
  27                 #define __NR_ioprio_get         1275
  28         #elif defined(__mips__)
  29                 #define __NR_ioprio_set         4284
  30                 #define __NR_ioprio_get         4285
  31         #else
  32                 #error "Unsupported arch"
  33         #endif
  34 #endif
  35
  36 #if defined(_syscall3) && defined(_syscall2)
  37
  38 _syscall3(int, ioprio_set, int, which, int, who, int, ioprio);
  39 _syscall2(int, ioprio_get, int, which, int, who);
  40
  41 #else
  42
  43 static inline int ioprio_set(int which, int who, int ioprio)
  44 {
  45         return syscall(__NR_ioprio_set, which, who, ioprio);
  46 }
  47
  48 static inline int ioprio_get(int which, int who)
  49 {
  50         return syscall(__NR_ioprio_get, which, who);
  51 }
  52
  53 #endif
  54
  55 #define IOPRIO_CLASS_SHIFT      13
  56
  57 enum {
  58         IOPRIO_WHO_PROCESS = 1,
  59         IOPRIO_WHO_PGRP,
  60         IOPRIO_WHO_USER,
  61 };
  62
  63 const char *to_prio[] = { "none", "realtime", "best-effort", "idle", };
  64
  65 void setIoPrio(int prio_class, int prio)
  66 {
  67         if (prio_class < 0 || prio_class > 3)
  68         {
  69                 eDebug("prio class(%d) out of valid range (0..3)", prio_class);
  70                 return;
  71         }
  72         if (prio < 0 || prio > 7)
  73         {
  74                 eDebug("prio level(%d) out of range (0..7)", prio);
  75                 return;
  76         }
  77         if (ioprio_set(IOPRIO_WHO_PROCESS, 0 /*pid 0 .. current process*/, prio | prio_class << IOPRIO_CLASS_SHIFT) == -1)
  78                 eDebug("setIoPrio failed (%m) !");
  79         else
  80                 eDebug("setIoPrio %s level %d ok", to_prio[prio_class], prio);
  81 }
  82
  83 void printIoPrio()
  84 {
  85         int pid = 0, ioprio = ioprio_get(IOPRIO_WHO_PROCESS, pid);
  86
  87         eDebug("pid=%d, %d", pid, ioprio);
  88
  89         if (ioprio == -1)
  90                 eDebug("ioprio_get(%m)");
  91         else {
  92                 int ioprio_class = ioprio >> IOPRIO_CLASS_SHIFT;
  93                 ioprio = ioprio & 0xff;
  94                 eDebug("%s: prio %d", to_prio[ioprio_class], ioprio);
  95         }
  96 }