servicemp3.cpp: more simple/flexible streaming detection
[enigma2.git] / 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 }