examples/performance-thread/pthread_shim/pthread_shim.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2015 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <sys/types.h>
  37 #include <errno.h>
  38 #define __USE_GNU
  39 #include <sched.h>
  40 #include <dlfcn.h>
  41
  42 #include <rte_log.h>
  43
  44 #include "lthread_api.h"
  45 #include "pthread_shim.h"
  46
  47 #define RTE_LOGTYPE_PTHREAD_SHIM RTE_LOGTYPE_USER3
  48
  49 #define POSIX_ERRNO(x)  (x)
  50
  51 /* some releases of FreeBSD 10, e.g. 10.0, don't have CPU_COUNT macro */
  52 #ifndef CPU_COUNT
  53 #define CPU_COUNT(x) __cpu_count(x)
  54
  55 static inline unsigned int
  56 __cpu_count(const rte_cpuset_t *cpuset)
  57 {
  58         unsigned int i, count = 0;
  59         for (i = 0; i < RTE_MAX_LCORE; i++)
  60                 if (CPU_ISSET(i, cpuset))
  61                         count++;
  62         return count;
  63 }
  64 #endif
  65
  66 /*
  67  * this flag determines at run time if we override pthread
  68  * calls and map then to equivalent lthread calls
  69  * or of we call the standard pthread function
  70  */
  71 static __thread int override;
  72
  73
  74 /*
  75  * this structures contains function pointers that will be
  76  * initialised to the loaded address of the real
  77  * pthread library API functions
  78  */
  79 struct pthread_lib_funcs {
  80 int (*f_pthread_barrier_destroy)
  81         (pthread_barrier_t *);
  82 int (*f_pthread_barrier_init)
  83         (pthread_barrier_t *, const pthread_barrierattr_t *, unsigned);
  84 int (*f_pthread_barrier_wait)
  85         (pthread_barrier_t *);
  86 int (*f_pthread_cond_broadcast)
  87         (pthread_cond_t *);
  88 int (*f_pthread_cond_destroy)
  89         (pthread_cond_t *);
  90 int (*f_pthread_cond_init)
  91         (pthread_cond_t *, const pthread_condattr_t *);
  92 int (*f_pthread_cond_signal)
  93         (pthread_cond_t *);
  94 int (*f_pthread_cond_timedwait)
  95         (pthread_cond_t *, pthread_mutex_t *, const struct timespec *);
  96 int (*f_pthread_cond_wait)
  97         (pthread_cond_t *, pthread_mutex_t *);
  98 int (*f_pthread_create)
  99         (pthread_t *, const pthread_attr_t *, void *(*)(void *), void *);
 100 int (*f_pthread_detach)
 101         (pthread_t);
 102 int (*f_pthread_equal)
 103         (pthread_t, pthread_t);
 104 void (*f_pthread_exit)
 105         (void *);
 106 void * (*f_pthread_getspecific)
 107         (pthread_key_t);
 108 int (*f_pthread_getcpuclockid)
 109         (pthread_t, clockid_t *);
 110 int (*f_pthread_join)
 111         (pthread_t, void **);
 112 int (*f_pthread_key_create)
 113         (pthread_key_t *, void (*) (void *));
 114 int (*f_pthread_key_delete)
 115         (pthread_key_t);
 116 int (*f_pthread_mutex_destroy)
 117         (pthread_mutex_t *__mutex);
 118 int (*f_pthread_mutex_init)
 119         (pthread_mutex_t *__mutex, const pthread_mutexattr_t *);
 120 int (*f_pthread_mutex_lock)
 121         (pthread_mutex_t *__mutex);
 122 int (*f_pthread_mutex_trylock)
 123         (pthread_mutex_t *__mutex);
 124 int (*f_pthread_mutex_timedlock)
 125         (pthread_mutex_t *__mutex, const struct timespec *);
 126 int (*f_pthread_mutex_unlock)
 127         (pthread_mutex_t *__mutex);
 128 int (*f_pthread_once)
 129         (pthread_once_t *, void (*) (void));
 130 int (*f_pthread_rwlock_destroy)
 131         (pthread_rwlock_t *__rwlock);
 132 int (*f_pthread_rwlock_init)
 133         (pthread_rwlock_t *__rwlock, const pthread_rwlockattr_t *);
 134 int (*f_pthread_rwlock_rdlock)
 135         (pthread_rwlock_t *__rwlock);
 136 int (*f_pthread_rwlock_timedrdlock)
 137         (pthread_rwlock_t *__rwlock, const struct timespec *);
 138 int (*f_pthread_rwlock_timedwrlock)
 139         (pthread_rwlock_t *__rwlock, const struct timespec *);
 140 int (*f_pthread_rwlock_tryrdlock)
 141         (pthread_rwlock_t *__rwlock);
 142 int (*f_pthread_rwlock_trywrlock)
 143         (pthread_rwlock_t *__rwlock);
 144 int (*f_pthread_rwlock_unlock)
 145         (pthread_rwlock_t *__rwlock);
 146 int (*f_pthread_rwlock_wrlock)
 147         (pthread_rwlock_t *__rwlock);
 148 pthread_t (*f_pthread_self)
 149         (void);
 150 int (*f_pthread_setspecific)
 151         (pthread_key_t, const void *);
 152 int (*f_pthread_spin_init)
 153         (pthread_spinlock_t *__spin, int);
 154 int (*f_pthread_spin_destroy)
 155         (pthread_spinlock_t *__spin);
 156 int (*f_pthread_spin_lock)
 157         (pthread_spinlock_t *__spin);
 158 int (*f_pthread_spin_trylock)
 159         (pthread_spinlock_t *__spin);
 160 int (*f_pthread_spin_unlock)
 161         (pthread_spinlock_t *__spin);
 162 int (*f_pthread_cancel)
 163         (pthread_t);
 164 int (*f_pthread_setcancelstate)
 165         (int, int *);
 166 int (*f_pthread_setcanceltype)
 167         (int, int *);
 168 void (*f_pthread_testcancel)
 169         (void);
 170 int (*f_pthread_getschedparam)
 171         (pthread_t pthread, int *, struct sched_param *);
 172 int (*f_pthread_setschedparam)
 173         (pthread_t, int, const struct sched_param *);
 174 int (*f_pthread_yield)
 175         (void);
 176 int (*f_pthread_setaffinity_np)
 177         (pthread_t thread, size_t cpusetsize, const rte_cpuset_t *cpuset);
 178 int (*f_nanosleep)
 179         (const struct timespec *req, struct timespec *rem);
 180 } _sys_pthread_funcs = {
 181         .f_pthread_barrier_destroy = NULL,
 182 };
 183
 184
 185 /*
 186  * this macro obtains the loaded address of a library function
 187  * and saves it.
 188  */
 189 static void *__libc_dl_handle = RTLD_NEXT;
 190
 191 #define get_addr_of_loaded_symbol(name) do {                            \
 192         char *error_str;                                                \
 193         _sys_pthread_funcs.f_##name = dlsym(__libc_dl_handle, (#name)); \
 194         error_str = dlerror();                                          \
 195         if (error_str != NULL) {                                        \
 196                 fprintf(stderr, "%s\n", error_str);                     \
 197         }                                                               \
 198 } while (0)
 199
 200
 201 /*
 202  * The constructor function initialises the
 203  * function pointers for pthread library functions
 204  */
 205 void
 206 pthread_intercept_ctor(void)__attribute__((constructor));
 207 void
 208 pthread_intercept_ctor(void)
 209 {
 210         override = 0;
 211         /*
 212          * Get the original functions
 213          */
 214         get_addr_of_loaded_symbol(pthread_barrier_destroy);
 215         get_addr_of_loaded_symbol(pthread_barrier_init);
 216         get_addr_of_loaded_symbol(pthread_barrier_wait);
 217         get_addr_of_loaded_symbol(pthread_cond_broadcast);
 218         get_addr_of_loaded_symbol(pthread_cond_destroy);
 219         get_addr_of_loaded_symbol(pthread_cond_init);
 220         get_addr_of_loaded_symbol(pthread_cond_signal);
 221         get_addr_of_loaded_symbol(pthread_cond_timedwait);
 222         get_addr_of_loaded_symbol(pthread_cond_wait);
 223         get_addr_of_loaded_symbol(pthread_create);
 224         get_addr_of_loaded_symbol(pthread_detach);
 225         get_addr_of_loaded_symbol(pthread_equal);
 226         get_addr_of_loaded_symbol(pthread_exit);
 227         get_addr_of_loaded_symbol(pthread_getspecific);
 228         get_addr_of_loaded_symbol(pthread_getcpuclockid);
 229         get_addr_of_loaded_symbol(pthread_join);
 230         get_addr_of_loaded_symbol(pthread_key_create);
 231         get_addr_of_loaded_symbol(pthread_key_delete);
 232         get_addr_of_loaded_symbol(pthread_mutex_destroy);
 233         get_addr_of_loaded_symbol(pthread_mutex_init);
 234         get_addr_of_loaded_symbol(pthread_mutex_lock);
 235         get_addr_of_loaded_symbol(pthread_mutex_trylock);
 236         get_addr_of_loaded_symbol(pthread_mutex_timedlock);
 237         get_addr_of_loaded_symbol(pthread_mutex_unlock);
 238         get_addr_of_loaded_symbol(pthread_once);
 239         get_addr_of_loaded_symbol(pthread_rwlock_destroy);
 240         get_addr_of_loaded_symbol(pthread_rwlock_init);
 241         get_addr_of_loaded_symbol(pthread_rwlock_rdlock);
 242         get_addr_of_loaded_symbol(pthread_rwlock_timedrdlock);
 243         get_addr_of_loaded_symbol(pthread_rwlock_timedwrlock);
 244         get_addr_of_loaded_symbol(pthread_rwlock_tryrdlock);
 245         get_addr_of_loaded_symbol(pthread_rwlock_trywrlock);
 246         get_addr_of_loaded_symbol(pthread_rwlock_unlock);
 247         get_addr_of_loaded_symbol(pthread_rwlock_wrlock);
 248         get_addr_of_loaded_symbol(pthread_self);
 249         get_addr_of_loaded_symbol(pthread_setspecific);
 250         get_addr_of_loaded_symbol(pthread_spin_init);
 251         get_addr_of_loaded_symbol(pthread_spin_destroy);
 252         get_addr_of_loaded_symbol(pthread_spin_lock);
 253         get_addr_of_loaded_symbol(pthread_spin_trylock);
 254         get_addr_of_loaded_symbol(pthread_spin_unlock);
 255         get_addr_of_loaded_symbol(pthread_cancel);
 256         get_addr_of_loaded_symbol(pthread_setcancelstate);
 257         get_addr_of_loaded_symbol(pthread_setcanceltype);
 258         get_addr_of_loaded_symbol(pthread_testcancel);
 259         get_addr_of_loaded_symbol(pthread_getschedparam);
 260         get_addr_of_loaded_symbol(pthread_setschedparam);
 261         get_addr_of_loaded_symbol(pthread_yield);
 262         get_addr_of_loaded_symbol(pthread_setaffinity_np);
 263         get_addr_of_loaded_symbol(nanosleep);
 264 }
 265
 266
 267 /*
 268  * Enable/Disable pthread override
 269  * state
 270  *  0 disable
 271  *  1 enable
 272  */
 273 void pthread_override_set(int state)
 274 {
 275         override = state;
 276 }
 277
 278
 279 /*
 280  * Return pthread override state
 281  * return
 282  *  0 disable
 283  *  1 enable
 284  */
 285 int pthread_override_get(void)
 286 {
 287         return override;
 288 }
 289
 290 /*
 291  * This macro is used to catch and log
 292  * invocation of stubs for unimplemented pthread
 293  * API functions.
 294  */
 295 #define NOT_IMPLEMENTED do {                            \
 296         if (override) {                                 \
 297                 RTE_LOG(WARNING,                        \
 298                         PTHREAD_SHIM,                   \
 299                         "WARNING %s NOT IMPLEMENTED\n", \
 300                         __func__);                      \
 301         }                                               \
 302 } while (0)
 303
 304 /*
 305  * pthread API override functions follow
 306  * Note in this example code only a subset of functions are
 307  * implemented.
 308  *
 309  * The stub functions provided will issue a warning log
 310  * message if an unimplemented function is invoked
 311  *
 312  */
 313
 314 int pthread_barrier_destroy(pthread_barrier_t *a)
 315 {
 316         NOT_IMPLEMENTED;
 317         return _sys_pthread_funcs.f_pthread_barrier_destroy(a);
 318 }
 319
 320 int
 321 pthread_barrier_init(pthread_barrier_t *a,
 322                      const pthread_barrierattr_t *b, unsigned c)
 323 {
 324         NOT_IMPLEMENTED;
 325         return _sys_pthread_funcs.f_pthread_barrier_init(a, b, c);
 326 }
 327
 328 int pthread_barrier_wait(pthread_barrier_t *a)
 329 {
 330         NOT_IMPLEMENTED;
 331         return _sys_pthread_funcs.f_pthread_barrier_wait(a);
 332 }
 333
 334 int pthread_cond_broadcast(pthread_cond_t *cond)
 335 {
 336         if (override) {
 337
 338                 lthread_cond_broadcast(*(struct lthread_cond **)cond);
 339                 return 0;
 340         }
 341         return _sys_pthread_funcs.f_pthread_cond_broadcast(cond);
 342 }
 343
 344 int pthread_mutex_destroy(pthread_mutex_t *mutex)
 345 {
 346         if (override)
 347                 return lthread_mutex_destroy(*(struct lthread_mutex **)mutex);
 348         return _sys_pthread_funcs.f_pthread_mutex_destroy(mutex);
 349 }
 350
 351 int pthread_cond_destroy(pthread_cond_t *cond)
 352 {
 353         if (override)
 354                 return lthread_cond_destroy(*(struct lthread_cond **)cond);
 355         return _sys_pthread_funcs.f_pthread_cond_destroy(cond);
 356 }
 357
 358 int pthread_cond_init(pthread_cond_t *cond, const pthread_condattr_t *attr)
 359 {
 360         if (override)
 361                 return lthread_cond_init(NULL,
 362                                 (struct lthread_cond **)cond,
 363                                 (const struct lthread_condattr *) attr);
 364         return _sys_pthread_funcs.f_pthread_cond_init(cond, attr);
 365 }
 366
 367 int pthread_cond_signal(pthread_cond_t *cond)
 368 {
 369         if (override) {
 370                 lthread_cond_signal(*(struct lthread_cond **)cond);
 371                 return 0;
 372         }
 373         return _sys_pthread_funcs.f_pthread_cond_signal(cond);
 374 }
 375
 376 int
 377 pthread_cond_timedwait(pthread_cond_t *__restrict cond,
 378                        pthread_mutex_t *__restrict mutex,
 379                        const struct timespec *__restrict time)
 380 {
 381         NOT_IMPLEMENTED;
 382         return _sys_pthread_funcs.f_pthread_cond_timedwait(cond, mutex, time);
 383 }
 384
 385 int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex)
 386 {
 387         if (override) {
 388                 pthread_mutex_unlock(mutex);
 389                 int rv = lthread_cond_wait(*(struct lthread_cond **)cond, 0);
 390
 391                 pthread_mutex_lock(mutex);
 392                 return rv;
 393         }
 394         return _sys_pthread_funcs.f_pthread_cond_wait(cond, mutex);
 395 }
 396
 397 int
 398 pthread_create(pthread_t *__restrict tid,
 399                 const pthread_attr_t *__restrict attr,
 400                 lthread_func_t func,
 401                void *__restrict arg)
 402 {
 403         if (override) {
 404                 int lcore = -1;
 405
 406                 if (attr != NULL) {
 407                         /* determine CPU being requested */
 408                         rte_cpuset_t cpuset;
 409
 410                         CPU_ZERO(&cpuset);
 411                         pthread_attr_getaffinity_np(attr,
 412                                                 sizeof(rte_cpuset_t),
 413                                                 &cpuset);
 414
 415                         if (CPU_COUNT(&cpuset) != 1)
 416                                 return POSIX_ERRNO(EINVAL);
 417
 418                         for (lcore = 0; lcore < LTHREAD_MAX_LCORES; lcore++) {
 419                                 if (!CPU_ISSET(lcore, &cpuset))
 420                                         continue;
 421                                 break;
 422                         }
 423                 }
 424                 return lthread_create((struct lthread **)tid, lcore,
 425                                       func, arg);
 426         }
 427         return _sys_pthread_funcs.f_pthread_create(tid, attr, func, arg);
 428 }
 429
 430 int pthread_detach(pthread_t tid)
 431 {
 432         if (override) {
 433                 struct lthread *lt = (struct lthread *)tid;
 434
 435                 if (lt == lthread_current()) {
 436                         lthread_detach();
 437                         return 0;
 438                 }
 439                 NOT_IMPLEMENTED;
 440         }
 441         return _sys_pthread_funcs.f_pthread_detach(tid);
 442 }
 443
 444 int pthread_equal(pthread_t a, pthread_t b)
 445 {
 446         NOT_IMPLEMENTED;
 447         return _sys_pthread_funcs.f_pthread_equal(a, b);
 448 }
 449
 450 void pthread_exit_override(void *v)
 451 {
 452         if (override) {
 453                 lthread_exit(v);
 454                 return;
 455         }
 456         _sys_pthread_funcs.f_pthread_exit(v);
 457 }
 458
 459 void
 460 *pthread_getspecific(pthread_key_t key)
 461 {
 462         if (override)
 463                 return lthread_getspecific((unsigned int) key);
 464         return _sys_pthread_funcs.f_pthread_getspecific(key);
 465 }
 466
 467 int pthread_getcpuclockid(pthread_t a, clockid_t *b)
 468 {
 469         NOT_IMPLEMENTED;
 470         return _sys_pthread_funcs.f_pthread_getcpuclockid(a, b);
 471 }
 472
 473 int pthread_join(pthread_t tid, void **val)
 474 {
 475         if (override)
 476                 return lthread_join((struct lthread *)tid, val);
 477         return _sys_pthread_funcs.f_pthread_join(tid, val);
 478 }
 479
 480 int pthread_key_create(pthread_key_t *keyptr, void (*dtor) (void *))
 481 {
 482         if (override)
 483                 return lthread_key_create((unsigned int *)keyptr, dtor);
 484         return _sys_pthread_funcs.f_pthread_key_create(keyptr, dtor);
 485 }
 486
 487 int pthread_key_delete(pthread_key_t key)
 488 {
 489         if (override) {
 490                 lthread_key_delete((unsigned int) key);
 491                 return 0;
 492         }
 493         return _sys_pthread_funcs.f_pthread_key_delete(key);
 494 }
 495
 496
 497 int
 498 pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr)
 499 {
 500         if (override)
 501                 return lthread_mutex_init(NULL,
 502                                 (struct lthread_mutex **)mutex,
 503                                 (const struct lthread_mutexattr *)attr);
 504         return _sys_pthread_funcs.f_pthread_mutex_init(mutex, attr);
 505 }
 506
 507 int pthread_mutex_lock(pthread_mutex_t *mutex)
 508 {
 509         if (override)
 510                 return lthread_mutex_lock(*(struct lthread_mutex **)mutex);
 511         return _sys_pthread_funcs.f_pthread_mutex_lock(mutex);
 512 }
 513
 514 int pthread_mutex_trylock(pthread_mutex_t *mutex)
 515 {
 516         if (override)
 517                 return lthread_mutex_trylock(*(struct lthread_mutex **)mutex);
 518         return _sys_pthread_funcs.f_pthread_mutex_trylock(mutex);
 519 }
 520
 521 int pthread_mutex_timedlock(pthread_mutex_t *mutex, const struct timespec *b)
 522 {
 523         NOT_IMPLEMENTED;
 524         return _sys_pthread_funcs.f_pthread_mutex_timedlock(mutex, b);
 525 }
 526
 527 int pthread_mutex_unlock(pthread_mutex_t *mutex)
 528 {
 529         if (override)
 530                 return lthread_mutex_unlock(*(struct lthread_mutex **)mutex);
 531         return _sys_pthread_funcs.f_pthread_mutex_unlock(mutex);
 532 }
 533
 534 int pthread_once(pthread_once_t *a, void (b) (void))
 535 {
 536         NOT_IMPLEMENTED;
 537         return _sys_pthread_funcs.f_pthread_once(a, b);
 538 }
 539
 540 int pthread_rwlock_destroy(pthread_rwlock_t *a)
 541 {
 542         NOT_IMPLEMENTED;
 543         return _sys_pthread_funcs.f_pthread_rwlock_destroy(a);
 544 }
 545
 546 int pthread_rwlock_init(pthread_rwlock_t *a, const pthread_rwlockattr_t *b)
 547 {
 548         NOT_IMPLEMENTED;
 549         return _sys_pthread_funcs.f_pthread_rwlock_init(a, b);
 550 }
 551
 552 int pthread_rwlock_rdlock(pthread_rwlock_t *a)
 553 {
 554         NOT_IMPLEMENTED;
 555         return _sys_pthread_funcs.f_pthread_rwlock_rdlock(a);
 556 }
 557
 558 int pthread_rwlock_timedrdlock(pthread_rwlock_t *a, const struct timespec *b)
 559 {
 560         NOT_IMPLEMENTED;
 561         return _sys_pthread_funcs.f_pthread_rwlock_timedrdlock(a, b);
 562 }
 563
 564 int pthread_rwlock_timedwrlock(pthread_rwlock_t *a, const struct timespec *b)
 565 {
 566         NOT_IMPLEMENTED;
 567         return _sys_pthread_funcs.f_pthread_rwlock_timedwrlock(a, b);
 568 }
 569
 570 int pthread_rwlock_tryrdlock(pthread_rwlock_t *a)
 571 {
 572         NOT_IMPLEMENTED;
 573         return _sys_pthread_funcs.f_pthread_rwlock_tryrdlock(a);
 574 }
 575
 576 int pthread_rwlock_trywrlock(pthread_rwlock_t *a)
 577 {
 578         NOT_IMPLEMENTED;
 579         return _sys_pthread_funcs.f_pthread_rwlock_trywrlock(a);
 580 }
 581
 582 int pthread_rwlock_unlock(pthread_rwlock_t *a)
 583 {
 584         NOT_IMPLEMENTED;
 585         return _sys_pthread_funcs.f_pthread_rwlock_unlock(a);
 586 }
 587
 588 int pthread_rwlock_wrlock(pthread_rwlock_t *a)
 589 {
 590         NOT_IMPLEMENTED;
 591         return _sys_pthread_funcs.f_pthread_rwlock_wrlock(a);
 592 }
 593
 594 #ifdef RTE_EXEC_ENV_LINUXAPP
 595 int
 596 pthread_yield(void)
 597 {
 598         if (override) {
 599                 lthread_yield();
 600                 return 0;
 601         }
 602         return _sys_pthread_funcs.f_pthread_yield();
 603 }
 604 #else
 605 void
 606 pthread_yield(void)
 607 {
 608         if (override)
 609                 lthread_yield();
 610         else
 611                 _sys_pthread_funcs.f_pthread_yield();
 612 }
 613 #endif
 614
 615 pthread_t pthread_self(void)
 616 {
 617         if (override)
 618                 return (pthread_t) lthread_current();
 619         return _sys_pthread_funcs.f_pthread_self();
 620 }
 621
 622 int pthread_setspecific(pthread_key_t key, const void *data)
 623 {
 624         if (override) {
 625                 int rv =  lthread_setspecific((unsigned int)key, data);
 626                 return rv;
 627         }
 628         return _sys_pthread_funcs.f_pthread_setspecific(key, data);
 629 }
 630
 631 int pthread_spin_init(pthread_spinlock_t *a, int b)
 632 {
 633         NOT_IMPLEMENTED;
 634         return _sys_pthread_funcs.f_pthread_spin_init(a, b);
 635 }
 636
 637 int pthread_spin_destroy(pthread_spinlock_t *a)
 638 {
 639         NOT_IMPLEMENTED;
 640         return _sys_pthread_funcs.f_pthread_spin_destroy(a);
 641 }
 642
 643 int pthread_spin_lock(pthread_spinlock_t *a)
 644 {
 645         NOT_IMPLEMENTED;
 646         return _sys_pthread_funcs.f_pthread_spin_lock(a);
 647 }
 648
 649 int pthread_spin_trylock(pthread_spinlock_t *a)
 650 {
 651         NOT_IMPLEMENTED;
 652         return _sys_pthread_funcs.f_pthread_spin_trylock(a);
 653 }
 654
 655 int pthread_spin_unlock(pthread_spinlock_t *a)
 656 {
 657         NOT_IMPLEMENTED;
 658         return _sys_pthread_funcs.f_pthread_spin_unlock(a);
 659 }
 660
 661 int pthread_cancel(pthread_t tid)
 662 {
 663         if (override) {
 664                 lthread_cancel(*(struct lthread **)tid);
 665                 return 0;
 666         }
 667         return _sys_pthread_funcs.f_pthread_cancel(tid);
 668 }
 669
 670 int pthread_setcancelstate(int a, int *b)
 671 {
 672         NOT_IMPLEMENTED;
 673         return _sys_pthread_funcs.f_pthread_setcancelstate(a, b);
 674 }
 675
 676 int pthread_setcanceltype(int a, int *b)
 677 {
 678         NOT_IMPLEMENTED;
 679         return _sys_pthread_funcs.f_pthread_setcanceltype(a, b);
 680 }
 681
 682 void pthread_testcancel(void)
 683 {
 684         NOT_IMPLEMENTED;
 685         return _sys_pthread_funcs.f_pthread_testcancel();
 686 }
 687
 688
 689 int pthread_getschedparam(pthread_t tid, int *a, struct sched_param *b)
 690 {
 691         NOT_IMPLEMENTED;
 692         return _sys_pthread_funcs.f_pthread_getschedparam(tid, a, b);
 693 }
 694
 695 int pthread_setschedparam(pthread_t a, int b, const struct sched_param *c)
 696 {
 697         NOT_IMPLEMENTED;
 698         return _sys_pthread_funcs.f_pthread_setschedparam(a, b, c);
 699 }
 700
 701
 702 int nanosleep(const struct timespec *req, struct timespec *rem)
 703 {
 704         if (override) {
 705                 uint64_t ns = req->tv_sec * 1000000000 + req->tv_nsec;
 706
 707                 lthread_sleep(ns);
 708                 return 0;
 709         }
 710         return _sys_pthread_funcs.f_nanosleep(req, rem);
 711 }
 712
 713 int
 714 pthread_setaffinity_np(pthread_t thread, size_t cpusetsize,
 715                        const rte_cpuset_t *cpuset)
 716 {
 717         if (override) {
 718                 /* we only allow affinity with a single CPU */
 719                 if (CPU_COUNT(cpuset) != 1)
 720                         return POSIX_ERRNO(EINVAL);
 721
 722                 /* we only allow the current thread to sets its own affinity */
 723                 struct lthread *lt = (struct lthread *)thread;
 724
 725                 if (lthread_current() != lt)
 726                         return POSIX_ERRNO(EINVAL);
 727
 728                 /* determine the CPU being requested */
 729                 int i;
 730
 731                 for (i = 0; i < LTHREAD_MAX_LCORES; i++) {
 732                         if (!CPU_ISSET(i, cpuset))
 733                                 continue;
 734                         break;
 735                 }
 736                 /* check requested core is allowed */
 737                 if (i == LTHREAD_MAX_LCORES)
 738                         return POSIX_ERRNO(EINVAL);
 739
 740                 /* finally we can set affinity to the requested lcore */
 741                 lthread_set_affinity(i);
 742                 return 0;
 743         }
 744         return _sys_pthread_funcs.f_pthread_setaffinity_np(thread, cpusetsize,
 745                                                            cpuset);
 746 }