shim_semget.c 28 KB

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