iosched.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * Copyright 2009, 2010 Daniel Borkmann.
   4  * Copyright 2010 Marek Polacek.
   5  * Subject to the GPL, version 2.
   6  */
   7
   8 #include <sys/syscall.h>
   9
  10 #include "iosched.h"
  11 #include "die.h"
  12
  13 #define IOPRIO_CLASS_SHIFT      13
  14
  15 enum {
  16         ioprio_class_none,
  17         ioprio_class_rt,
  18         ioprio_class_be,
  19         ioprio_class_idle,
  20 };
  21
  22 enum {
  23         ioprio_who_process = 1,
  24         ioprio_who_pgrp,
  25         ioprio_who_user,
  26 };
  27
  28 static const char *const to_prio[] = {
  29         "none",
  30         "realtime",
  31         "best-effort",
  32         "idle",
  33 };
  34
  35 static inline int ioprio_set(int which, int who, int ioprio)
  36 {
  37         return syscall(SYS_ioprio_set, which, who, ioprio);
  38 }
  39
  40 static inline int ioprio_get(int which, int who)
  41 {
  42         return syscall(SYS_ioprio_get, which, who);
  43 }
  44
  45 static void ioprio_setpid(pid_t pid, int ioprio, int ioclass)
  46 {
  47         int ret = ioprio_set(ioprio_who_process, pid,
  48                              ioprio | ioclass << IOPRIO_CLASS_SHIFT);
  49         if (ret < 0)
  50                 panic("Failed to set io prio for pid!\n");
  51 }
  52
  53 void ioprio_print(void)
  54 {
  55         int ioprio = ioprio_get(ioprio_who_process, getpid());
  56         if (ioprio < 0)
  57                 panic("Failed to fetch io prio for pid!\n");
  58         else {
  59                 int ioclass = ioprio >> IOPRIO_CLASS_SHIFT;
  60                 if (ioclass != ioprio_class_idle) {
  61                         ioprio &= 0xff;
  62                         printf("%s: prio %d\n", to_prio[ioclass], ioprio);
  63                 } else
  64                         printf("%s\n", to_prio[ioclass]);
  65         }
  66 }
  67
  68 void set_ioprio_rt(void)
  69 {
  70         ioprio_setpid(getpid(), 4, ioprio_class_rt);
  71 }
  72
  73 void set_ioprio_be(void)
  74 {
  75         ioprio_setpid(getpid(), 4, ioprio_class_be);
  76 }