Changes in / [44aad8f:c680a4b]


Ignore:
Files:
1 deleted
5 edited

Legend:

Unmodified
Added
Removed
  • benchmark/io/readv.cfa

    r44aad8f rc680a4b  
    1818
    1919extern bool traceHeapOn();
    20 extern ssize_t cfa_preadv2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
     20extern ssize_t async_preadv2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
    2121
    2222int fd;
     
    3434
    3535        while(__atomic_load_n(&run, __ATOMIC_RELAXED)) {
    36                 cfa_preadv2(fd, &iov, 1, 0, 0);
     36                async_preadv2(fd, &iov, 1, 0, 0);
    3737                __atomic_fetch_add( &count, 1, __ATOMIC_SEQ_CST );
    3838        }
     
    4040
    4141int main(int argc, char * argv[]) {
    42         #if !defined(__CFA_NO_STATISTICS__)
    43                 print_stats_at_exit( *active_cluster() );
    44         #endif
    45 
    4642        double duration   = 5.0;
    4743        unsigned long int nthreads = 2;
     
    5046        printf("Setting local\n");
    5147        setlocale(LC_NUMERIC, "");
     48
    5249
    5350        arg_loop:
  • examples/io/simple/server.cfa

    r44aad8f rc680a4b  
    5151
    5252//----------
    53 extern ssize_t cfa_recvmsg(int sockfd, struct msghdr *msg, int flags);
    54 extern int cfa_accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags);
    55 extern int cfa_close(int fd);
     53extern ssize_t async_recvmsg(int sockfd, struct msghdr *msg, int flags);
     54extern int async_accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags);
     55extern int async_close(int fd);
    5656
    5757//----------
     
    8888        struct sockaddr_in cli_addr;
    8989        __socklen_t clilen = sizeof(cli_addr);
    90         int newsock = cfa_accept4(sock, (struct sockaddr *) &cli_addr, &clilen, 0);
     90        int newsock = async_accept4(sock, (struct sockaddr *) &cli_addr, &clilen, 0);
    9191        if (newsock < 0) {
    9292                error( printer, "accept", -newsock);
     
    9797
    9898        while(1) {
    99                 int res = cfa_recvmsg(newsock, &msg, 0);
     99                int res = async_recvmsg(newsock, &msg, 0);
    100100                if(res == 0) break;
    101101                if(res < 0) {
     
    107107        }
    108108
    109         ret = cfa_close(newsock);
     109        ret = async_close(newsock);
    110110      if(ret < 0) {
    111111            error( printer, "close new", -ret);
     
    113113      }
    114114
    115         ret = cfa_close(sock);
     115        ret = async_close(sock);
    116116      if(ret < 0) {
    117117            error( printer, "close old", -ret);
  • libcfa/src/concurrency/io.cfa

    r44aad8f rc680a4b  
    1 //
    2 // Cforall Version 1.0.0 Copyright (C) 2020 University of Waterloo
    3 //
    4 // The contents of this file are covered under the licence agreement in the
    5 // file "LICENCE" distributed with Cforall.
    6 //
    7 // io.cfa --
    8 //
    9 // Author           : Thierry Delisle
    10 // Created On       : Thu Apr 23 17:31:00 2020
    11 // Last Modified By :
    12 // Last Modified On :
    13 // Update Count     :
    14 //
    15 
    161#include "kernel.hfa"
    172
     
    238        void __kernel_io_shutdown( cluster & this ) {
    249                // Nothing to do without io_uring
     10        }
     11
     12        bool is_async( void (*)() ) {
     13                return false;
    2514        }
    2615
     
    10190
    10291                // Requires features
    103                 #if defined(IORING_FEAT_SINGLE_MMAP)
    104                         // adjust the size according to the parameters
    105                         if ((params.features & IORING_FEAT_SINGLE_MMAP) != 0) {
    106                                 cq->ring_sz = sq->ring_sz = max(cq->ring_sz, sq->ring_sz);
    107                         }
    108                 #endif
     92                // // adjust the size according to the parameters
     93                // if ((params.features & IORING_FEAT_SINGLE_MMAP) != 0) {
     94                //      cq->ring_sz = sq->ring_sz = max(cq->ring_sz, sq->ring_sz);
     95                // }
    10996
    11097                // mmap the Submit Queue into existence
     
    114101                }
    115102
     103                // mmap the Completion Queue into existence (may or may not be needed)
    116104                // Requires features
    117                 #if defined(IORING_FEAT_SINGLE_MMAP)
    118                         // mmap the Completion Queue into existence (may or may not be needed)
    119                         if ((params.features & IORING_FEAT_SINGLE_MMAP) != 0) {
    120                                 cq->ring_ptr = sq->ring_ptr;
    121                         }
    122                         else
    123                 #endif
    124                 {
     105                // if ((params.features & IORING_FEAT_SINGLE_MMAP) != 0) {
     106                //      cq->ring_ptr = sq->ring_ptr;
     107                // }
     108                // else {
    125109                        // We need multiple call to MMAP
    126110                        cq.ring_ptr = mmap(0, cq.ring_sz, PROT_READ | PROT_WRITE, MAP_SHARED | MAP_POPULATE, fd, IORING_OFF_CQ_RING);
     
    129113                                abort("KERNEL ERROR: IO_URING MMAP2 - %s\n", strerror(errno));
    130114                        }
    131                 }
     115                // }
    132116
    133117                // mmap the submit queue entries
     
    176160                (this.io.submit){ min(*sq.num, *cq.num) };
    177161
    178                 // Initialize statistics
    179                 #if !defined(__CFA_NO_STATISTICS__)
    180                         this.io.submit_q.stats.submit_avg.val = 0;
    181                         this.io.submit_q.stats.submit_avg.cnt = 0;
    182                         this.io.completion_q.stats.completed_avg.val = 0;
    183                         this.io.completion_q.stats.completed_avg.cnt = 0;
    184                 #endif
    185 
    186162                // Create the poller thread
    187163                this.io.stack = __create_pthread( &this.io.poller, __io_poller, &this );
     
    198174                pthread_join( this.io.poller, 0p );
    199175                free( this.io.stack );
    200 
    201                 // print statistics
    202                 #if !defined(__CFA_NO_STATISTICS__)
    203                         if(this.print_stats) {
    204                                 __cfaabi_bits_print_safe( STDERR_FILENO,
    205                                         "----- I/O uRing Stats -----\n"
    206                                         "- total submit calls  : %llu\n"
    207                                         "- avg submit          : %lf\n"
    208                                         "- total wait calls    : %llu\n"
    209                                         "- avg completion/wait : %lf\n",
    210                                         this.io.submit_q.stats.submit_avg.cnt,
    211                                         ((double)this.io.submit_q.stats.submit_avg.val) / this.io.submit_q.stats.submit_avg.cnt,
    212                                         this.io.completion_q.stats.completed_avg.cnt,
    213                                         ((double)this.io.completion_q.stats.completed_avg.val) / this.io.completion_q.stats.completed_avg.cnt
    214                                 );
    215                         }
    216                 #endif
    217176
    218177                // Shutdown the io rings
     
    245204        // Process a single completion message from the io_uring
    246205        // This is NOT thread-safe
    247         int __drain_io( struct io_ring & ring, sigset_t & mask, int waitcnt ) {
    248                 int ret = syscall( __NR_io_uring_enter, ring.fd, 0, waitcnt, IORING_ENTER_GETEVENTS, &mask, _NSIG / 8);
    249                 if( ret < 0 ) {
    250                         switch((int)errno) {
    251                         case EAGAIN:
    252                         case EINTR:
    253                                 return -EAGAIN;
    254                         default:
    255                                 abort( "KERNEL ERROR: IO_URING WAIT - %s\n", strerror(errno) );
    256                         }
    257                 }
    258 
    259                 // Drain the queue
     206        static bool __io_process(struct io_ring & ring) {
    260207                unsigned head = *ring.completion_q.head;
    261208                unsigned tail = __atomic_load_n(ring.completion_q.tail, __ATOMIC_ACQUIRE);
    262209
    263                 // Nothing was new return 0
    264                 if (head == tail) {
    265                         #if !defined(__CFA_NO_STATISTICS__)
    266                                 ring.completion_q.stats.completed_avg.cnt += 1;
    267                         #endif
    268                         return 0;
    269                 }
    270 
    271                 uint32_t count = tail - head;
    272                 for(i; count) {
    273                         unsigned idx = (head + i) & (*ring.completion_q.mask);
    274                         struct io_uring_cqe & cqe = ring.completion_q.cqes[idx];
    275 
    276                         /* paranoid */ verify(&cqe);
    277 
    278                         struct io_user_data * data = (struct io_user_data *)cqe.user_data;
    279                         // __cfaabi_bits_print_safe( STDERR_FILENO, "Performed reading io cqe %p, result %d for %p\n", data, cqe.res, data->thrd );
    280 
    281                         data->result = cqe.res;
    282                         __unpark( data->thrd __cfaabi_dbg_ctx2 );
    283                 }
     210                if (head == tail) return false;
     211
     212                unsigned idx = head & (*ring.completion_q.mask);
     213                struct io_uring_cqe & cqe = ring.completion_q.cqes[idx];
     214
     215                /* paranoid */ verify(&cqe);
     216
     217                struct io_user_data * data = (struct io_user_data *)cqe.user_data;
     218                // __cfaabi_bits_print_safe( STDERR_FILENO, "Performed reading io cqe %p, result %d for %p\n", data, cqe.res, data->thrd );
     219
     220                data->result = cqe.res;
     221                __unpark( data->thrd __cfaabi_dbg_ctx2 );
    284222
    285223                // Allow new submissions to happen
    286                 V(ring.submit, count);
     224                V(ring.submit);
    287225
    288226                // Mark to the kernel that the cqe has been seen
    289227                // Ensure that the kernel only sees the new value of the head index after the CQEs have been read.
    290                 __atomic_fetch_add( ring.completion_q.head, count, __ATOMIC_RELAXED );
    291 
    292                 // Update statistics
    293                 #if !defined(__CFA_NO_STATISTICS__)
    294                         ring.completion_q.stats.completed_avg.val += count;
    295                         ring.completion_q.stats.completed_avg.cnt += 1;
    296                 #endif
    297 
    298                 return count;
     228                __atomic_fetch_add( ring.completion_q.head, 1, __ATOMIC_RELAXED );
     229
     230                return true;
    299231        }
    300232
     
    314246                verify( (*ring.completion_q.head) == (*ring.completion_q.tail) );
    315247
    316                 while(!__atomic_load_n(&ring.done, __ATOMIC_SEQ_CST)) {
    317                         __drain_io( ring, mask, 1 );
     248                LOOP: while(!__atomic_load_n(&ring.done, __ATOMIC_SEQ_CST)) {
     249                        int ret = syscall( __NR_io_uring_enter, ring.fd, 0, 1, IORING_ENTER_GETEVENTS, &mask, _NSIG / 8);
     250                        if( ret < 0 ) {
     251                                switch((int)errno) {
     252                                case EAGAIN:
     253                                case EINTR:
     254                                        continue LOOP;
     255                                default:
     256                                        abort( "KERNEL ERROR: IO_URING WAIT - %s\n", strerror(errno) );
     257                                }
     258                        }
     259
     260                        // Drain the queue
     261                        while(__io_process(ring)) {}
    318262                }
    319263
     
    349293//
    350294
    351         static inline [* struct io_uring_sqe, uint32_t] __submit_alloc( struct io_ring & ring ) {
    352                 // Wait for a spot to be available
    353                 P(ring.submit);
    354 
    355                 // Allocate the sqe
    356                 uint32_t idx = __atomic_fetch_add(&ring.submit_q.alloc, 1ul32, __ATOMIC_SEQ_CST);
    357 
    358                 // Validate that we didn't overflow anything
    359                 // Check that nothing overflowed
    360                 /* paranoid */ verify( true );
    361 
    362                 // Check that it goes head -> tail -> alloc and never head -> alloc -> tail
    363                 /* paranoid */ verify( true );
    364 
    365                 // Return the sqe
    366                 return [&ring.submit_q.sqes[ idx & (*ring.submit_q.mask)], idx];
    367         }
    368 
    369         static inline void __submit( struct io_ring & ring, uint32_t idx ) {
    370                 // get mutual exclusion
    371                 lock(ring.submit_q.lock __cfaabi_dbg_ctx2);
    372 
    373                 // Append to the list of ready entries
    374                 uint32_t * tail = ring.submit_q.tail;
    375                 const uint32_t mask = *ring.submit_q.mask;
    376 
    377                 ring.submit_q.array[ (*tail) & mask ] = idx & mask;
    378                 __atomic_fetch_add(tail, 1ul32, __ATOMIC_SEQ_CST);
    379 
    380                 // Submit however, many entries need to be submitted
    381                 int ret = syscall( __NR_io_uring_enter, ring.fd, 1, 0, 0, 0p, 0);
    382                 // __cfaabi_bits_print_safe( STDERR_FILENO, "Performed io_submit, returned %d\n", ret );
    383                 if( ret < 0 ) {
    384                         switch((int)errno) {
    385                         default:
    386                                 abort( "KERNEL ERROR: IO_URING SUBMIT - %s\n", strerror(errno) );
    387                         }
     295static inline [* struct io_uring_sqe, uint32_t] __submit_alloc( struct io_ring & ring ) {
     296        // Wait for a spot to be available
     297        P(ring.submit);
     298
     299        // Allocate the sqe
     300        uint32_t idx = __atomic_fetch_add(&ring.submit_q.alloc, 1ul32, __ATOMIC_SEQ_CST);
     301
     302        // Validate that we didn't overflow anything
     303        // Check that nothing overflowed
     304        /* paranoid */ verify( true );
     305
     306        // Check that it goes head -> tail -> alloc and never head -> alloc -> tail
     307        /* paranoid */ verify( true );
     308
     309        // Return the sqe
     310        return [&ring.submit_q.sqes[ idx & (*ring.submit_q.mask)], idx];
     311}
     312
     313static inline void __submit( struct io_ring & ring, uint32_t idx ) {
     314        // get mutual exclusion
     315        lock(ring.submit_q.lock __cfaabi_dbg_ctx2);
     316
     317        // Append to the list of ready entries
     318        uint32_t * tail = ring.submit_q.tail;
     319        const uint32_t mask = *ring.submit_q.mask;
     320
     321        ring.submit_q.array[ (*tail) & mask ] = idx & mask;
     322        __atomic_fetch_add(tail, 1ul32, __ATOMIC_SEQ_CST);
     323
     324        // Submit however, many entries need to be submitted
     325        int ret = syscall( __NR_io_uring_enter, ring.fd, 1, 0, 0, 0p, 0);
     326        // __cfaabi_bits_print_safe( STDERR_FILENO, "Performed io_submit, returned %d\n", ret );
     327        if( ret < 0 ) {
     328                switch((int)errno) {
     329                default:
     330                        abort( "KERNEL ERROR: IO_URING SUBMIT - %s\n", strerror(errno) );
    388331                }
    389 
    390                 // update statistics
    391                 #if !defined(__CFA_NO_STATISTICS__)
    392                         ring.submit_q.stats.submit_avg.val += 1;
    393                         ring.submit_q.stats.submit_avg.cnt += 1;
    394                 #endif
    395 
    396                 unlock(ring.submit_q.lock);
    397                 // Make sure that idx was submitted
    398                 // Be careful to not get false positive if we cycled the entire list or that someone else submitted for us
    399         }
    400 
    401         static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd) {
    402                 this.opcode = opcode;
    403                 #if !defined(IOSQE_ASYNC)
    404                         this.flags = 0;
    405                 #else
    406                         this.flags = IOSQE_ASYNC;
    407                 #endif
    408                 this.ioprio = 0;
    409                 this.fd = fd;
    410                 this.off = 0;
    411                 this.addr = 0;
    412                 this.len = 0;
    413                 this.rw_flags = 0;
    414                 this.__pad2[0] = this.__pad2[1] = this.__pad2[2] = 0;
    415         }
    416 
    417         static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd, void * addr, uint32_t len, uint64_t off ) {
    418                 (this){ opcode, fd };
    419                 this.off = off;
    420                 this.addr = (uint64_t)addr;
    421                 this.len = len;
    422         }
    423 #endif
     332        }
     333
     334        unlock(ring.submit_q.lock);
     335        // Make sure that idx was submitted
     336        // Be careful to not get false positive if we cycled the entire list or that someone else submitted for us
     337}
     338
     339static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd) {
     340        this.opcode = opcode;
     341        #if !defined(IOSQE_ASYNC)
     342                this.flags = 0;
     343        #else
     344                this.flags = IOSQE_ASYNC;
     345        #endif
     346        this.ioprio = 0;
     347        this.fd = fd;
     348        this.off = 0;
     349        this.addr = 0;
     350        this.len = 0;
     351        this.rw_flags = 0;
     352        this.__pad2[0] = this.__pad2[1] = this.__pad2[2] = 0;
     353}
     354
     355static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd, void * addr, uint32_t len, uint64_t off ) {
     356        (this){ opcode, fd };
     357        this.off = off;
     358        this.addr = (uint64_t)addr;
     359        this.len = len;
     360}
    424361
    425362//=============================================================================================
    426363// I/O Interface
    427364//=============================================================================================
    428 #if defined(HAVE_LINUX_IO_URING_H)
     365        extern "C" {
     366                #define __USE_GNU
     367                #define _GNU_SOURCE
     368                #include <fcntl.h>
     369                #include <sys/uio.h>
     370                #include <sys/socket.h>
     371                #include <sys/stat.h>
     372        }
     373
    429374        #define __submit_prelude \
    430375                struct io_ring & ring = active_cluster()->io; \
     
    440385                park( __cfaabi_dbg_ctx ); \
    441386                return data.result;
    442 #endif
    443 
    444 // Some forward declarations
    445 extern "C" {
    446         #include <sys/types.h>
    447         struct iovec;
    448         extern ssize_t preadv2 (int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
    449         extern ssize_t pwritev2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
    450 
    451         extern int fsync(int fd);
    452         extern int sync_file_range(int fd, int64_t offset, int64_t nbytes, unsigned int flags);
    453 
    454         struct msghdr;
    455         struct sockaddr;
    456         extern ssize_t sendmsg(int sockfd, const struct msghdr *msg, int flags);
    457         extern ssize_t recvmsg(int sockfd, struct msghdr *msg, int flags);
    458         extern ssize_t send(int sockfd, const void *buf, size_t len, int flags);
    459         extern ssize_t recv(int sockfd, void *buf, size_t len, int flags);
    460         extern int accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags);
    461         extern int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
    462 
    463         extern int fallocate(int fd, int mode, uint64_t offset, uint64_t len);
    464         extern int posix_fadvise(int fd, uint64_t offset, uint64_t len, int advice);
    465         extern int madvise(void *addr, size_t length, int advice);
    466 
    467         extern int openat(int dirfd, const char *pathname, int flags, mode_t mode);
    468         extern int close(int fd);
    469 
    470         struct statx;
    471         extern int statx(int dirfd, const char *pathname, int flags, unsigned int mask, struct statx *statxbuf);
    472 
    473         extern ssize_t read (int fd, void *buf, size_t count);
    474 }
    475387
    476388//-----------------------------------------------------------------------------
    477389// Asynchronous operations
    478 ssize_t cfa_preadv2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
    479         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_READV)
    480                 return preadv2(fd, iov, iovcnt, offset, flags);
    481         #else
    482                 __submit_prelude
    483 
    484                 (*sqe){ IORING_OP_READV, fd, iov, iovcnt, offset };
    485 
    486                 __submit_wait
    487         #endif
    488 }
    489 
    490 ssize_t cfa_pwritev2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
    491         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_WRITEV)
    492                 return pwritev2(fd, iov, iovcnt, offset, flags);
    493         #else
    494                 __submit_prelude
    495 
    496                 (*sqe){ IORING_OP_WRITEV, fd, iov, iovcnt, offset };
    497 
    498                 __submit_wait
    499         #endif
    500 }
    501 
    502 int cfa_fsync(int fd) {
    503         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_FSYNC)
    504                 return fsync(fd);
    505         #else
    506                 __submit_prelude
    507 
    508                 (*sqe){ IORING_OP_FSYNC, fd };
    509 
    510                 __submit_wait
    511         #endif
    512 }
    513 
    514 int cfa_sync_file_range(int fd, int64_t offset, int64_t nbytes, unsigned int flags) {
    515         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_SYNC_FILE_RANGE)
    516                 return sync_file_range(fd, offset, nbytes, flags);
    517         #else
    518                 __submit_prelude
    519 
    520                 (*sqe){ IORING_OP_SYNC_FILE_RANGE, fd };
    521                 sqe->off = offset;
    522                 sqe->len = nbytes;
    523                 sqe->sync_range_flags = flags;
    524 
    525                 __submit_wait
    526         #endif
    527 }
    528 
    529 
    530 ssize_t cfa_sendmsg(int sockfd, const struct msghdr *msg, int flags) {
    531         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_SENDMSG)
    532                 return recv(sockfd, msg, flags);
    533         #else
    534                 __submit_prelude
    535 
    536                 (*sqe){ IORING_OP_SENDMSG, sockfd, msg, 1, 0 };
    537                 sqe->msg_flags = flags;
    538 
    539                 __submit_wait
    540         #endif
    541 }
    542 
    543 ssize_t cfa_recvmsg(int sockfd, struct msghdr *msg, int flags) {
    544         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_RECVMSG)
    545                 return recv(sockfd, msg, flags);
    546         #else
    547                 __submit_prelude
    548 
    549                 (*sqe){ IORING_OP_RECVMSG, sockfd, msg, 1, 0 };
    550                 sqe->msg_flags = flags;
    551 
    552                 __submit_wait
    553         #endif
    554 }
    555 
    556 ssize_t cfa_send(int sockfd, const void *buf, size_t len, int flags) {
    557         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_SEND)
    558                 return send( sockfd, buf, len, flags );
    559         #else
    560                 __submit_prelude
    561 
    562                 (*sqe){ IORING_OP_SEND, sockfd };
    563                 sqe->addr = (uint64_t)buf;
    564                 sqe->len = len;
    565                 sqe->msg_flags = flags;
    566 
    567                 __submit_wait
    568         #endif
    569 }
    570 
    571 ssize_t cfa_recv(int sockfd, void *buf, size_t len, int flags) {
    572         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_RECV)
    573                 return recv( sockfd, buf, len, flags );
    574         #else
    575                 __submit_prelude
    576 
    577                 (*sqe){ IORING_OP_RECV, sockfd };
    578                 sqe->addr = (uint64_t)buf;
    579                 sqe->len = len;
    580                 sqe->msg_flags = flags;
    581 
    582                 __submit_wait
    583         #endif
    584 }
    585 
    586 int cfa_accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags) {
    587         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_ACCEPT)
    588                 return accept4( sockfd, addr, addrlen, flags );
    589         #else
    590                 __submit_prelude
    591 
    592                 (*sqe){ IORING_OP_ACCEPT, sockfd };
    593                 sqe->addr = addr;
    594                 sqe->addr2 = addrlen;
    595                 sqe->accept_flags = flags;
    596 
    597                 __submit_wait
    598         #endif
    599 }
    600 
    601 int cfa_connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen) {
    602         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_CONNECT)
    603                 return connect( sockfd, addr, addrlen );
    604         #else
    605                 __submit_prelude
    606 
    607                 (*sqe){ IORING_OP_CONNECT, sockfd };
    608                 sqe->addr = (uint64_t)addr;
    609                 sqe->off = addrlen;
    610 
    611                 __submit_wait
    612         #endif
    613 }
    614 
    615 int cfa_fallocate(int fd, int mode, uint64_t offset, uint64_t len) {
    616         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_FALLOCATE)
    617                 return fallocate( fd, mode, offset, len );
    618         #else
    619                 __submit_prelude
    620 
    621                 (*sqe){ IORING_OP_FALLOCATE, fd };
    622                 sqe->off = offset;
    623                 sqe->len = length;
    624                 sqe->mode = mode;
    625 
    626                 __submit_wait
    627         #endif
    628 }
    629 
    630 int cfa_fadvise(int fd, uint64_t offset, uint64_t len, int advice) {
    631         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_FADVISE)
    632                 return posix_fadvise( fd, offset, len, advice );
    633         #else
    634                 __submit_prelude
    635 
    636                 (*sqe){ IORING_OP_FADVISE, fd };
    637                 sqe->off = (uint64_t)offset;
    638                 sqe->len = length;
    639                 sqe->fadvise_advice = advice;
    640 
    641                 __submit_wait
    642         #endif
    643 }
    644 
    645 int cfa_madvise(void *addr, size_t length, int advice) {
    646         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_MADVISE)
    647                 return madvise( addr, length, advice );
    648         #else
    649                 __submit_prelude
    650 
    651                 (*sqe){ IORING_OP_MADVISE, 0 };
    652                 sqe->addr = (uint64_t)addr;
    653                 sqe->len = length;
    654                 sqe->fadvise_advice = advice;
    655 
    656                 __submit_wait
    657         #endif
    658 }
    659 
    660 int cfa_openat(int dirfd, const char *pathname, int flags, mode_t mode) {
    661         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_OPENAT)
    662                 return openat( dirfd, pathname, flags, mode );
    663         #else
    664                 __submit_prelude
    665 
    666                 (*sqe){ IORING_OP_OPENAT, dirfd };
    667                 sqe->addr = (uint64_t)pathname;
    668                 sqe->open_flags = flags;
    669                 sqe->mode = mode;
    670 
    671                 __submit_wait
    672         #endif
    673 }
    674 
    675 int cfa_close(int fd) {
    676         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_CLOSE)
    677                 return close( fd );
    678         #else
    679                 __submit_prelude
    680 
    681                 (*sqe){ IORING_OP_CLOSE, fd };
    682 
    683                 __submit_wait
    684         #endif
    685 }
    686 
    687 int cfa_statx(int dirfd, const char *pathname, int flags, unsigned int mask, struct statx *statxbuf) {
    688         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_STATX)
    689                 //return statx( dirfd, pathname, flags, mask, statxbuf );
    690                 return syscall( __NR_io_uring_setup, dirfd, pathname, flags, mask, statxbuf );
    691         #else
    692                 __submit_prelude
    693 
    694                 (*sqe){ IORING_OP_STATX, dirfd };
    695                 sqe->addr = (uint64_t)pathname;
    696                 sqe->statx_flags = flags;
    697                 sqe->len = mask;
    698                 sqe->off = (uint64_t)statxbuf;
    699 
    700                 __submit_wait
    701         #endif
    702 }
    703 
    704 
    705 ssize_t cfa_read(int fd, void *buf, size_t count) {
    706         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_READ)
    707                 return read( fd, buf, count );
    708         #else
    709                 __submit_prelude
    710 
    711                 (*sqe){ IORING_OP_READ, fd, buf, count, 0 };
    712 
    713                 __submit_wait
    714         #endif
    715 }
    716 
    717 ssize_t cfa_write(int fd, void *buf, size_t count) {
    718         #if !defined(HAVE_LINUX_IO_URING_H) || !defined(IORING_OP_WRITE)
    719                 return read( fd, buf, count );
    720         #else
    721                 __submit_prelude
    722 
    723                 (*sqe){ IORING_OP_WRITE, fd, buf, count, 0 };
    724 
    725                 __submit_wait
    726         #endif
    727 }
     390        ssize_t async_preadv2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
     391                #if !defined(IORING_OP_READV)
     392                        return preadv2(fd, iov, iovcnt, offset, flags);
     393                #else
     394                        __submit_prelude
     395
     396                        (*sqe){ IORING_OP_READV, fd, iov, iovcnt, offset };
     397
     398                        __submit_wait
     399                #endif
     400        }
     401
     402        ssize_t async_pwritev2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
     403                #if !defined(IORING_OP_WRITEV)
     404                        return pwritev2(fd, iov, iovcnt, offset, flags);
     405                #else
     406                        __submit_prelude
     407
     408                        (*sqe){ IORING_OP_WRITEV, fd, iov, iovcnt, offset };
     409
     410                        __submit_wait
     411                #endif
     412        }
     413
     414        int async_fsync(int fd) {
     415                #if !defined(IORING_OP_FSYNC)
     416                        return fsync(fd);
     417                #else
     418                        __submit_prelude
     419
     420                        (*sqe){ IORING_OP_FSYNC, fd };
     421
     422                        __submit_wait
     423                #endif
     424        }
     425
     426        int async_sync_file_range(int fd, int64_t offset, int64_t nbytes, unsigned int flags) {
     427                #if !defined(IORING_OP_SYNC_FILE_RANGE)
     428                        return sync_file_range(fd, offset, nbytes, flags);
     429                #else
     430                        __submit_prelude
     431
     432                        (*sqe){ IORING_OP_SYNC_FILE_RANGE, fd };
     433                        sqe->off = offset;
     434                        sqe->len = nbytes;
     435                        sqe->sync_range_flags = flags;
     436
     437                        __submit_wait
     438                #endif
     439        }
     440
     441
     442        ssize_t async_sendmsg(int sockfd, const struct msghdr *msg, int flags) {
     443                #if !defined(IORING_OP_SENDMSG)
     444                        return recv(sockfd, msg, flags);
     445                #else
     446                        __submit_prelude
     447
     448                        (*sqe){ IORING_OP_SENDMSG, sockfd, msg, 1, 0 };
     449                        sqe->msg_flags = flags;
     450
     451                        __submit_wait
     452                #endif
     453        }
     454
     455        ssize_t async_recvmsg(int sockfd, struct msghdr *msg, int flags) {
     456                #if !defined(IORING_OP_RECVMSG)
     457                        return recv(sockfd, msg, flags);
     458                #else
     459                        __submit_prelude
     460
     461                        (*sqe){ IORING_OP_RECVMSG, sockfd, msg, 1, 0 };
     462                        sqe->msg_flags = flags;
     463
     464                        __submit_wait
     465                #endif
     466        }
     467
     468        ssize_t async_send(int sockfd, const void *buf, size_t len, int flags) {
     469                #if !defined(IORING_OP_SEND)
     470                        return send( sockfd, buf, len, flags );
     471                #else
     472                        __submit_prelude
     473
     474                        (*sqe){ IORING_OP_SEND, sockfd };
     475                        sqe->addr = (uint64_t)buf;
     476                        sqe->len = len;
     477                        sqe->msg_flags = flags;
     478
     479                        __submit_wait
     480                #endif
     481        }
     482
     483        ssize_t async_recv(int sockfd, void *buf, size_t len, int flags) {
     484                #if !defined(IORING_OP_RECV)
     485                        return recv( sockfd, buf, len, flags );
     486                #else
     487                        __submit_prelude
     488
     489                        (*sqe){ IORING_OP_RECV, sockfd };
     490                        sqe->addr = (uint64_t)buf;
     491                        sqe->len = len;
     492                        sqe->msg_flags = flags;
     493
     494                        __submit_wait
     495                #endif
     496        }
     497
     498        int async_accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags) {
     499                #if !defined(IORING_OP_ACCEPT)
     500                        __SOCKADDR_ARG _addr;
     501                        _addr.__sockaddr__ = addr;
     502                        return accept4( sockfd, _addr, addrlen, flags );
     503                #else
     504                        __submit_prelude
     505
     506                        (*sqe){ IORING_OP_ACCEPT, sockfd };
     507                        sqe->addr = addr;
     508                        sqe->addr2 = addrlen;
     509                        sqe->accept_flags = flags;
     510
     511                        __submit_wait
     512                #endif
     513        }
     514
     515        int async_connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen) {
     516                #if !defined(IORING_OP_CONNECT)
     517                        __CONST_SOCKADDR_ARG _addr;
     518                        _addr.__sockaddr__ = addr;
     519                        return connect( sockfd, _addr, addrlen );
     520                #else
     521                        __submit_prelude
     522
     523                        (*sqe){ IORING_OP_CONNECT, sockfd };
     524                        sqe->addr = (uint64_t)addr;
     525                        sqe->off = addrlen;
     526
     527                        __submit_wait
     528                #endif
     529        }
     530
     531        int async_fallocate(int fd, int mode, uint64_t offset, uint64_t len) {
     532                #if !defined(IORING_OP_FALLOCATE)
     533                        return fallocate( fd, mode, offset, len );
     534                #else
     535                        __submit_prelude
     536
     537                        (*sqe){ IORING_OP_FALLOCATE, fd };
     538                        sqe->off = offset;
     539                        sqe->len = length;
     540                        sqe->mode = mode;
     541
     542                        __submit_wait
     543                #endif
     544        }
     545
     546        int async_fadvise(int fd, uint64_t offset, uint64_t len, int advice) {
     547                #if !defined(IORING_OP_FADVISE)
     548                        return posix_fadvise( fd, offset, len, advice );
     549                #else
     550                        __submit_prelude
     551
     552                        (*sqe){ IORING_OP_FADVISE, fd };
     553                        sqe->off = (uint64_t)offset;
     554                        sqe->len = length;
     555                        sqe->fadvise_advice = advice;
     556
     557                        __submit_wait
     558                #endif
     559        }
     560
     561        int async_madvise(void *addr, size_t length, int advice) {
     562                #if !defined(IORING_OP_MADVISE)
     563                        return madvise( addr, length, advice );
     564                #else
     565                        __submit_prelude
     566
     567                        (*sqe){ IORING_OP_MADVISE, 0 };
     568                        sqe->addr = (uint64_t)addr;
     569                        sqe->len = length;
     570                        sqe->fadvise_advice = advice;
     571
     572                        __submit_wait
     573                #endif
     574        }
     575
     576        int async_openat(int dirfd, const char *pathname, int flags, mode_t mode) {
     577                #if !defined(IORING_OP_OPENAT)
     578                        return openat( dirfd, pathname, flags, mode );
     579                #else
     580                        __submit_prelude
     581
     582                        (*sqe){ IORING_OP_OPENAT, dirfd };
     583                        sqe->addr = (uint64_t)pathname;
     584                        sqe->open_flags = flags;
     585                        sqe->mode = mode;
     586
     587                        __submit_wait
     588                #endif
     589        }
     590
     591        int async_close(int fd) {
     592                #if !defined(IORING_OP_CLOSE)
     593                        return close( fd );
     594                #else
     595                        __submit_prelude
     596
     597                        (*sqe){ IORING_OP_CLOSE, fd };
     598
     599                        __submit_wait
     600                #endif
     601        }
     602
     603        int async_statx(int dirfd, const char *pathname, int flags, unsigned int mask, struct statx *statxbuf) {
     604                #if !defined(IORING_OP_STATX)
     605                        //return statx( dirfd, pathname, flags, mask, statxbuf );
     606                        return syscall( __NR_io_uring_setup, dirfd, pathname, flags, mask, statxbuf );
     607                #else
     608                        __submit_prelude
     609
     610                        (*sqe){ IORING_OP_STATX, dirfd };
     611                        sqe->addr = (uint64_t)pathname;
     612                        sqe->statx_flags = flags;
     613                        sqe->len = mask;
     614                        sqe->off = (uint64_t)statxbuf;
     615
     616                        __submit_wait
     617                #endif
     618        }
     619
     620
     621        ssize_t async_read(int fd, void *buf, size_t count) {
     622                #if !defined(IORING_OP_READ)
     623                        return read( fd, buf, count );
     624                #else
     625                        __submit_prelude
     626
     627                        (*sqe){ IORING_OP_READ, fd, buf, count, 0 };
     628
     629                        __submit_wait
     630                #endif
     631        }
     632
     633        ssize_t async_write(int fd, void *buf, size_t count) {
     634                #if !defined(IORING_OP_WRITE)
     635                        return read( fd, buf, count );
     636                #else
     637                        __submit_prelude
     638
     639                        (*sqe){ IORING_OP_WRITE, fd, buf, count, 0 };
     640
     641                        __submit_wait
     642                #endif
     643        }
    728644
    729645//-----------------------------------------------------------------------------
     
    731647
    732648// Macro magic to reduce the size of the following switch case
    733 #define IS_DEFINED_APPLY(f, ...) f(__VA_ARGS__)
    734 #define IS_DEFINED_SECOND(first, second, ...) second
    735 #define IS_DEFINED_TEST(expansion) _CFA_IO_FEATURE_##expansion
    736 #define IS_DEFINED(macro) IS_DEFINED_APPLY( IS_DEFINED_SECOND,IS_DEFINED_TEST(macro) false, true)
    737 
    738 bool has_user_level_blocking( fptr_t func ) {
    739         #if defined(HAVE_LINUX_IO_URING_H)
     649        #define IS_DEFINED_APPLY(f, ...) f(__VA_ARGS__)
     650        #define IS_DEFINED_SECOND(first, second, ...) second
     651        #define IS_DEFINED_TEST(expansion) _CFA_IO_FEATURE_##expansion
     652        #define IS_DEFINED(macro) IS_DEFINED_APPLY( IS_DEFINED_SECOND,IS_DEFINED_TEST(macro) false, true)
     653
     654        bool is_async( fptr_t func ) {
     655
    740656                if( /*func == (fptr_t)preadv2 || */
    741                         func == (fptr_t)cfa_preadv2 )
     657                        func == (fptr_t)async_preadv2 )
    742658                        #define _CFA_IO_FEATURE_IORING_OP_READV ,
    743659                        return IS_DEFINED(IORING_OP_READV);
    744660
    745661                if( /*func == (fptr_t)pwritev2 || */
    746                         func == (fptr_t)cfa_pwritev2 )
     662                      func == (fptr_t)async_pwritev2 )
    747663                        #define _CFA_IO_FEATURE_IORING_OP_WRITEV ,
    748664                        return IS_DEFINED(IORING_OP_WRITEV);
    749665
    750666                if( /*func == (fptr_t)fsync || */
    751                         func == (fptr_t)cfa_fsync )
     667                      func == (fptr_t)async_fsync )
    752668                        #define _CFA_IO_FEATURE_IORING_OP_FSYNC ,
    753669                        return IS_DEFINED(IORING_OP_FSYNC);
    754670
    755671                if( /*func == (fptr_t)ync_file_range || */
    756                         func == (fptr_t)cfa_sync_file_range )
     672                      func == (fptr_t)async_sync_file_range )
    757673                        #define _CFA_IO_FEATURE_IORING_OP_SYNC_FILE_RANGE ,
    758674                        return IS_DEFINED(IORING_OP_SYNC_FILE_RANGE);
    759675
    760676                if( /*func == (fptr_t)sendmsg || */
    761                         func == (fptr_t)cfa_sendmsg )
     677                      func == (fptr_t)async_sendmsg )
    762678                        #define _CFA_IO_FEATURE_IORING_OP_SENDMSG ,
    763679                        return IS_DEFINED(IORING_OP_SENDMSG);
    764680
    765681                if( /*func == (fptr_t)recvmsg || */
    766                         func == (fptr_t)cfa_recvmsg )
     682                      func == (fptr_t)async_recvmsg )
    767683                        #define _CFA_IO_FEATURE_IORING_OP_RECVMSG ,
    768684                        return IS_DEFINED(IORING_OP_RECVMSG);
    769685
    770686                if( /*func == (fptr_t)send || */
    771                         func == (fptr_t)cfa_send )
     687                        func == (fptr_t)async_send )
    772688                        #define _CFA_IO_FEATURE_IORING_OP_SEND ,
    773689                        return IS_DEFINED(IORING_OP_SEND);
    774690
    775691                if( /*func == (fptr_t)recv || */
    776                         func == (fptr_t)cfa_recv )
     692                        func == (fptr_t)async_recv )
    777693                        #define _CFA_IO_FEATURE_IORING_OP_RECV ,
    778694                        return IS_DEFINED(IORING_OP_RECV);
    779695
    780696                if( /*func == (fptr_t)accept4 || */
    781                         func == (fptr_t)cfa_accept4 )
     697                        func == (fptr_t)async_accept4 )
    782698                        #define _CFA_IO_FEATURE_IORING_OP_ACCEPT ,
    783699                        return IS_DEFINED(IORING_OP_ACCEPT);
    784700
    785701                if( /*func == (fptr_t)connect || */
    786                         func == (fptr_t)cfa_connect )
     702                        func == (fptr_t)async_connect )
    787703                        #define _CFA_IO_FEATURE_IORING_OP_CONNECT ,
    788704                        return IS_DEFINED(IORING_OP_CONNECT);
    789705
    790706                if( /*func == (fptr_t)fallocate || */
    791                         func == (fptr_t)cfa_fallocate )
     707                        func == (fptr_t)async_fallocate )
    792708                        #define _CFA_IO_FEATURE_IORING_OP_FALLOCATE ,
    793709                        return IS_DEFINED(IORING_OP_FALLOCATE);
    794710
    795                 if( /*func == (fptr_t)posix_fadvise || */
    796                         func == (fptr_t)cfa_fadvise )
     711                if( /*func == (fptr_t)fadvise || */
     712                        func == (fptr_t)async_fadvise )
    797713                        #define _CFA_IO_FEATURE_IORING_OP_FADVISE ,
    798714                        return IS_DEFINED(IORING_OP_FADVISE);
    799715
    800716                if( /*func == (fptr_t)madvise || */
    801                         func == (fptr_t)cfa_madvise )
     717                        func == (fptr_t)async_madvise )
    802718                        #define _CFA_IO_FEATURE_IORING_OP_MADVISE ,
    803719                        return IS_DEFINED(IORING_OP_MADVISE);
    804720
    805721                if( /*func == (fptr_t)openat || */
    806                         func == (fptr_t)cfa_openat )
     722                        func == (fptr_t)async_openat )
    807723                        #define _CFA_IO_FEATURE_IORING_OP_OPENAT ,
    808724                        return IS_DEFINED(IORING_OP_OPENAT);
    809725
    810726                if( /*func == (fptr_t)close || */
    811                         func == (fptr_t)cfa_close )
     727                        func == (fptr_t)async_close )
    812728                        #define _CFA_IO_FEATURE_IORING_OP_CLOSE ,
    813729                        return IS_DEFINED(IORING_OP_CLOSE);
    814730
    815731                if( /*func == (fptr_t)statx || */
    816                         func == (fptr_t)cfa_statx )
     732                        func == (fptr_t)async_statx )
    817733                        #define _CFA_IO_FEATURE_IORING_OP_STATX ,
    818734                        return IS_DEFINED(IORING_OP_STATX);
    819735
    820736                if( /*func == (fptr_t)read || */
    821                         func == (fptr_t)cfa_read )
     737                      func == (fptr_t)async_read )
    822738                        #define _CFA_IO_FEATURE_IORING_OP_READ ,
    823739                        return IS_DEFINED(IORING_OP_READ);
    824740
    825741                if( /*func == (fptr_t)write || */
    826                         func == (fptr_t)cfa_write )
     742                      func == (fptr_t)async_write )
    827743                        #define _CFA_IO_FEATURE_IORING_OP_WRITE ,
    828744                        return IS_DEFINED(IORING_OP_WRITE);
    829         #endif
    830 
    831         return false;
    832 }
     745
     746                return false;
     747        }
     748
     749#endif
  • libcfa/src/concurrency/kernel.cfa

    r44aad8f rc680a4b  
    257257        ready_queue{};
    258258        ready_queue_lock{};
    259 
    260         #if !defined(__CFA_NO_STATISTICS__)
    261                 print_stats = false;
    262         #endif
    263259
    264260        procs{ __get };
     
    10081004}
    10091005
    1010 bool V(semaphore & this, unsigned diff) with( this ) {
    1011         $thread * thrd = 0p;
    1012         lock( lock __cfaabi_dbg_ctx2 );
    1013         int release = max(-count, (int)diff);
    1014         count += diff;
    1015         for(release) {
    1016                 unpark( pop_head( waiting ) __cfaabi_dbg_ctx2 );
    1017         }
    1018 
    1019         unlock( lock );
    1020 
    1021         return thrd != 0p;
    1022 }
    1023 
    10241006//-----------------------------------------------------------------------------
    10251007// Global Queues
  • libcfa/src/concurrency/kernel.hfa

    r44aad8f rc680a4b  
    4040void   P (semaphore & this);
    4141bool   V (semaphore & this);
    42 bool   V (semaphore & this, unsigned count);
    4342
    4443
     
    145144        void * ring_ptr;
    146145        size_t ring_sz;
    147 
    148         // Statistics
    149         #if !defined(__CFA_NO_STATISTICS__)
    150                 struct {
    151                         struct {
    152                                 unsigned long long int val;
    153                                 unsigned long long int cnt;
    154                         } submit_avg;
    155                 } stats;
    156         #endif
    157146};
    158147
     
    175164        void * ring_ptr;
    176165        size_t ring_sz;
    177 
    178         // Statistics
    179         #if !defined(__CFA_NO_STATISTICS__)
    180                 struct {
    181                         struct {
    182                                 unsigned long long int val;
    183                                 unsigned long long int cnt;
    184                         } completed_avg;
    185                 } stats;
    186         #endif
    187166};
    188167
     
    234213                struct io_ring io;
    235214        #endif
    236 
    237         #if !defined(__CFA_NO_STATISTICS__)
    238                 bool print_stats;
    239         #endif
    240215};
    241216extern Duration default_preemption();
     
    252227static inline struct processor * active_processor() { return TL_GET( this_processor ); } // UNSAFE
    253228static inline struct cluster   * active_cluster  () { return TL_GET( this_processor )->cltr; }
    254 
    255 #if !defined(__CFA_NO_STATISTICS__)
    256         static inline void print_stats_at_exit( cluster & this ) {
    257                 this.print_stats = true;
    258         }
    259 #endif
    260229
    261230// Local Variables: //
Note: See TracChangeset for help on using the changeset viewer.