blob: e74c5963dc7a9db7038e4ad3f5f7de1ed490f20f [file] [log] [blame]
Namhyung Kim12864b32012-04-26 14:15:22 +09001/*
2 * Helper functions for handling target threads/cpus
3 *
4 * Copyright (C) 2012, LG Electronics, Namhyung Kim <namhyung.kim@lge.com>
5 *
6 * Released under the GPL v2.
7 */
8
9#include "target.h"
10#include "debug.h"
11
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090012#include <pwd.h>
Namhyung Kim16ad2ff2012-05-07 14:09:02 +090013#include <string.h>
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090014
Namhyung Kim12864b32012-04-26 14:15:22 +090015
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030016enum target_errno target__validate(struct target *target)
Namhyung Kim12864b32012-04-26 14:15:22 +090017{
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030018 enum target_errno ret = TARGET_ERRNO__SUCCESS;
Namhyung Kim60bbdda2012-05-07 14:09:00 +090019
Namhyung Kim12864b32012-04-26 14:15:22 +090020 if (target->pid)
21 target->tid = target->pid;
22
23 /* CPU and PID are mutually exclusive */
24 if (target->tid && target->cpu_list) {
Namhyung Kim12864b32012-04-26 14:15:22 +090025 target->cpu_list = NULL;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030026 if (ret == TARGET_ERRNO__SUCCESS)
27 ret = TARGET_ERRNO__PID_OVERRIDE_CPU;
Namhyung Kim12864b32012-04-26 14:15:22 +090028 }
29
30 /* UID and PID are mutually exclusive */
31 if (target->tid && target->uid_str) {
Namhyung Kim12864b32012-04-26 14:15:22 +090032 target->uid_str = NULL;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030033 if (ret == TARGET_ERRNO__SUCCESS)
34 ret = TARGET_ERRNO__PID_OVERRIDE_UID;
Namhyung Kim12864b32012-04-26 14:15:22 +090035 }
36
37 /* UID and CPU are mutually exclusive */
38 if (target->uid_str && target->cpu_list) {
Namhyung Kim12864b32012-04-26 14:15:22 +090039 target->cpu_list = NULL;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030040 if (ret == TARGET_ERRNO__SUCCESS)
41 ret = TARGET_ERRNO__UID_OVERRIDE_CPU;
Namhyung Kim12864b32012-04-26 14:15:22 +090042 }
43
Namhyung Kim60bbdda2012-05-07 14:09:00 +090044 /* PID and SYSTEM are mutually exclusive */
45 if (target->tid && target->system_wide) {
Namhyung Kim12864b32012-04-26 14:15:22 +090046 target->system_wide = false;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030047 if (ret == TARGET_ERRNO__SUCCESS)
48 ret = TARGET_ERRNO__PID_OVERRIDE_SYSTEM;
Namhyung Kim12864b32012-04-26 14:15:22 +090049 }
Namhyung Kim60bbdda2012-05-07 14:09:00 +090050
51 /* UID and SYSTEM are mutually exclusive */
52 if (target->uid_str && target->system_wide) {
53 target->system_wide = false;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030054 if (ret == TARGET_ERRNO__SUCCESS)
55 ret = TARGET_ERRNO__UID_OVERRIDE_SYSTEM;
Namhyung Kim60bbdda2012-05-07 14:09:00 +090056 }
57
Adrian Hunter3aa59392013-11-15 15:52:29 +020058 /* THREAD and SYSTEM/CPU are mutually exclusive */
59 if (target->per_thread && (target->system_wide || target->cpu_list)) {
60 target->per_thread = false;
61 if (ret == TARGET_ERRNO__SUCCESS)
62 ret = TARGET_ERRNO__SYSTEM_OVERRIDE_THREAD;
63 }
64
Namhyung Kim60bbdda2012-05-07 14:09:00 +090065 return ret;
Namhyung Kim12864b32012-04-26 14:15:22 +090066}
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090067
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030068enum target_errno target__parse_uid(struct target *target)
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090069{
70 struct passwd pwd, *result;
71 char buf[1024];
72 const char *str = target->uid_str;
73
74 target->uid = UINT_MAX;
75 if (str == NULL)
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030076 return TARGET_ERRNO__SUCCESS;
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090077
78 /* Try user name first */
79 getpwnam_r(str, &pwd, buf, sizeof(buf), &result);
80
81 if (result == NULL) {
82 /*
83 * The user name not found. Maybe it's a UID number.
84 */
85 char *endptr;
86 int uid = strtol(str, &endptr, 10);
87
88 if (*endptr != '\0')
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030089 return TARGET_ERRNO__INVALID_UID;
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090090
91 getpwuid_r(uid, &pwd, buf, sizeof(buf), &result);
92
93 if (result == NULL)
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030094 return TARGET_ERRNO__USER_NOT_FOUND;
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090095 }
96
97 target->uid = result->pw_uid;
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -030098 return TARGET_ERRNO__SUCCESS;
Namhyung Kimdfe78ad2012-05-07 14:09:01 +090099}
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900100
101/*
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300102 * This must have a same ordering as the enum target_errno.
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900103 */
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300104static const char *target__error_str[] = {
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900105 "PID/TID switch overriding CPU",
106 "PID/TID switch overriding UID",
107 "UID switch overriding CPU",
108 "PID/TID switch overriding SYSTEM",
109 "UID switch overriding SYSTEM",
Adrian Hunter3aa59392013-11-15 15:52:29 +0200110 "SYSTEM/CPU switch overriding PER-THREAD",
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900111 "Invalid User: %s",
112 "Problems obtaining information for user %s",
113};
114
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300115int target__strerror(struct target *target, int errnum,
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900116 char *buf, size_t buflen)
117{
118 int idx;
119 const char *msg;
120
Namhyung Kim0ecf4f02012-07-26 10:50:10 +0900121 BUG_ON(buflen == 0);
Kirill A. Shutemov4cc49d42012-07-24 00:06:54 +0300122
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900123 if (errnum >= 0) {
Kirill A. Shutemov4cc49d42012-07-24 00:06:54 +0300124 const char *err = strerror_r(errnum, buf, buflen);
125
126 if (err != buf) {
127 size_t len = strlen(err);
Irina Tirdea60ff92f2012-08-29 01:22:16 +0300128 memcpy(buf, err, min(buflen - 1, len));
129 *(buf + min(buflen - 1, len)) = '\0';
Kirill A. Shutemov4cc49d42012-07-24 00:06:54 +0300130 }
131
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900132 return 0;
133 }
134
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300135 if (errnum < __TARGET_ERRNO__START || errnum >= __TARGET_ERRNO__END)
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900136 return -1;
137
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300138 idx = errnum - __TARGET_ERRNO__START;
139 msg = target__error_str[idx];
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900140
141 switch (errnum) {
Adrian Hunter3aa59392013-11-15 15:52:29 +0200142 case TARGET_ERRNO__PID_OVERRIDE_CPU ...
143 TARGET_ERRNO__SYSTEM_OVERRIDE_THREAD:
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900144 snprintf(buf, buflen, "%s", msg);
145 break;
146
Arnaldo Carvalho de Melo602ad872013-11-12 16:46:16 -0300147 case TARGET_ERRNO__INVALID_UID:
148 case TARGET_ERRNO__USER_NOT_FOUND:
Namhyung Kim16ad2ff2012-05-07 14:09:02 +0900149 snprintf(buf, buflen, msg, target->uid_str);
150 break;
151
152 default:
153 /* cannot reach here */
154 break;
155 }
156
157 return 0;
158}