shim_semget.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921
  1. /* Copyright (C) 2014 Stony Brook University
  2. This file is part of Graphene Library OS.
  3. Graphene Library OS is free software: you can redistribute it and/or
  4. modify it under the terms of the GNU Lesser General Public License
  5. as published by the Free Software Foundation, either version 3 of the
  6. License, or (at your option) any later version.
  7. Graphene Library OS is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU Lesser General Public License for more details.
  11. You should have received a copy of the GNU Lesser General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>. */
  13. /*
  14. * shim_semget.c
  15. *
  16. * Implementation of system call "semget", "semop", "semtimedop" and "semctl".
  17. */
  18. #include <errno.h>
  19. #include <list.h>
  20. #include <pal.h>
  21. #include <pal_error.h>
  22. #include <shim_handle.h>
  23. #include <shim_internal.h>
  24. #include <shim_ipc.h>
  25. #include <shim_profile.h>
  26. #include <shim_sysv.h>
  27. #include <shim_table.h>
  28. #include <shim_utils.h>
  29. #define SEM_HASH_LEN 8
  30. #define SEM_HASH_NUM (1 << SEM_HASH_LEN)
  31. #define SEM_HASH_MASK (SEM_HASH_NUM - 1)
  32. #define SEM_HASH(idx) ((idx) & SEM_HASH_MASK)
  33. /* The sem_list links shim_sem_handle objects by the list field.
  34. * The sem_key_hlist links them by key_hlist, and qid_hlist by qid_hlist */
  35. DEFINE_LISTP(shim_sem_handle);
  36. static LISTP_TYPE(shim_sem_handle) sem_list;
  37. static LISTP_TYPE(shim_sem_handle) sem_key_hlist[SEM_HASH_NUM];
  38. static LISTP_TYPE(shim_sem_handle) sem_sid_hlist[SEM_HASH_NUM];
  39. static struct shim_lock sem_list_lock;
  40. DEFINE_PROFILE_CATEGORY(sysv_sem, );
  41. #define SEM_TO_HANDLE(semhdl) container_of((semhdl), struct shim_handle, info.sem)
  42. static int __add_sem_handle(unsigned long key, IDTYPE semid, int nsems, bool owned,
  43. struct shim_sem_handle** semhdl) {
  44. LISTP_TYPE(shim_sem_handle)* key_head =
  45. (key != IPC_PRIVATE) ? &sem_key_hlist[SEM_HASH(key)] : NULL;
  46. LISTP_TYPE(shim_sem_handle)* sid_head = semid ? &sem_sid_hlist[SEM_HASH(semid)] : NULL;
  47. struct shim_sem_handle* tmp;
  48. int ret = 0;
  49. if (key_head)
  50. LISTP_FOR_EACH_ENTRY(tmp, key_head, key_hlist) {
  51. if (tmp->semkey == key) {
  52. if (tmp->semid == semid)
  53. goto out;
  54. return -EEXIST;
  55. }
  56. }
  57. if (sid_head)
  58. LISTP_FOR_EACH_ENTRY(tmp, sid_head, sid_hlist) {
  59. if (tmp->semid == semid) {
  60. if (key)
  61. tmp->semkey = key;
  62. goto out;
  63. }
  64. }
  65. struct shim_handle* hdl = get_new_handle();
  66. if (!hdl)
  67. return -ENOMEM;
  68. tmp = &hdl->info.sem;
  69. hdl->type = TYPE_SEM;
  70. tmp->semkey = key;
  71. tmp->semid = semid;
  72. tmp->owned = owned;
  73. tmp->event = DkNotificationEventCreate(PAL_FALSE);
  74. if (owned && nsems) {
  75. tmp->nsems = nsems;
  76. tmp->sems = malloc(sizeof(struct sem_obj) * nsems);
  77. if (!tmp->sems) {
  78. ret = -ENOMEM;
  79. goto failed;
  80. }
  81. for (int i = 0; i < nsems; i++) {
  82. tmp->sems[i].num = i;
  83. tmp->sems[i].val = 0;
  84. tmp->sems[i].host_sem_id = 0;
  85. tmp->sems[i].host_sem = NULL;
  86. INIT_LISTP(&tmp->sems[i].ops);
  87. INIT_LISTP(&tmp->sems[i].next_ops);
  88. }
  89. }
  90. INIT_LISTP(&tmp->migrated);
  91. INIT_LIST_HEAD(tmp, list);
  92. get_handle(hdl);
  93. LISTP_ADD_TAIL(tmp, &sem_list, list);
  94. INIT_LIST_HEAD(tmp, key_hlist);
  95. if (key_head) {
  96. get_handle(hdl);
  97. LISTP_ADD(tmp, key_head, key_hlist);
  98. }
  99. if (sid_head) {
  100. get_handle(hdl);
  101. LISTP_ADD(tmp, sid_head, sid_hlist);
  102. }
  103. out:
  104. if (!semhdl) {
  105. put_handle(hdl);
  106. return 0;
  107. }
  108. *semhdl = tmp;
  109. return 0;
  110. failed:
  111. put_handle(hdl);
  112. return ret;
  113. }
  114. int add_sem_handle(unsigned long key, IDTYPE id, int nsems, bool owned) {
  115. lock(&sem_list_lock);
  116. int ret = __add_sem_handle(key, id, nsems, owned, NULL);
  117. unlock(&sem_list_lock);
  118. return ret;
  119. }
  120. struct shim_sem_handle* get_sem_handle_by_key(unsigned long key) {
  121. LISTP_TYPE(shim_sem_handle)* key_head = &sem_key_hlist[SEM_HASH(key)];
  122. struct shim_sem_handle* tmp;
  123. struct shim_sem_handle* found = NULL;
  124. lock(&sem_list_lock);
  125. LISTP_FOR_EACH_ENTRY(tmp, key_head, key_hlist) {
  126. if (tmp->semkey == key) {
  127. found = tmp;
  128. break;
  129. }
  130. }
  131. if (found)
  132. get_handle(SEM_TO_HANDLE(found));
  133. unlock(&sem_list_lock);
  134. return found;
  135. }
  136. struct shim_sem_handle* get_sem_handle_by_id(IDTYPE semid) {
  137. LISTP_TYPE(shim_sem_handle)* sid_head = &sem_sid_hlist[SEM_HASH(semid)];
  138. struct shim_sem_handle* tmp;
  139. struct shim_sem_handle* found = NULL;
  140. lock(&sem_list_lock);
  141. LISTP_FOR_EACH_ENTRY(tmp, sid_head, sid_hlist) {
  142. if (tmp->semid == semid) {
  143. found = tmp;
  144. break;
  145. }
  146. }
  147. if (found)
  148. get_handle(SEM_TO_HANDLE(found));
  149. unlock(&sem_list_lock);
  150. return found;
  151. }
  152. void put_sem_handle(struct shim_sem_handle* sem) {
  153. put_handle(SEM_TO_HANDLE(sem));
  154. }
  155. static int __del_sem_handle(struct shim_sem_handle* sem) {
  156. if (sem->deleted)
  157. return 0;
  158. sem->deleted = true;
  159. struct shim_handle* hdl = SEM_TO_HANDLE(sem);
  160. lock(&sem_list_lock);
  161. LISTP_DEL_INIT(sem, &sem_list, list);
  162. put_handle(hdl);
  163. if (!LIST_EMPTY(sem, key_hlist)) {
  164. // DEP: Yuck
  165. LISTP_TYPE(shim_sem_handle)* key_head = &sem_key_hlist[SEM_HASH(sem->semkey)];
  166. LISTP_DEL_INIT(sem, key_head, key_hlist);
  167. put_handle(hdl);
  168. }
  169. if (!LIST_EMPTY(sem, sid_hlist)) {
  170. // DEP: Yuck
  171. LISTP_TYPE(shim_sem_handle)* sid_head = &sem_sid_hlist[SEM_HASH(sem->semid)];
  172. LISTP_DEL_INIT(sem, sid_head, sid_hlist);
  173. put_handle(hdl);
  174. }
  175. unlock(&sem_list_lock);
  176. return 0;
  177. }
  178. int del_sem_handle(struct shim_sem_handle* sem) {
  179. struct shim_handle* hdl = SEM_TO_HANDLE(sem);
  180. lock(&hdl->lock);
  181. int ret = del_sem_handle(sem);
  182. unlock(&hdl->lock);
  183. return ret;
  184. }
  185. static void __try_create_lock(void) {
  186. create_lock_runtime(&sem_list_lock);
  187. }
  188. int shim_do_semget(key_t key, int nsems, int semflg) {
  189. INC_PROFILE_OCCURENCE(syscall_use_ipc);
  190. IDTYPE semid = 0;
  191. int ret;
  192. __try_create_lock();
  193. if (key != IPC_PRIVATE) {
  194. struct shim_sem_handle* sem = get_sem_handle_by_key(key);
  195. if (sem) {
  196. semid = sem->semid;
  197. put_sem_handle(sem);
  198. return (semflg & IPC_EXCL) ? -EEXIST : (int)semid;
  199. }
  200. }
  201. struct sysv_key k;
  202. k.key = key;
  203. k.type = SYSV_SEM;
  204. if (semflg & IPC_CREAT) {
  205. do {
  206. semid = allocate_sysv(0, 0);
  207. if (!semid)
  208. semid = ipc_sysv_lease_send(NULL);
  209. } while (!semid);
  210. if (key != IPC_PRIVATE) {
  211. if ((ret = ipc_sysv_tellkey_send(NULL, 0, &k, semid, 0)) < 0) {
  212. release_sysv(semid);
  213. return ret;
  214. }
  215. }
  216. add_sem_handle(key, semid, nsems, true);
  217. } else {
  218. if ((ret = ipc_sysv_findkey_send(&k)) < 0)
  219. return ret;
  220. semid = ret;
  221. if ((ret = ipc_sysv_query_send(semid)) < 0)
  222. return ret;
  223. }
  224. return semid;
  225. }
  226. static int connect_sem_handle(int semid, int nsems, struct shim_sem_handle** semp) {
  227. struct shim_sem_handle* sem = get_sem_handle_by_id(semid);
  228. int ret;
  229. if (!sem) {
  230. if ((ret = ipc_sysv_query_send(semid)) < 0)
  231. return ret;
  232. if (!sem) {
  233. lock(&sem_list_lock);
  234. ret = __add_sem_handle(IPC_PRIVATE, semid, nsems, false, &sem);
  235. unlock(&sem_list_lock);
  236. if (ret < 0)
  237. return ret;
  238. }
  239. }
  240. *semp = sem;
  241. return 0;
  242. }
  243. int recover_sem_ownership(struct shim_sem_handle* sem, struct sem_backup* backups, int nbackups,
  244. struct sem_client_backup* clients, int nclients) {
  245. struct shim_handle* hdl = SEM_TO_HANDLE(sem);
  246. lock(&hdl->lock);
  247. assert(!sem->owned);
  248. assert(!sem->nsems && !sem->sems);
  249. sem->nsems = nbackups;
  250. if (!sem->sems && !(sem->sems = malloc(sizeof(struct sem_obj) * nbackups)))
  251. goto out;
  252. for (int i = 0; i < nbackups; i++) {
  253. sem->sems[i].num = i;
  254. sem->sems[i].val = backups[i].val;
  255. sem->sems[i].zcnt = backups[i].zcnt;
  256. sem->sems[i].ncnt = backups[i].ncnt;
  257. sem->sems[i].pid = backups[i].pid;
  258. INIT_LISTP(&sem->sems[i].ops);
  259. INIT_LISTP(&sem->sems[i].next_ops);
  260. }
  261. for (int i = 0; i < nclients; i++) {
  262. struct sem_ops* op = malloc(sizeof(struct sem_ops));
  263. if (!op)
  264. continue;
  265. op->stat.completed = false;
  266. op->stat.failed = false;
  267. op->stat.nops = clients[i].nops;
  268. op->stat.current = clients[i].current;
  269. op->stat.timeout = -1;
  270. op->client.vmid = clients[i].vmid;
  271. op->client.port = NULL;
  272. op->client.seq = clients[i].seq;
  273. INIT_LIST_HEAD(op, progress);
  274. LISTP_ADD_TAIL(op, &sem->migrated, progress);
  275. }
  276. sem->owned = true;
  277. DkEventSet(sem->event);
  278. out:
  279. unlock(&hdl->lock);
  280. return 0;
  281. }
  282. static int __do_semop(int semid, struct sembuf* sops, unsigned int nsops, unsigned long timeout) {
  283. int ret;
  284. struct shim_sem_handle* sem;
  285. size_t nsems = 0;
  286. for (size_t i = 0; i < nsops; i++)
  287. if (sops[i].sem_num >= nsems)
  288. nsems = sops[i].sem_num + 1;
  289. __try_create_lock();
  290. if ((ret = connect_sem_handle(semid, nsems, &sem)) < 0)
  291. return ret;
  292. ret = submit_sysv_sem(sem, sops, nsops, timeout, NULL);
  293. put_sem_handle(sem);
  294. return ret;
  295. }
  296. int shim_do_semop(int semid, struct sembuf* sops, unsigned int nsops) {
  297. INC_PROFILE_OCCURENCE(syscall_use_ipc);
  298. return __do_semop(semid, sops, nsops, IPC_SEM_NOTIMEOUT);
  299. }
  300. int shim_do_semtimedop(int semid, struct sembuf* sops, unsigned int nsops,
  301. const struct timespec* timeout) {
  302. INC_PROFILE_OCCURENCE(syscall_use_ipc);
  303. return __do_semop(semid, sops, nsops, timeout->tv_sec * 1000000000ULL + timeout->tv_nsec);
  304. }
  305. int shim_do_semctl(int semid, int semnum, int cmd, unsigned long arg) {
  306. INC_PROFILE_OCCURENCE(syscall_use_ipc);
  307. struct shim_sem_handle* sem;
  308. int ret;
  309. __try_create_lock();
  310. if ((ret = connect_sem_handle(semid, 0, &sem)) < 0)
  311. return ret;
  312. struct shim_handle* hdl = SEM_TO_HANDLE(sem);
  313. lock(&hdl->lock);
  314. switch (cmd) {
  315. case IPC_RMID: {
  316. if (!sem->owned) {
  317. ret = ipc_sysv_delres_send(NULL, 0, semid, SYSV_SEM);
  318. if (ret < 0)
  319. goto out;
  320. }
  321. __del_sem_handle(sem);
  322. goto out;
  323. }
  324. }
  325. if (sem->owned) {
  326. if (sem->deleted) {
  327. ret = -EIDRM;
  328. goto out;
  329. }
  330. switch (cmd) {
  331. case GETALL:
  332. for (int i = 0; i < sem->nsems; i++) {
  333. unsigned short val = sem->sems[i].val;
  334. ((unsigned short*)arg)[i] = val;
  335. }
  336. break;
  337. case GETNCNT:
  338. ret = sem->sems[semnum].ncnt;
  339. break;
  340. case GETPID:
  341. ret = sem->sems[semnum].pid;
  342. break;
  343. case GETVAL:
  344. ret = sem->sems[semnum].val;
  345. break;
  346. case GETZCNT:
  347. ret = sem->sems[semnum].zcnt;
  348. break;
  349. case SETALL:
  350. for (int i = 0; i < sem->nsems; i++) {
  351. unsigned short val = ((unsigned short*)arg)[i];
  352. sem->sems[i].val = val;
  353. }
  354. break;
  355. case SETVAL: {
  356. unsigned short val = arg;
  357. sem->sems[semnum].val = val;
  358. break;
  359. }
  360. }
  361. } else {
  362. switch (cmd) {
  363. case GETALL:
  364. case SETALL: {
  365. int valsize = sem->nsems * sizeof(unsigned short);
  366. ret = ipc_sysv_semctl_send(sem->semid, 0, cmd, (unsigned short*)arg, valsize);
  367. break;
  368. }
  369. case GETVAL:
  370. case GETNCNT:
  371. case GETPID:
  372. case GETZCNT: {
  373. int valsize = sizeof(unsigned short);
  374. unsigned short val;
  375. ret = ipc_sysv_semctl_send(sem->semid, semnum, cmd, &val, valsize);
  376. if (!ret)
  377. ret = val;
  378. break;
  379. }
  380. case SETVAL: {
  381. unsigned short val = arg;
  382. ret = ipc_sysv_semctl_send(sem->semid, semnum, cmd, &val, sizeof(unsigned short));
  383. break;
  384. }
  385. }
  386. }
  387. out:
  388. unlock(&hdl->lock);
  389. put_sem_handle(sem);
  390. return ret;
  391. }
  392. static bool __handle_sysv_sems(struct shim_sem_handle* sem) {
  393. bool progressed = false;
  394. bool setevent = false;
  395. struct sem_obj* sobj;
  396. for (sobj = sem->sems; sobj < &sem->sems[sem->nsems]; sobj++)
  397. LISTP_SPLICE_TAIL_INIT(&sobj->next_ops, &sobj->ops, progress, sem_ops);
  398. for (sobj = sem->sems; sobj < &sem->sems[sem->nsems]; sobj++) {
  399. struct sem_ops* sops;
  400. struct sem_ops* n;
  401. LISTP_FOR_EACH_ENTRY_SAFE(sops, n, &sobj->ops, progress) {
  402. struct sembuf* op = &sops->ops[sops->stat.current];
  403. assert(op->sem_num == sobj->num);
  404. // first_iter is a variable defined by LISTP_FOR_EACH_ENTRY_SAFE
  405. // The second part of this assertion is only valid after the first attempt
  406. assert(first_iter || (sops != n));
  407. if (sops->stat.completed)
  408. goto send_result;
  409. again:
  410. if (op->sem_op > 0) {
  411. sobj->val += op->sem_op;
  412. debug("sem %u: add %u => %u\n", sobj->num, op->sem_op, sobj->val);
  413. } else if (op->sem_op < 0) {
  414. if (sobj->val < -op->sem_op) {
  415. if (op->sem_flg & IPC_NOWAIT) {
  416. debug("sem %u: wait for %u failed\n", sobj->num, -op->sem_op);
  417. goto failed;
  418. }
  419. continue;
  420. }
  421. sobj->val -= -op->sem_op;
  422. debug("sem %u: wait for %u => %u\n", sobj->num, -op->sem_op, sobj->val);
  423. } else {
  424. if (sobj->val) {
  425. if (op->sem_flg & IPC_NOWAIT) {
  426. debug("sem %u: wait for 0 failed\n", sobj->num);
  427. goto failed;
  428. }
  429. continue;
  430. }
  431. debug("sem %u: wait for 0\n", sobj->num);
  432. }
  433. progressed = true;
  434. sops->stat.current++;
  435. if (sops->stat.current == sops->stat.nops) {
  436. sops->stat.completed = true;
  437. goto send_result;
  438. }
  439. op = &sops->ops[sops->stat.current];
  440. if (op->sem_num != sobj->num) {
  441. LISTP_MOVE_TAIL(sops, &sem->sems[op->sem_num].next_ops, &sobj->ops, progress);
  442. continue;
  443. }
  444. goto again;
  445. failed:
  446. progressed = true;
  447. sops->stat.failed = true;
  448. send_result:
  449. /* Chia-Che 10/17/17: If the code reaches this point, sops should
  450. * still be in sobj->ops. */
  451. LISTP_DEL_INIT(sops, &sobj->ops, progress);
  452. sem->nreqs--;
  453. if (!sops->client.vmid) {
  454. setevent = true;
  455. continue;
  456. }
  457. size_t total_msg_size = get_ipc_msg_size(sizeof(struct shim_ipc_resp));
  458. struct shim_ipc_msg* resp_msg = __alloca(total_msg_size);
  459. init_ipc_msg(resp_msg, IPC_RESP, total_msg_size, sops->client.vmid);
  460. resp_msg->seq = sops->client.seq;
  461. struct shim_ipc_resp* resp = (struct shim_ipc_resp*)resp_msg->msg;
  462. resp->retval = sops->stat.completed ? 0 : -EAGAIN;
  463. send_ipc_message(resp_msg, sops->client.port);
  464. put_ipc_port(sops->client.port);
  465. sops->client.vmid = 0;
  466. sops->client.port = NULL;
  467. sops->client.seq = 0;
  468. free(sops);
  469. }
  470. }
  471. if (setevent)
  472. DkEventSet(sem->event);
  473. return progressed;
  474. }
  475. static void __handle_one_sysv_sem(struct shim_sem_handle* sem, struct sem_stat* stat,
  476. struct sembuf* sops) {
  477. bool progressed = false;
  478. again:
  479. while (stat->current < stat->nops) {
  480. struct sem_obj* sobj = &sem->sems[sops[stat->current].sem_num];
  481. struct sembuf* op = &sops[stat->current];
  482. if (op->sem_op > 0) {
  483. progressed = true;
  484. sobj->val += op->sem_op;
  485. debug("sem %u: add %u => %u\n", sobj->num, op->sem_op, sobj->val);
  486. } else if (op->sem_op < 0) {
  487. if (sobj->val < -op->sem_op) {
  488. if (op->sem_flg & IPC_NOWAIT) {
  489. stat->failed = true;
  490. debug("sem %u: wait for %u failed\n", sobj->num, -op->sem_op);
  491. return;
  492. }
  493. goto failed;
  494. }
  495. progressed = true;
  496. sobj->val -= -op->sem_op;
  497. debug("sem %u: wait for %u => %u\n", sobj->num, -op->sem_op, sobj->val);
  498. } else {
  499. if (sobj->val) {
  500. if (op->sem_flg & IPC_NOWAIT) {
  501. stat->failed = true;
  502. debug("sem %u: wait for 0 failed\n", sobj->num);
  503. return;
  504. }
  505. goto failed;
  506. }
  507. progressed = true;
  508. debug("sem %u: wait for 0\n", sobj->num);
  509. }
  510. stat->current++;
  511. }
  512. stat->completed = true;
  513. failed:
  514. if (progressed) {
  515. while (__handle_sysv_sems(sem))
  516. ;
  517. progressed = false;
  518. if (!stat->completed)
  519. goto again;
  520. }
  521. }
  522. #if MIGRATE_SYSV_SEM == 1
  523. static int sem_balance_migrate(struct shim_handle* hdl, struct sysv_client* client);
  524. static struct sysv_balance_policy sem_policy = {
  525. .score_decay = SEM_SCORE_DECAY,
  526. .score_max = SEM_SCORE_MAX,
  527. .balance_threshold = SEM_BALANCE_THRESHOLD,
  528. .migrate = &sem_balance_migrate,
  529. };
  530. #endif
  531. DEFINE_PROFILE_CATEGORY(submit_sysv_sem, sysv_sem);
  532. DEFINE_PROFILE_INTERVAL(sem_prepare_stat, submit_sysv_sem);
  533. DEFINE_PROFILE_INTERVAL(sem_lock_handle, submit_sysv_sem);
  534. DEFINE_PROFILE_INTERVAL(sem_count_score, submit_sysv_sem);
  535. DEFINE_PROFILE_INTERVAL(sem_handle_by_shared_semaphore, submit_sysv_sem);
  536. DEFINE_PROFILE_INTERVAL(sem_send_ipc_movres, submit_sysv_sem);
  537. DEFINE_PROFILE_INTERVAL(sem_send_ipc_semop, submit_sysv_sem);
  538. DEFINE_PROFILE_INTERVAL(sem_handle_one_sysv_sem, submit_sysv_sem);
  539. DEFINE_PROFILE_INTERVAL(sem_send_ipc_response, submit_sysv_sem);
  540. DEFINE_PROFILE_INTERVAL(sem_alloc_semop, submit_sysv_sem);
  541. DEFINE_PROFILE_INTERVAL(sem_append_semop, submit_sysv_sem);
  542. DEFINE_PROFILE_INTERVAL(sem_wait_for_complete, submit_sysv_sem);
  543. int submit_sysv_sem(struct shim_sem_handle* sem, struct sembuf* sops, int nsops,
  544. unsigned long timeout, struct sysv_client* client) {
  545. BEGIN_PROFILE_INTERVAL();
  546. int ret = 0;
  547. struct shim_handle* hdl = SEM_TO_HANDLE(sem);
  548. struct sem_ops* sem_ops = NULL;
  549. bool malloced = false;
  550. struct sem_stat stat;
  551. stat.nops = nsops;
  552. stat.current = 0;
  553. stat.timeout = timeout;
  554. stat.completed = false;
  555. stat.failed = false;
  556. SAVE_PROFILE_INTERVAL(sem_prepare_stat);
  557. lock(&hdl->lock);
  558. SAVE_PROFILE_INTERVAL(sem_lock_handle);
  559. if (sem->deleted) {
  560. ret = -EIDRM;
  561. goto out_locked;
  562. }
  563. IDTYPE semid = sem->semid;
  564. bool sendreply = false;
  565. unsigned long seq = client ? client->seq : 0;
  566. int score = 0;
  567. for (int i = 0; i < nsops; i++) {
  568. struct sembuf* op = &sops[i];
  569. if (op->sem_op > 0) {
  570. score += SEM_POSITIVE_SCORE(op->sem_num);
  571. } else if (op->sem_op < 0) {
  572. score += SEM_NEGATIVE_SCORE(-op->sem_num);
  573. sendreply = true;
  574. } else {
  575. score += SEM_ZERO_SCORE;
  576. sendreply = true;
  577. }
  578. }
  579. SAVE_PROFILE_INTERVAL(sem_count_score);
  580. if (sem->deleted) {
  581. if (!client || sendreply) {
  582. ret = -EIDRM;
  583. goto out_locked;
  584. }
  585. ret = ipc_sysv_delres_send(client->port, client->vmid, sem->semid, SYSV_SEM);
  586. goto out_locked;
  587. }
  588. #if MIGRATE_SYSV_SEM == 1
  589. if (sem->owned) {
  590. __balance_sysv_score(&sem_policy, hdl, sem->scores, MAX_SYSV_CLIENTS, client, score);
  591. if (!sem->owned && client) {
  592. struct shim_ipc_info* owner = sem->owner;
  593. assert(owner);
  594. ret = ipc_sysv_movres_send(client, owner->vmid, qstrgetstr(&owner->uri), sem->lease,
  595. sem->semid, SYSV_SEM);
  596. goto out_locked;
  597. }
  598. }
  599. #endif
  600. if (!sem->owned) {
  601. if (client) {
  602. struct shim_ipc_info* owner = sem->owner;
  603. ret = owner ? ipc_sysv_movres_send(client, owner->vmid, qstrgetstr(&owner->uri),
  604. sem->lease, sem->semid, SYSV_SEM)
  605. : -ECONNREFUSED;
  606. SAVE_PROFILE_INTERVAL(sem_send_ipc_movres);
  607. goto out_locked;
  608. }
  609. unowned:
  610. unlock(&hdl->lock);
  611. ret = ipc_sysv_semop_send(semid, sops, nsops, timeout, &seq);
  612. if (ret != -EAGAIN && ret != -ECONNREFUSED)
  613. goto out;
  614. lock(&hdl->lock);
  615. SAVE_PROFILE_INTERVAL(sem_send_ipc_semop);
  616. if (!sem->owned)
  617. goto out_locked;
  618. }
  619. if (seq) {
  620. struct sem_ops* op;
  621. LISTP_FOR_EACH_ENTRY(op, &sem->migrated, progress) {
  622. if (op->client.vmid == (client ? client->vmid : cur_process.vmid) &&
  623. seq == op->client.seq) {
  624. LISTP_DEL_INIT(op, &sem->migrated, progress);
  625. sem_ops = op;
  626. stat = sem_ops->stat;
  627. malloced = true;
  628. break;
  629. }
  630. }
  631. }
  632. __handle_one_sysv_sem(sem, &stat, sops);
  633. SAVE_PROFILE_INTERVAL(sem_handle_one_sysv_sem);
  634. if (stat.completed || stat.failed) {
  635. ret = stat.completed ? 0 : -EAGAIN;
  636. if (client && sendreply) {
  637. size_t total_msg_size = get_ipc_msg_size(sizeof(struct shim_ipc_resp));
  638. struct shim_ipc_msg* resp_msg = __alloca(total_msg_size);
  639. init_ipc_msg(resp_msg, IPC_RESP, total_msg_size, client->vmid);
  640. resp_msg->seq = client->seq;
  641. struct shim_ipc_resp* resp = (struct shim_ipc_resp*)resp_msg->msg;
  642. resp->retval = ret;
  643. ret = send_ipc_message(resp_msg, client->port);
  644. }
  645. SAVE_PROFILE_INTERVAL(sem_send_ipc_response);
  646. goto out_locked;
  647. }
  648. if (client) {
  649. assert(sendreply);
  650. if (!sem_ops || !malloced) {
  651. sem_ops = malloc(sizeof(struct sem_ops) + sizeof(struct sembuf) * nsops);
  652. if (!sem_ops) {
  653. ret = -ENOMEM;
  654. goto out_locked;
  655. }
  656. sem_ops->client.vmid = 0;
  657. sem_ops->client.port = NULL;
  658. sem_ops->client.seq = 0;
  659. INIT_LIST_HEAD(sem_ops, progress);
  660. malloced = true;
  661. SAVE_PROFILE_INTERVAL(sem_alloc_semop);
  662. }
  663. } else {
  664. if (!sem_ops) {
  665. sem_ops = __alloca(sizeof(struct sem_ops) + sizeof(struct sembuf) * nsops);
  666. sem_ops->client.vmid = 0;
  667. sem_ops->client.port = NULL;
  668. sem_ops->client.seq = 0;
  669. INIT_LIST_HEAD(sem_ops, progress);
  670. SAVE_PROFILE_INTERVAL(sem_alloc_semop);
  671. }
  672. }
  673. sem_ops->stat = stat;
  674. for (int i = 0; i < nsops; i++) {
  675. sem_ops->ops[i] = sops[i];
  676. }
  677. LISTP_TYPE(sem_ops)* next_ops = &sem->sems[sops[stat.current].sem_num].next_ops;
  678. assert(LIST_EMPTY(sem_ops, progress));
  679. LISTP_ADD_TAIL(sem_ops, next_ops, progress);
  680. // CHECK_LIST_HEAD(next_ops);
  681. sem->nreqs++;
  682. SAVE_PROFILE_INTERVAL(sem_append_semop);
  683. if (client) {
  684. assert(sendreply);
  685. add_ipc_port(client->port, client->vmid, IPC_PORT_SYSVCON, NULL);
  686. get_ipc_port(client->port);
  687. sem_ops->client = *client;
  688. sem_ops = NULL;
  689. goto out_locked;
  690. }
  691. while (!sem_ops->stat.completed && !sem_ops->stat.failed) {
  692. if (!sem->owned) {
  693. /* Chia-Che 10/17/17: sem_ops may move from semaphore to semaphore
  694. base on its current state */
  695. next_ops = &sem->sems[sem_ops->ops[sem_ops->stat.current].sem_num].next_ops;
  696. LISTP_DEL_INIT(sem_ops, next_ops, progress);
  697. goto unowned;
  698. }
  699. unlock(&hdl->lock);
  700. object_wait_with_retry(sem->event);
  701. lock(&hdl->lock);
  702. SAVE_PROFILE_INTERVAL(sem_wait_for_complete);
  703. }
  704. ret = sem_ops->stat.completed ? 0 : -EAGAIN;
  705. out_locked:
  706. unlock(&hdl->lock);
  707. out:
  708. if (sem_ops && malloced)
  709. free(sem_ops);
  710. return ret;
  711. }
  712. #if MIGRATE_SYSV_SEM == 1
  713. static int sem_balance_migrate(struct shim_handle* hdl, struct sysv_client* src) {
  714. struct shim_sem_handle* sem = &hdl->info.sem;
  715. int ret = 0;
  716. debug("trigger semaphore balancing, migrate to process %u\n", src->vmid);
  717. struct sem_backup* sem_backups = __alloca(sizeof(struct sem_backup) * sem->nsems);
  718. struct sem_client_backup* clients = __alloca(sizeof(struct sem_client_backup) * sem->nreqs);
  719. int sem_cnt = 0, client_cnt = 0;
  720. struct sem_obj* sobj;
  721. for (sobj = sem->sems; sobj < &sem->sems[sem->nsems]; sobj++) {
  722. assert(sem_cnt < sem->nsems);
  723. struct sem_backup* b = sem_backups + (sem_cnt++);
  724. b->val = sobj->val;
  725. b->zcnt = sobj->zcnt;
  726. b->ncnt = sobj->ncnt;
  727. b->pid = sobj->pid;
  728. LISTP_SPLICE_TAIL(&sobj->next_ops, &sobj->ops, progress, sem_ops);
  729. struct sem_ops* sops;
  730. LISTP_FOR_EACH_ENTRY(sops, &sobj->ops, progress) {
  731. assert(client_cnt < sem->nreqs);
  732. struct sem_client_backup* c = clients + (client_cnt)++;
  733. c->vmid = sops->client.vmid;
  734. c->seq = sops->client.seq;
  735. c->current = sops->stat.current;
  736. c->nops = sops->stat.nops;
  737. }
  738. }
  739. struct shim_ipc_info* info = lookup_ipc_info(src->vmid);
  740. if (!info)
  741. goto out;
  742. ipc_sysv_sublease_send(src->vmid, sem->semid, qstrgetstr(&info->uri), &sem->lease);
  743. ret = ipc_sysv_semmov_send(src->port, src->vmid, sem->semid, sem->lease, sem_backups, sem_cnt,
  744. clients, client_cnt, sem->scores, MAX_SYSV_CLIENTS);
  745. if (ret < 0)
  746. goto failed_info;
  747. sem->owned = false;
  748. sem->owner = info;
  749. for (sobj = sem->sems; sobj < &sem->sems[sem->nsems]; sobj++) {
  750. struct sem_ops* sops;
  751. struct sem_ops* n;
  752. LISTP_FOR_EACH_ENTRY_SAFE(sops, n, &sobj->ops, progress) {
  753. LISTP_DEL_INIT(sops, &sobj->ops, progress);
  754. sem->nreqs--;
  755. sops->stat.failed = true;
  756. if (!sops->client.vmid)
  757. continue;
  758. ipc_sysv_movres_send(&sops->client, src->vmid, qstrgetstr(&info->uri), sem->lease,
  759. sem->semid, SYSV_SEM);
  760. put_ipc_port(sops->client.port);
  761. free(sops);
  762. }
  763. }
  764. sem->nsems = 0;
  765. free(sem->sems);
  766. sem->sems = NULL;
  767. ret = 0;
  768. DkEventSet(sem->event);
  769. goto out;
  770. failed_info:
  771. put_ipc_info(info);
  772. out:
  773. return ret;
  774. }
  775. #endif