pthread_getaffinity.c 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. /****************************************************************************
  2. * sched/pthread/pthread_getaffinity.c
  3. *
  4. * Copyright (C) 2016, 2018 Gregory Nutt. All rights reserved.
  5. * Author: Gregory Nutt <gnutt@nuttx.org>
  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. * 1. Redistributions of source code must retain the above copyright
  12. * notice, this list of conditions and the following disclaimer.
  13. * 2. 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. * 3. Neither the name NuttX nor the names of its contributors may be
  18. * used to endorse or promote products derived from this software
  19. * 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
  24. * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  25. * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26. * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27. * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
  28. * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29. * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  31. * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  32. * POSSIBILITY OF SUCH DAMAGE.
  33. *
  34. ****************************************************************************/
  35. /****************************************************************************
  36. * Included Files
  37. ****************************************************************************/
  38. #include <nuttx/config.h>
  39. #include <sys/types.h>
  40. #include <pthread.h>
  41. #include <sched.h>
  42. #include <errno.h>
  43. #include <debug.h>
  44. #include <nuttx/sched.h>
  45. #include "pthread/pthread.h"
  46. #ifdef CONFIG_SMP
  47. /****************************************************************************
  48. * Public Functions
  49. ****************************************************************************/
  50. /****************************************************************************
  51. * Name: pthread_getschedparam
  52. *
  53. * Description:
  54. * The pthread_getaffinity_np() function returns the CPU affinity mask
  55. * of the thread thread in the buffer pointed to by cpuset.
  56. *
  57. * The argument cpusetsize is the length (in bytes) of the buffer
  58. * pointed to by cpuset. Typically, this argument would be specified as
  59. * sizeof(cpu_set_t).
  60. *
  61. * Input Parameters:
  62. * thread - The ID of thread whose affinity set will be retrieved.
  63. * cpusetsize - Size of cpuset. MUST be sizeofcpu_set_t().
  64. * cpuset - The location to return the thread's new affinity set.
  65. *
  66. * Returned Value:
  67. * 0 if successful. Otherwise, an errno value is returned indicating the
  68. * nature of the failure.
  69. *
  70. ****************************************************************************/
  71. int pthread_getaffinity_np(pthread_t thread, size_t cpusetsize,
  72. FAR cpu_set_t *cpuset)
  73. {
  74. int ret;
  75. sinfo("thread ID=%d cpusetsize=%d cpuset=%p\n",
  76. (int)thread, (int)cpusetsize, cpusetsize);
  77. DEBUGASSERT(thread > 0 && cpusetsize == sizeof(cpu_set_t) &&
  78. cpuset != NULL);
  79. /* Let nxsched_getaffinity do all of the work */
  80. ret = nxsched_getaffinity((pid_t)thread, cpusetsize, cpuset);
  81. if (ret < 0)
  82. {
  83. /* If nxsched_getaffinity() fails, return the positive errno */
  84. ret = -ret;
  85. }
  86. return ret;
  87. }
  88. #endif /* CONFIG_SMP */