Branch data     Line data    Source code

       1                 :            : /* SPDX-License-Identifier: LGPL-2.1+ */
       2                 :            : 
       3                 :            : #include <pthread.h>
       4                 :            : 
       5                 :            : #include "process-util.h"
       6                 :            : #include "set.h"
       7                 :            : #include "tests.h"
       8                 :            : 
       9                 :            : #define NUM 100
      10                 :            : 
      11                 :          8 : static void* thread(void *p) {
      12                 :          8 :         Set **s = p;
      13                 :            : 
      14         [ -  + ]:          8 :         assert_se(s);
      15         [ -  + ]:          8 :         assert_se(*s);
      16                 :            : 
      17         [ -  + ]:          8 :         assert_se(!is_main_thread());
      18         [ -  + ]:          8 :         assert_se(set_size(*s) == NUM);
      19                 :          8 :         *s = set_free(*s);
      20                 :            : 
      21                 :          8 :         return NULL;
      22                 :            : }
      23                 :            : 
      24                 :          8 : static void test_one(const char *val) {
      25                 :            :         pthread_t t;
      26                 :          8 :         int x[NUM] = {};
      27                 :            :         unsigned i;
      28                 :            :         Set *s;
      29                 :            : 
      30         [ +  - ]:          8 :         log_info("Testing with SYSTEMD_MEMPOOL=%s", val);
      31         [ -  + ]:          8 :         assert_se(setenv("SYSTEMD_MEMPOOL", val, true) == 0);
      32         [ -  + ]:          8 :         assert_se(is_main_thread());
      33                 :            : 
      34         [ -  + ]:          8 :         assert_se(s = set_new(NULL));
      35         [ +  + ]:        808 :         for (i = 0; i < NUM; i++)
      36         [ -  + ]:        800 :                 assert_se(set_put(s, &x[i]));
      37                 :            : 
      38         [ -  + ]:          8 :         assert_se(pthread_create(&t, NULL, thread, &s) == 0);
      39         [ -  + ]:          8 :         assert_se(pthread_join(t, NULL) == 0);
      40                 :            : 
      41         [ -  + ]:          8 :         assert_se(!s);
      42                 :          8 : }
      43                 :            : 
      44                 :          4 : int main(int argc, char *argv[]) {
      45                 :          4 :         test_setup_logging(LOG_DEBUG);
      46                 :            : 
      47                 :          4 :         test_one("0");
      48                 :            :         /* The value $SYSTEMD_MEMPOOL= is cached. So the following
      49                 :            :          * test should also succeed. */
      50                 :          4 :         test_one("1");
      51                 :            : 
      52                 :          4 :         return 0;
      53                 :            : }