shim_semget.c 28 KB

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