Skip to content

Commit 522c669

Browse files
committed
chore: add nullptr gaurds
1 parent c94712f commit 522c669

File tree

3 files changed

+13
-6
lines changed

3 files changed

+13
-6
lines changed

libc/src/sched/linux/sched_getcpuisset.cpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,9 @@
88

99
#include "src/sched/sched_getcpuisset.h"
1010

11-
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12-
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
11+
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12+
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
13+
#include "src/__support/macros/null_check.h" // LIBC_CRASH_ON_NULLPTR
1314

1415
#include "hdr/sched_macros.h" // NCPUBITS
1516
#include "hdr/types/cpu_set_t.h"
@@ -19,6 +20,8 @@ namespace LIBC_NAMESPACE_DECL {
1920

2021
LLVM_LIBC_FUNCTION(int, __sched_getcpuisset,
2122
(int cpu, const size_t cpuset_size, cpu_set_t *set)) {
23+
LIBC_CRASH_ON_NULLPTR(set);
24+
2225
if (static_cast<size_t>(cpu) / 8 < cpuset_size) {
2326
const size_t element_index = static_cast<size_t>(cpu) / NCPUBITS;
2427
const size_t bit_position = static_cast<size_t>(cpu) % NCPUBITS;

libc/src/sched/linux/sched_setcpuset.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,9 @@
88

99
#include "src/sched/sched_setcpuset.h"
1010

11-
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12-
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
11+
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12+
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
13+
#include "src/__support/macros/null_check.h" // LIBC_CRASH_ON_NULLPTR
1314

1415
#include "hdr/sched_macros.h" // NCPUBITS
1516
#include "hdr/types/cpu_set_t.h"
@@ -19,6 +20,7 @@ namespace LIBC_NAMESPACE_DECL {
1920

2021
LLVM_LIBC_FUNCTION(void, __sched_setcpuset,
2122
(int cpu, const size_t cpuset_size, cpu_set_t *set)) {
23+
LIBC_CRASH_ON_NULLPTR(set);
2224
if (static_cast<size_t>(cpu) / 8 < cpuset_size) {
2325
const size_t element_index = static_cast<size_t>(cpu) / NCPUBITS;
2426
const size_t bit_position = static_cast<size_t>(cpu) % NCPUBITS;

libc/src/sched/linux/sched_setcpuzero.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,9 @@
88

99
#include "src/sched/sched_setcpuzero.h"
1010

11-
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12-
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
11+
#include "src/__support/common.h" // LLVM_LIBC_FUNCTION
12+
#include "src/__support/macros/config.h" // LIBC_NAMESPACE_DECL
13+
#include "src/__support/macros/null_check.h" // LIBC_CRASH_ON_NULLPTR
1314

1415
#include "hdr/types/cpu_set_t.h"
1516
#include "hdr/types/size_t.h"
@@ -18,6 +19,7 @@ namespace LIBC_NAMESPACE_DECL {
1819

1920
LLVM_LIBC_FUNCTION(void, __sched_setcpuzero,
2021
(const size_t cpuset_size, cpu_set_t *set)) {
22+
LIBC_CRASH_ON_NULLPTR(set);
2123
__builtin_memset(set, 0, cpuset_size);
2224
}
2325

0 commit comments

Comments
 (0)