[5869cea] | 1 | /*
|
---|
| 2 | This is an example that uses io_uring with attach mode.
|
---|
| 3 | It demonstrates the what happens if some thread submits to
|
---|
| 4 | the original ring and then spins, while a second thread waits
|
---|
| 5 | on the attached ring. This deadlocks because even if rings are
|
---|
| 6 | attached, they don't share a completion queue. Therefore all
|
---|
| 7 | completion queues must be polled for progress to be made.
|
---|
| 8 |
|
---|
| 9 | It uses liburing for simplicity.
|
---|
| 10 | */
|
---|
| 11 |
|
---|
| 12 | #include <stdio.h>
|
---|
| 13 | #include <stdlib.h>
|
---|
| 14 | #include <string.h>
|
---|
| 15 |
|
---|
| 16 | #include <errno.h>
|
---|
| 17 | #include <fcntl.h>
|
---|
| 18 | #include <unistd.h>
|
---|
| 19 |
|
---|
| 20 | #include <pthread.h>
|
---|
| 21 | #include <liburing.h>
|
---|
| 22 |
|
---|
| 23 | volatile bool done = false;
|
---|
| 24 |
|
---|
| 25 | void * submitter(void * arg) {
|
---|
| 26 | struct io_uring * ring = (struct io_uring *)arg;
|
---|
| 27 |
|
---|
| 28 | const char * path = __FILE__;
|
---|
| 29 |
|
---|
| 30 | int fd = open(path, 0, O_RDONLY);
|
---|
| 31 | if( fd < 0) {
|
---|
| 32 | fprintf(stderr, "Can't open file: (%d) %s\n", errno, strerror(errno));
|
---|
| 33 | exit(EXIT_FAILURE);
|
---|
| 34 | }
|
---|
| 35 | printf("Opened file: %d\n", fd);
|
---|
| 36 |
|
---|
| 37 | char buffer[1024];
|
---|
| 38 | struct io_uring_sqe * sqe = io_uring_get_sqe(ring);
|
---|
| 39 | io_uring_prep_read(sqe, fd, buffer, 1024, 0);
|
---|
| 40 | io_uring_submit(ring);
|
---|
| 41 |
|
---|
| 42 | printf("Submitted read\n");
|
---|
| 43 |
|
---|
| 44 | printf("Spinning\n");
|
---|
| 45 | while(!__atomic_load_n(&done, __ATOMIC_SEQ_CST));
|
---|
| 46 |
|
---|
| 47 | return NULL;
|
---|
| 48 | }
|
---|
| 49 |
|
---|
| 50 | void * poller(void * arg) {
|
---|
| 51 | struct io_uring * ring = (struct io_uring *)arg;
|
---|
| 52 | struct io_uring_cqe *cqe;
|
---|
| 53 |
|
---|
| 54 | printf("Waiting for results\n");
|
---|
| 55 |
|
---|
| 56 | int ret = io_uring_wait_cqe(ring, &cqe);
|
---|
| 57 | if( ret < 0) {
|
---|
| 58 | fprintf(stderr, "ioring wait failed: (%d) %s\n", -ret, strerror(-ret));
|
---|
| 59 | exit(EXIT_FAILURE);
|
---|
| 60 | }
|
---|
| 61 | io_uring_cqe_seen(ring, cqe);
|
---|
| 62 |
|
---|
| 63 | printf("Got result\n");
|
---|
| 64 |
|
---|
| 65 | __atomic_store_n(&done, true, __ATOMIC_SEQ_CST);
|
---|
| 66 | return NULL;
|
---|
| 67 | }
|
---|
| 68 |
|
---|
| 69 | int main() {
|
---|
| 70 | struct io_uring base_ring;
|
---|
| 71 | io_uring_queue_init(8, &base_ring, 0);
|
---|
| 72 | printf("Created first ring: %d\n", base_ring.ring_fd);
|
---|
| 73 |
|
---|
| 74 | struct io_uring attached_ring;
|
---|
| 75 | struct io_uring_params par;
|
---|
| 76 | memset(&par, '\0', sizeof(struct io_uring_params));
|
---|
| 77 | par.flags |= IORING_SETUP_ATTACH_WQ;
|
---|
| 78 | par.wq_fd = base_ring.ring_fd;
|
---|
| 79 | io_uring_queue_init_params(8, &attached_ring, &par);
|
---|
| 80 | printf("Attached second ring: %d\n", attached_ring.ring_fd);
|
---|
| 81 |
|
---|
| 82 | printf("Forking\n");
|
---|
| 83 | pthread_t s, p;
|
---|
| 84 |
|
---|
| 85 | int ret = pthread_create(&s, NULL, submitter, &base_ring);
|
---|
| 86 | if( ret < 0) {
|
---|
| 87 | fprintf(stderr, "pthread create 1 error: (%d) %s\n", errno, strerror(errno));
|
---|
| 88 | exit(EXIT_FAILURE);
|
---|
| 89 | }
|
---|
| 90 |
|
---|
| 91 | ret = pthread_create(&p, NULL, poller, &attached_ring);
|
---|
| 92 | if( ret < 0) {
|
---|
| 93 | fprintf(stderr, "pthread create 2 error: (%d) %s\n", errno, strerror(errno));
|
---|
| 94 | exit(EXIT_FAILURE);
|
---|
| 95 | }
|
---|
| 96 |
|
---|
| 97 | ret = pthread_join(p, NULL);
|
---|
| 98 | if( ret < 0) {
|
---|
| 99 | fprintf(stderr, "pthread join 2 error: (%d) %s\n", errno, strerror(errno));
|
---|
| 100 | exit(EXIT_FAILURE);
|
---|
| 101 | }
|
---|
| 102 |
|
---|
| 103 | ret = pthread_join(s, NULL);
|
---|
| 104 | if( ret < 0) {
|
---|
| 105 | fprintf(stderr, "pthread join 1 error: (%d) %s\n", errno, strerror(errno));
|
---|
| 106 | exit(EXIT_FAILURE);
|
---|
| 107 | }
|
---|
| 108 | }
|
---|