Mailing List Archive

[PATCH] Add npth socket test case
* tests/t-socket.c: New

Signed-off-by: yaowenbin1 <yaowenbin1@huawei.com>
---
tests/t-socket.c | 270 +++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 270 insertions(+)
create mode 100644 tests/t-socket.c

diff --git a/tests/t-socket.c b/tests/t-socket.c
new file mode 100644
index 0000000..8e980d4
--- /dev/null
+++ b/tests/t-socket.c
@@ -0,0 +1,270 @@
+/* t-socket.c
+ *
+ * This file is free software; as a special exception the author gives
+ * unlimited permission to copy and/or distribute it, with or without
+ * modifications, as long as this notice is preserved.
+ *
+ * This file is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY, to the extent permitted by law; without even the
+ * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ */
+
+#include "t-support.h"
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <strings.h>
+
+#define NAME "Socket"
+#define DATA "Test npth socket interfaces. . ."
+
+static void *
+thread_one (void *arg)
+{
+ int sock, msgsock, rval;
+ struct sockaddr_un server;
+ char buf[1024];
+ struct timeval tv;
+ struct timespec ts;
+
+ info_msg ("thread-one started");
+ tv.tv_sec = 0;
+ tv.tv_usec = 100;
+ ts.tv_sec = 0;
+ ts.tv_nsec = 1000;
+
+ sock = socket(AF_UNIX, SOCK_STREAM, 0);
+ if (sock < 0) {
+ perror("opening stream socket");
+ exit(1);
+ }
+ server.sun_family = AF_UNIX;
+ strcpy(server.sun_path, NAME);
+ if (bind(sock, (struct sockaddr *) &server, sizeof(struct sockaddr_un))) {
+ perror("binding stream socket");
+ exit(1);
+ }
+ info_msg("Socket has name %s\n", server.sun_path);
+ listen(sock, 5);
+ msgsock = npth_accept(sock, 0, 0);
+ if (msgsock == -1)
+ perror("accept");
+ else do {
+ bzero(buf, sizeof(buf));
+ if ((rval = npth_read(msgsock, buf, 1024)) < 0)
+ perror("reading stream message");
+ else if (rval == 0)
+ info_msg("Ending connection\n");
+ else
+ info_msg("-->%s\n", buf);
+ } while (rval > 0);
+
+ npth_select(0, NULL, NULL, NULL, &tv);
+ npth_pselect(0, NULL, NULL, NULL, &ts, NULL);
+ close(msgsock);
+ close(sock);
+ unlink(NAME);
+ info_msg ("thread-one terminated");
+
+ return (void*)4711;
+}
+
+
+static void *
+thread_two (void *arg)
+{
+
+ int sock;
+ struct sockaddr_un server;
+ char buf[1024];
+
+ info_msg ("thread-two started");
+ sock = socket(AF_UNIX, SOCK_STREAM, 0);
+ if (sock < 0) {
+ perror("opening stream socket");
+ exit(1);
+ }
+ server.sun_family = AF_UNIX;
+ strcpy(server.sun_path, NAME);
+
+ if (npth_connect(sock, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) < 0) {
+ close(sock);
+ perror("connecting stream socket");
+ exit(1);
+ }
+ if (npth_write(sock, DATA, sizeof(DATA)) < 0)
+ perror("writing on stream socket");
+ close(sock);
+
+ info_msg ("thread-two terminated");
+
+ return (void*)4722;
+}
+
+static void *
+thread_three (void *arg)
+{
+ int sock, msgsock, rval;
+ struct sockaddr_un server;
+ char buf[1024];
+ struct msghdr msg;
+ struct iovec io;
+
+ msg.msg_name = NULL;
+ io.iov_base = buf;
+ io.iov_len = 1024;
+ msg.msg_iov = &io;
+ msg.msg_iovlen = 1;
+
+ info_msg ("thread-three started");
+
+ sock = socket(AF_UNIX, SOCK_STREAM, 0);
+ if (sock < 0) {
+ perror("opening stream socket");
+ exit(1);
+ }
+ server.sun_family = AF_UNIX;
+ strcpy(server.sun_path, NAME);
+ if (bind(sock, (struct sockaddr *) &server, sizeof(struct sockaddr_un))) {
+ perror("binding stream socket");
+ exit(1);
+ }
+ info_msg("Socket has name %s\n", server.sun_path);
+ listen(sock, 5);
+ msgsock = npth_accept(sock, 0, 0);
+ if (msgsock == -1)
+ perror("accept");
+ else {
+ bzero(buf, sizeof(buf));
+ ssize_t recv_size = npth_recvmsg(msgsock, &msg, 0);
+ char * temp = msg.msg_iov[0].iov_base;
+ temp[recv_size] = '\0';
+ info_msg("get message:%s", temp);
+ };
+
+ close(msgsock);
+ close(sock);
+ unlink(NAME);
+ info_msg ("thread-three terminated");
+
+ return (void*)4711;
+}
+
+
+static void *
+thread_four (void *arg)
+{
+
+ int sock;
+ struct sockaddr_un server;
+ char buf[1024] = "test sendmsg and recvmsg\n";
+ struct msghdr msg;
+ struct iovec io;
+
+ msg.msg_name = NULL;
+ io.iov_base = buf;
+ io.iov_len = sizeof(buf);
+ msg.msg_iov = &io;
+ msg.msg_iovlen = 1;
+
+ info_msg ("thread-four started");
+ sock = socket(AF_UNIX, SOCK_STREAM, 0);
+ if (sock < 0) {
+ perror("opening stream socket");
+ exit(1);
+ }
+ server.sun_family = AF_UNIX;
+ strcpy(server.sun_path, NAME);
+
+ if (npth_connect(sock, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) < 0) {
+ close(sock);
+ perror("connecting stream socket");
+ exit(1);
+ }
+ npth_sendmsg(sock, &msg, 0);
+ close(sock);
+
+ info_msg ("thread-four terminated");
+
+ return (void*)4722;
+}
+
+int
+main (int argc, char *argv[])
+{
+ int rc;
+ npth_attr_t tattr;
+ int state;
+ npth_t tid1, tid2;
+ npth_t tid3, tid4;
+ void *retval;
+
+ if (argc >= 2 && !strcmp (argv[1], "--verbose"))
+ opt_verbose = 1;
+
+ rc = npth_init ();
+ fail_if_err (rc);
+
+ rc = npth_attr_init (&tattr);
+ fail_if_err (rc);
+ rc = npth_attr_getdetachstate (&tattr, &state);
+ fail_if_err (rc);
+ if ( state != NPTH_CREATE_JOINABLE )
+ fail_msg ("new tattr is not joinable");
+
+ info_msg ("creating thread-one");
+ rc = npth_create (&tid1, &tattr, thread_one, NULL);
+ fail_if_err (rc);
+ npth_setname_np (tid1, "thread-one");
+
+ npth_usleep(100);
+
+ info_msg ("creating thread-two");
+ rc = npth_create (&tid2, &tattr, thread_two, NULL);
+ fail_if_err (rc);
+ npth_setname_np (tid2, "thread-two");
+
+ rc = npth_attr_destroy (&tattr);
+ fail_if_err (rc);
+
+ info_msg ("waiting for thread-one to terminate");
+ rc = npth_join (tid1, &retval);
+ fail_if_err (rc);
+ if (retval != (void*)4711)
+ fail_msg ("thread-one returned an unexpected value");
+
+ info_msg ("waiting for thread-two to terminate");
+ rc = npth_join (tid2, &retval);
+ fail_if_err (rc);
+ if (retval != (void*)4722)
+ fail_msg ("thread-two returned an unexpected value");
+
+ info_msg ("creating thread-three");
+ rc = npth_create (&tid3, NULL, thread_three, NULL);
+ fail_if_err (rc);
+ npth_setname_np (tid3, "thread-three");
+
+ npth_usleep(100);
+
+ info_msg ("creating thread-four");
+ rc = npth_create (&tid4, NULL, thread_four, NULL);
+ fail_if_err (rc);
+ npth_setname_np (tid4, "thread-two");
+
+ info_msg ("waiting for thread-three to terminate");
+ rc = npth_join (tid3, &retval);
+ fail_if_err (rc);
+ if (retval != (void*)4711)
+ fail_msg ("thread-three returned an unexpected value");
+
+ info_msg ("waiting for thread-four to terminate");
+ rc = npth_join (tid4, &retval);
+ fail_if_err (rc);
+ if (retval != (void*)4722)
+ fail_msg ("thread-four returned an unexpected value");
+
+ return 0;
+}
--
2.27.0


_______________________________________________
Gnupg-users mailing list
Gnupg-users@gnupg.org
https://lists.gnupg.org/mailman/listinfo/gnupg-users