^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) #include "tests.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #include "machine.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #include "thread.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) #include "debug.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) int test__thread_maps_share(struct test *test __maybe_unused, int subtest __maybe_unused)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) struct machines machines;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) struct machine *machine;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) /* thread group */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) struct thread *leader;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) struct thread *t1, *t2, *t3;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) struct maps *maps;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) /* other process */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) struct thread *other, *other_leader;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) struct maps *other_maps;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) * This test create 2 processes abstractions (struct thread)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) * with several threads and checks they properly share and
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) * maintain maps info (struct maps).
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) * thread group (pid: 0, tids: 0, 1, 2, 3)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) * other group (pid: 4, tids: 4, 5)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) machines__init(&machines);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) machine = &machines.host;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) /* create process with 4 threads */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) leader = machine__findnew_thread(machine, 0, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) t1 = machine__findnew_thread(machine, 0, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) t2 = machine__findnew_thread(machine, 0, 2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) t3 = machine__findnew_thread(machine, 0, 3);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) /* and create 1 separated process, without thread leader */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) other = machine__findnew_thread(machine, 4, 5);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) TEST_ASSERT_VAL("failed to create threads",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) leader && t1 && t2 && t3 && other);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) maps = leader->maps;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&maps->refcnt), 4);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) /* test the maps pointer is shared */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) TEST_ASSERT_VAL("maps don't match", maps == t1->maps);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) TEST_ASSERT_VAL("maps don't match", maps == t2->maps);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) TEST_ASSERT_VAL("maps don't match", maps == t3->maps);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) * Verify the other leader was created by previous call.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) * It should have shared maps with no change in
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) * refcnt.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) other_leader = machine__find_thread(machine, 4, 4);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) TEST_ASSERT_VAL("failed to find other leader", other_leader);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) * Ok, now that all the rbtree related operations were done,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) * lets remove all of them from there so that we can do the
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) * refcounting tests.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) machine__remove_thread(machine, leader);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) machine__remove_thread(machine, t1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) machine__remove_thread(machine, t2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) machine__remove_thread(machine, t3);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70) machine__remove_thread(machine, other);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) machine__remove_thread(machine, other_leader);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) other_maps = other->maps;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&other_maps->refcnt), 2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) TEST_ASSERT_VAL("maps don't match", other_maps == other_leader->maps);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) /* release thread group */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) thread__put(leader);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&maps->refcnt), 3);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) thread__put(t1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&maps->refcnt), 2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) thread__put(t2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&maps->refcnt), 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88) thread__put(t3);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) /* release other group */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91) thread__put(other_leader);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) TEST_ASSERT_EQUAL("wrong refcnt", refcount_read(&other_maps->refcnt), 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) thread__put(other);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96) machines__exit(&machines);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) }