| Index: third_party/grpc/test/core/support/thd_test.c
|
| diff --git a/third_party/grpc/test/core/support/thd_test.c b/third_party/grpc/test/core/support/thd_test.c
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f7807d280a2113d5a47a590ec6632e186508413e
|
| --- /dev/null
|
| +++ b/third_party/grpc/test/core/support/thd_test.c
|
| @@ -0,0 +1,115 @@
|
| +/*
|
| + *
|
| + * Copyright 2015, Google Inc.
|
| + * All rights reserved.
|
| + *
|
| + * Redistribution and use in source and binary forms, with or without
|
| + * modification, are permitted provided that the following conditions are
|
| + * met:
|
| + *
|
| + * * Redistributions of source code must retain the above copyright
|
| + * notice, this list of conditions and the following disclaimer.
|
| + * * Redistributions in binary form must reproduce the above
|
| + * copyright notice, this list of conditions and the following disclaimer
|
| + * in the documentation and/or other materials provided with the
|
| + * distribution.
|
| + * * Neither the name of Google Inc. nor the names of its
|
| + * contributors may be used to endorse or promote products derived from
|
| + * this software without specific prior written permission.
|
| + *
|
| + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| + *
|
| + */
|
| +
|
| +/* Test of gpr thread support. */
|
| +
|
| +#include <stdio.h>
|
| +#include <stdlib.h>
|
| +#include <grpc/support/log.h>
|
| +#include <grpc/support/sync.h>
|
| +#include <grpc/support/thd.h>
|
| +#include <grpc/support/time.h>
|
| +#include "test/core/util/test_config.h"
|
| +
|
| +struct test {
|
| + gpr_mu mu;
|
| + int n;
|
| + int is_done;
|
| + gpr_cv done_cv;
|
| +};
|
| +
|
| +/* A Thread body. Decrement t->n, and if is becomes zero, set t->done. */
|
| +static void thd_body(void *v) {
|
| + struct test *t = v;
|
| + gpr_mu_lock(&t->mu);
|
| + t->n--;
|
| + if (t->n == 0) {
|
| + t->is_done = 1;
|
| + gpr_cv_signal(&t->done_cv);
|
| + }
|
| + gpr_mu_unlock(&t->mu);
|
| +}
|
| +
|
| +static void thd_body_joinable(void *v) {}
|
| +
|
| +/* Test thread options work as expected */
|
| +static void test_options(void) {
|
| + gpr_thd_options options = gpr_thd_options_default();
|
| + GPR_ASSERT(!gpr_thd_options_is_joinable(&options));
|
| + GPR_ASSERT(gpr_thd_options_is_detached(&options));
|
| + gpr_thd_options_set_joinable(&options);
|
| + GPR_ASSERT(gpr_thd_options_is_joinable(&options));
|
| + GPR_ASSERT(!gpr_thd_options_is_detached(&options));
|
| + gpr_thd_options_set_detached(&options);
|
| + GPR_ASSERT(!gpr_thd_options_is_joinable(&options));
|
| + GPR_ASSERT(gpr_thd_options_is_detached(&options));
|
| +}
|
| +
|
| +/* Test that we can create a number of threads and wait for them. */
|
| +static void test(void) {
|
| + int i;
|
| + gpr_thd_id thd;
|
| + gpr_thd_id thds[1000];
|
| + struct test t;
|
| + int n = 1000;
|
| + gpr_thd_options options = gpr_thd_options_default();
|
| + gpr_mu_init(&t.mu);
|
| + gpr_cv_init(&t.done_cv);
|
| + t.n = n;
|
| + t.is_done = 0;
|
| + for (i = 0; i != n; i++) {
|
| + GPR_ASSERT(gpr_thd_new(&thd, &thd_body, &t, NULL));
|
| + }
|
| + gpr_mu_lock(&t.mu);
|
| + while (!t.is_done) {
|
| + gpr_cv_wait(&t.done_cv, &t.mu, gpr_inf_future(GPR_CLOCK_REALTIME));
|
| + }
|
| + gpr_mu_unlock(&t.mu);
|
| + GPR_ASSERT(t.n == 0);
|
| + gpr_thd_options_set_joinable(&options);
|
| + for (i = 0; i < n; i++) {
|
| + GPR_ASSERT(gpr_thd_new(&thds[i], &thd_body_joinable, NULL, &options));
|
| + }
|
| + for (i = 0; i < n; i++) {
|
| + gpr_thd_join(thds[i]);
|
| + }
|
| +}
|
| +
|
| +/* ------------------------------------------------- */
|
| +
|
| +int main(int argc, char *argv[]) {
|
| + grpc_test_init(argc, argv);
|
| + test_options();
|
| + test();
|
| + return 0;
|
| +}
|
|
|