init

Telegram/ThirdParty/dispatch/tests/dispatch_context_for_key.c

@@ -0,0 +1,158 @@
+/*
+ * Copyright (c) 2010-2011 Apple Inc. All rights reserved.
+ *
+ * @APPLE_APACHE_LICENSE_HEADER_START@
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * @APPLE_APACHE_LICENSE_HEADER_END@
+ */
+
+#include <dispatch/dispatch.h>
+#include <stdlib.h>
+#include <stdio.h>
+#if defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
+#include <unistd.h>
+#endif
+#include <assert.h>
+
+#include <bsdtests.h>
+#include "dispatch_test.h"
+
+#if DISPATCH_API_VERSION >= 20100825 && DISPATCH_API_VERSION != 20101110
+
+static const char *ctxts[] = {"ctxt for app", "ctxt for key 1",
+		"ctxt for key 2", "ctxt for key 1 bis", "ctxt for key 4"};
+volatile long ctxts_destroyed;
+static dispatch_group_t g;
+
+static void
+destructor(void *ctxt)
+{
+	fprintf(stderr, "destructor of %s\n", (char*)ctxt);
+	(void)__sync_add_and_fetch(&ctxts_destroyed, 1);
+	dispatch_group_leave(g);
+}
+
+static void
+test_context_for_key(void)
+{
+	g = dispatch_group_create();
+	dispatch_queue_t q = dispatch_queue_create("q", NULL);
+#if DISPATCH_API_VERSION >= 20100518
+	dispatch_queue_t tq = dispatch_queue_create("tq", DISPATCH_QUEUE_CONCURRENT);
+#else
+	dispatch_queue_t tq = dispatch_queue_create("tq", NULL);
+	dispatch_queue_set_width(tq, LONG_MAX);
+#endif
+	dispatch_queue_t ttq = dispatch_get_global_queue(0, 0);
+	dispatch_group_enter(g);
+#if DISPATCH_API_VERSION >= 20101011
+	dispatch_queue_set_specific(tq, &ctxts[4], (char *)ctxts[4], destructor);
+#else
+	dispatch_set_context_for_key(tq, &ctxts[4], ctxts[4], ttq, destructor);
+#endif
+	dispatch_set_target_queue(tq, ttq);
+	dispatch_group_enter(g);
+	dispatch_set_context(q, (char *)ctxts[0]);
+	dispatch_set_target_queue(q, tq);
+	dispatch_set_finalizer_f(q, destructor);
+
+	dispatch_async(q, ^{
+		dispatch_group_enter(g);
+#if DISPATCH_API_VERSION >= 20101011
+		dispatch_queue_set_specific(q, &ctxts[1], (char *)ctxts[1], destructor);
+#else
+		dispatch_set_context_for_key(q, &ctxts[1], ctxts[1], ttq, destructor);
+#endif
+	});
+	dispatch_retain(q);
+	dispatch_async(dispatch_get_global_queue(0, 0), ^{
+		dispatch_group_enter(g);
+#if DISPATCH_API_VERSION >= 20101011
+		dispatch_queue_set_specific(q, &ctxts[2], (char *)ctxts[2], destructor);
+#else
+		dispatch_set_context_for_key(q, &ctxts[2], ctxts[2], ttq, destructor);
+#endif
+		dispatch_async(dispatch_get_global_queue(0, 0), ^{
+			void *ctxt;
+#if DISPATCH_API_VERSION >= 20101011
+			ctxt = dispatch_queue_get_specific(q, &ctxts[2]);
+#else
+			ctxt = dispatch_get_context_for_key(q, &ctxts[2]);
+#endif
+			test_ptr("get context for key 2", ctxt, ctxts[2]);
+			dispatch_release(q);
+		});
+	});
+	dispatch_async(q, ^{
+		void *ctxt;
+#if DISPATCH_API_VERSION >= 20101011
+		ctxt = dispatch_get_specific(&ctxts[1]);
+		test_ptr("get current context for key 1", ctxt, ctxts[1]);
+		ctxt = dispatch_get_specific(&ctxts[4]);
+		test_ptr("get current context for key 4 (on target queue)", ctxt, ctxts[4]);
+		ctxt = dispatch_queue_get_specific(q, &ctxts[1]);
+#else
+		ctxt = dispatch_get_context_for_key(tq, &ctxts[4]);
+		test_ptr("get context for key 4 (on target queue)", ctxt, ctxts[4]);
+		ctxt = dispatch_get_context_for_key(q, &ctxts[1]);
+#endif
+		test_ptr("get context for key 1", ctxt, ctxts[1]);
+	});
+	dispatch_async(q, ^{
+		dispatch_group_enter(g);
+		void *ctxt;
+#if DISPATCH_API_VERSION >= 20101011
+		dispatch_queue_set_specific(q, &ctxts[1], (char *)ctxts[3], destructor);
+		ctxt = dispatch_queue_get_specific(q, &ctxts[1]);
+#else
+		dispatch_set_context_for_key(q, &ctxts[1], ctxts[3], ttq, destructor);
+		ctxt = dispatch_get_context_for_key(q, &ctxts[1]);
+#endif
+		test_ptr("get context for key 1", ctxt, ctxts[3]);
+	});
+	dispatch_async(q, ^{
+		void *ctxt;
+#if DISPATCH_API_VERSION >= 20101011
+		dispatch_queue_set_specific(q, &ctxts[1], NULL, destructor);
+		ctxt = dispatch_queue_get_specific(q, &ctxts[1]);
+#else
+		dispatch_set_context_for_key(q, &ctxts[1], NULL, ttq, destructor);
+		ctxt = dispatch_get_context_for_key(q, &ctxts[1]);
+#endif
+		test_ptr("get context for key 1", ctxt, NULL);
+	});
+	void *ctxt = dispatch_get_context(q);
+	test_ptr("get context for app", ctxt, ctxts[0]);
+	dispatch_release(tq);
+	dispatch_release(q);
+	dispatch_group_wait(g, DISPATCH_TIME_FOREVER);
+	test_long("contexts destroyed", ctxts_destroyed, 5);
+	dispatch_release(g);
+}
+#endif
+
+int
+main(void)
+{
+	dispatch_test_start("Dispatch Queue Specific"); // rdar://problem/8429188
+
+	dispatch_async(dispatch_get_main_queue(), ^{
+#if DISPATCH_API_VERSION >= 20100825 && DISPATCH_API_VERSION != 20101110
+		test_context_for_key();
+#endif
+		test_stop();
+	});
+	dispatch_main();
+}