You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@nuttx.apache.org by xi...@apache.org on 2021/03/05 15:51:58 UTC
[incubator-nuttx-apps] branch master updated: rpmsgsocket: add
rpmsgsocket test case
This is an automated email from the ASF dual-hosted git repository.
xiaoxiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-nuttx-apps.git
The following commit(s) were added to refs/heads/master by this push:
new d4259ac rpmsgsocket: add rpmsgsocket test case
d4259ac is described below
commit d4259acc156bc193b0581e3ce332d223537954bb
Author: ligd <li...@xiaomi.com>
AuthorDate: Wed Nov 18 20:40:58 2020 +0800
rpmsgsocket: add rpmsgsocket test case
Change-Id: Ia07b5915da2c9c96fa8137bcaecdf1b120c76853
Signed-off-by: ligd <li...@xiaomi.com>
---
examples/rpmsgsocket/Kconfig | 11 +
examples/rpmsgsocket/Make.defs | 23 ++
examples/rpmsgsocket/Makefile | 30 +++
examples/rpmsgsocket/rpsock_client.c | 509 +++++++++++++++++++++++++++++++++++
examples/rpmsgsocket/rpsock_server.c | 344 +++++++++++++++++++++++
5 files changed, 917 insertions(+)
diff --git a/examples/rpmsgsocket/Kconfig b/examples/rpmsgsocket/Kconfig
new file mode 100644
index 0000000..7be2c7c
--- /dev/null
+++ b/examples/rpmsgsocket/Kconfig
@@ -0,0 +1,11 @@
+#
+# For a description of the syntax of this configuration file,
+# see the file kconfig-language.txt in the NuttX tools repository.
+#
+
+config EXAMPLES_RPMSGSOCKET
+ tristate "Rpmsg scoekt example"
+ depends on NET_RPMSG
+ default n
+ ---help---
+ Enable the rpmsg socket test example
diff --git a/examples/rpmsgsocket/Make.defs b/examples/rpmsgsocket/Make.defs
new file mode 100644
index 0000000..9246239
--- /dev/null
+++ b/examples/rpmsgsocket/Make.defs
@@ -0,0 +1,23 @@
+############################################################################
+# apps/examples/rpmsgsocket/Make.defs
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership. The
+# ASF licenses this file to you 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.
+#
+############################################################################
+
+ifneq ($(CONFIG_EXAMPLES_RPMSGSOCKET),)
+CONFIGURED_APPS += $(APPDIR)/examples/rpmsgsocket
+endif
diff --git a/examples/rpmsgsocket/Makefile b/examples/rpmsgsocket/Makefile
new file mode 100644
index 0000000..acfc183
--- /dev/null
+++ b/examples/rpmsgsocket/Makefile
@@ -0,0 +1,30 @@
+############################################################################
+# apps/examples/rpmsgsocket/Makefile
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership. The
+# ASF licenses this file to you 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.
+#
+############################################################################
+
+include $(APPDIR)/Make.defs
+
+MAINSRC = rpsock_client.c rpsock_server.c
+
+PROGNAME = rpsock_client rpsock_server
+PRIORITY = SCHED_PRIORITY_DEFAULT
+STACKSIZE = $(CONFIG_DEFAULT_TASK_STACKSIZE)
+MODULE = $(CONFIG_EXAMPLES_RPMSGSOCKET)
+
+include $(APPDIR)/Application.mk
diff --git a/examples/rpmsgsocket/rpsock_client.c b/examples/rpmsgsocket/rpsock_client.c
new file mode 100644
index 0000000..a4af160
--- /dev/null
+++ b/examples/rpmsgsocket/rpsock_client.c
@@ -0,0 +1,509 @@
+/****************************************************************************
+ * apps/examples/rpmsgsocket/rpsock_client.c
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership. The
+ * ASF licenses this file to you 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.
+ *
+ ****************************************************************************/
+
+/****************************************************************************
+ * Included Files
+ ****************************************************************************/
+
+#include <nuttx/config.h>
+
+#include <sys/socket.h>
+#include <netpacket/rpmsg.h>
+
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <poll.h>
+#include <errno.h>
+
+/****************************************************************************
+ * Pre-processor Definitions
+ ****************************************************************************/
+
+#define ALIGN_UP(a) (((a) + 0x3) & ~0x3)
+#define SYNCSIZE CONFIG_NET_RPMSG_RXBUF_SIZE
+#define BUFSIZE SYNCSIZE * 2
+#define BUFHEAD 64
+
+/****************************************************************************
+ * Private types
+ ****************************************************************************/
+
+struct rpsock_arg_s
+{
+ int fd;
+ char *inbuf;
+ char *outbuf;
+ bool nonblock;
+ int bufsize;
+ int check;
+};
+
+/****************************************************************************
+ * Public Functions
+ ****************************************************************************/
+
+static void *rpsock_send_thread(pthread_addr_t pvarg)
+{
+ struct rpsock_arg_s *args = pvarg;
+ int bufsize = args->bufsize;
+ char *buf = args->outbuf;
+ int fd = args->fd;
+ int total = 0;
+ ssize_t ret;
+ int cnt = 0;
+
+ while (cnt < 2000)
+ {
+ volatile uint32_t *intp;
+ struct pollfd pfd;
+ char *tmp;
+ int snd;
+ int i;
+
+ intp = (uint32_t *)buf;
+ for (i = 0; i < bufsize / sizeof(uint32_t); i++)
+ {
+ intp[i] = cnt * bufsize / sizeof(uint32_t) + i;
+ }
+
+ tmp = buf;
+ snd = bufsize;
+ while (snd > 0)
+ {
+ if (args->nonblock)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = fd;
+ pfd.events = POLLOUT;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("client: poll failure: %d\n", errno);
+ break;
+ }
+ }
+
+ ret = send(fd, tmp, snd, 0);
+ if (ret > 0)
+ {
+ printf("client send data, cnt %d, total %d\n",
+ cnt, cnt * bufsize);
+ total += ret;
+ tmp += ret;
+ snd -= ret;
+ }
+ else if (ret < 0 && errno != EAGAIN)
+ {
+ printf("client send data failed errno %d\n", errno);
+ }
+ else if (args->nonblock)
+ {
+ usleep(10);
+ }
+ }
+
+ if (cnt && cnt % 1024 == 0)
+ {
+ sleep(3);
+ }
+
+ cnt++;
+ }
+
+ sleep(5);
+
+ snprintf(buf, 64, "endflags, send total %d", total);
+ send(fd, buf, 64, 0);
+
+ return NULL;
+}
+
+static int rpsock_unsync_test(struct rpsock_arg_s *args)
+{
+ pthread_attr_t attr;
+ pthread_t thread;
+ int total = 0;
+ int cnt = 0;
+ int ret;
+
+ pthread_attr_init(&attr);
+ pthread_attr_setstacksize(&attr, 10 * 1024);
+ ret = pthread_create(&thread, &attr, rpsock_send_thread,
+ (pthread_addr_t)args);
+ if (ret < 0)
+ {
+ return ret;
+ }
+
+ pthread_detach(thread);
+
+ while (1)
+ {
+ struct pollfd pfd;
+
+ if (args->nonblock)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = args->fd;
+ pfd.events = POLLIN;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("client: poll failure: %d\n", errno);
+ break;
+ }
+ }
+
+ ret = recv(args->fd, args->inbuf, args->bufsize, 0);
+ if (ret > 0)
+ {
+ uint32_t *intp;
+ int checks;
+ int i;
+
+ if (strncmp(args->inbuf, "endflags,", 9) == 0)
+ {
+ printf("client recv done, total %d, %s\n", total, args->inbuf);
+ break;
+ }
+
+ printf("client recv data, act len %d, total %d\n", ret, total);
+
+ if (args->check && ret > 4)
+ {
+ intp = (uint32_t *)(args->inbuf + (ALIGN_UP(total) - total));
+ checks = ret - (ALIGN_UP(total) - total);
+
+ for (i = 0; i < checks / sizeof(uint32_t); i++)
+ {
+ if (intp[i] != ALIGN_UP(total) / sizeof(uint32_t) + i)
+ {
+ printf("client check fail total %d, \
+ i %d, %08x, %08x\n",
+ ALIGN_UP(total), i, intp[i],
+ ALIGN_UP(total) / sizeof(uint32_t) + i);
+ }
+ }
+ }
+
+ total += ret;
+ }
+ else
+ {
+ printf("client recv data failed errno %d\n", errno);
+ }
+
+ if (cnt++ > 1000)
+ {
+ sleep(2);
+ cnt = 0;
+ }
+ }
+
+ return 0;
+}
+
+static int rpsock_stream_client(int argc, char *argv[])
+{
+ struct sockaddr_rpmsg myaddr;
+ struct rpsock_arg_s args;
+ bool nonblock = false;
+ int cnt = 0;
+ char *outbuf;
+ char *inbuf;
+ int sockfd;
+ int ret;
+
+ /* Allocate buffers */
+
+ outbuf = malloc(BUFSIZE);
+ inbuf = malloc(BUFSIZE);
+ if (!outbuf || !inbuf)
+ {
+ printf("client: failed to allocate buffers\n");
+ ret = -ENOMEM;
+ goto errout_with_buffers;
+ }
+
+ /* Create a new rpmsg domain socket */
+
+ if (strcmp(argv[2], "nonblock") == 0)
+ {
+ nonblock = true;
+ }
+
+ printf("client: create socket SOCK_STREAM nonblock %d\n", nonblock);
+
+ if (nonblock)
+ {
+ sockfd = socket(PF_RPMSG, SOCK_STREAM | SOCK_NONBLOCK, 0);
+ }
+ else
+ {
+ sockfd = socket(PF_RPMSG, SOCK_STREAM, 0);
+ }
+
+ if (sockfd < 0)
+ {
+ printf("client socket failure %d\n", errno);
+ goto errout_with_buffers;
+ }
+
+ /* Connect the socket to the server */
+
+ myaddr.rp_family = AF_RPMSG;
+ strncpy(myaddr.rp_name, argv[3], RPMSG_SOCKET_NAME_SIZE);
+ strncpy(myaddr.rp_cpu, argv[4], RPMSG_SOCKET_CPU_SIZE);
+
+ printf("client: Connecting to %s,%s...\n", myaddr.rp_cpu, myaddr.rp_name);
+ ret = connect(sockfd, (struct sockaddr *)&myaddr, sizeof(myaddr));
+ if (ret < 0)
+ {
+ printf("client: connect failure: %d\n", errno);
+ goto errout_with_socket;
+ }
+
+ printf("client: Connected\n");
+
+ while (1)
+ {
+ size_t sendsize = BUFHEAD + cnt * (random() % 64);
+ size_t recvsize = 0;
+ ssize_t act;
+ char *tmp;
+ int snd;
+ int *ptr;
+ int i;
+
+ if (sendsize > SYNCSIZE)
+ {
+ break;
+ }
+
+ snprintf(outbuf, BUFHEAD, "process%04d, msg%04d, name:%s",
+ getpid(), cnt, argv[3]);
+
+ ptr = (int *)(outbuf + BUFHEAD);
+ for (i = 0; i < (sendsize - BUFHEAD) / 4; i++)
+ {
+ ptr[i] = cnt * 100 + i;
+ }
+
+ printf("client send data, cnt %d, total len %d, BUFHEAD %s\n",
+ cnt, sendsize, outbuf);
+
+ tmp = outbuf;
+ snd = sendsize;
+ while (snd > 0)
+ {
+ ret = send(sockfd, tmp, snd, 0);
+ if (ret < 0)
+ {
+ continue;
+ }
+
+ tmp += ret;
+ snd -= ret;
+ }
+
+ tmp = inbuf;
+ while (1)
+ {
+ struct pollfd pfd;
+
+ if (nonblock)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = sockfd;
+ pfd.events = POLLIN;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("client: poll failure: %d\n", errno);
+ break;
+ }
+ }
+
+ act = recv(sockfd, tmp, SYNCSIZE, 0);
+ if (act == -EAGAIN)
+ {
+ continue;
+ }
+ else if (act < 0)
+ {
+ printf("client recv data failed %d\n", act);
+ break;
+ }
+
+ if (recvsize == 0)
+ {
+ printf("client recv data %s\n", tmp);
+ }
+
+ recvsize += act;
+ tmp += act;
+
+ if (recvsize >= sendsize)
+ {
+ printf("client recv total %d, send total %d\n",
+ recvsize, sendsize);
+ break;
+ }
+ }
+
+ ptr = (int *)(inbuf + BUFHEAD);
+ for (i = 0; i < (recvsize - BUFHEAD) / 4; i++)
+ {
+ if (ptr[i] != cnt * 100 + i)
+ {
+ printf("client check fail i %d, %d, %d\n", i, ptr[i], cnt + i);
+ }
+ }
+
+ cnt++;
+ }
+
+ args.fd = sockfd;
+ args.inbuf = inbuf;
+ args.outbuf = outbuf;
+ args.nonblock = nonblock;
+ args.bufsize = BUFSIZE;
+ args.check = true;
+
+ rpsock_unsync_test(&args);
+
+ printf("client: Terminating\n");
+
+errout_with_socket:
+ close(sockfd);
+
+errout_with_buffers:
+ free(outbuf);
+ free(inbuf);
+ return -errno;
+}
+
+static int rpsock_dgram_client(int argc, char *argv[])
+{
+ struct sockaddr_rpmsg myaddr;
+ struct rpsock_arg_s args;
+ bool nonblock = false;
+ char *outbuf;
+ char *inbuf;
+ int sockfd;
+ int ret;
+
+ /* Allocate buffers */
+
+ outbuf = malloc(BUFSIZE);
+ inbuf = malloc(BUFSIZE);
+ if (!outbuf || !inbuf)
+ {
+ printf("client: failed to allocate buffers\n");
+ ret = -ENOMEM;
+ goto errout_with_buffers;
+ }
+
+ /* Create a new rpmsg domain socket */
+
+ if (strcmp(argv[2], "nonblock") == 0)
+ {
+ nonblock = true;
+ }
+
+ printf("client: create socket SOCK_DGRAM nonblock %d\n", nonblock);
+
+ if (nonblock)
+ {
+ sockfd = socket(PF_RPMSG, SOCK_DGRAM | SOCK_NONBLOCK, 0);
+ }
+ else
+ {
+ sockfd = socket(PF_RPMSG, SOCK_DGRAM, 0);
+ }
+
+ if (sockfd < 0)
+ {
+ printf("client socket failure %d\n", errno);
+ goto errout_with_buffers;
+ }
+
+ /* Connect the socket to the server */
+
+ myaddr.rp_family = AF_RPMSG;
+ strncpy(myaddr.rp_name, argv[3], RPMSG_SOCKET_NAME_SIZE);
+ strncpy(myaddr.rp_cpu, argv[4], RPMSG_SOCKET_CPU_SIZE);
+
+ printf("client: Connecting to %s,%s...\n", myaddr.rp_cpu, myaddr.rp_name);
+ ret = connect(sockfd, (struct sockaddr *)&myaddr, sizeof(myaddr));
+ if (ret < 0)
+ {
+ printf("client: connect failure: %d\n", errno);
+ goto errout_with_socket;
+ }
+
+ printf("client: Connected\n");
+
+ args.fd = sockfd;
+ args.inbuf = inbuf;
+ args.outbuf = outbuf;
+ args.nonblock = nonblock;
+ args.bufsize = SYNCSIZE - 32;
+ args.check = false;
+
+ rpsock_unsync_test(&args);
+
+ printf("client: Terminating\n");
+
+errout_with_socket:
+ close(sockfd);
+
+errout_with_buffers:
+ free(outbuf);
+ free(inbuf);
+ return -errno;
+}
+
+int main(int argc, char *argv[])
+{
+ if (argc < 4)
+ {
+ printf("Usage: rpsock_client stream/dgram"
+ " block/nonblock rp_name rp_cpu\n");
+ return -EINVAL;
+ }
+
+ if (!strcmp(argv[1], "stream"))
+ {
+ return rpsock_stream_client(argc, argv);
+ }
+ else if (!strcmp(argv[1], "dgram"))
+ {
+ return rpsock_dgram_client(argc, argv);
+ }
+
+ return -EINVAL;
+}
diff --git a/examples/rpmsgsocket/rpsock_server.c b/examples/rpmsgsocket/rpsock_server.c
new file mode 100644
index 0000000..001d9db
--- /dev/null
+++ b/examples/rpmsgsocket/rpsock_server.c
@@ -0,0 +1,344 @@
+/****************************************************************************
+ * apps/examples/rpmsgsocket/rpsock_server.c
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership. The
+ * ASF licenses this file to you 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.
+ *
+ ****************************************************************************/
+
+/****************************************************************************
+ * Included Files
+ ****************************************************************************/
+
+#include <nuttx/config.h>
+
+#include <sys/socket.h>
+#include <netpacket/rpmsg.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <poll.h>
+#include <errno.h>
+
+/****************************************************************************
+ * Private types
+ ****************************************************************************/
+
+struct rpsock_arg_s
+{
+ int fd;
+ bool nonblock;
+ bool skippoll;
+};
+
+/****************************************************************************
+ * Public Functions
+ ****************************************************************************/
+
+static void *rpsock_thread(pthread_addr_t pvarg)
+{
+ struct rpsock_arg_s *args = pvarg;
+ struct pollfd pfd;
+ char buf[255];
+ ssize_t ret;
+
+ while (1)
+ {
+ char *tmp;
+ int snd;
+
+ if (args->nonblock && !args->skippoll)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = args->fd;
+ pfd.events = POLLIN;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("server poll failed errno %d\n", errno);
+ break;
+ }
+ }
+
+ args->skippoll = false;
+
+ ret = recv(args->fd, buf, sizeof(buf), 0);
+ if (ret == 0 || (ret < 0 && errno == ECONNRESET))
+ {
+ printf("server recv data normal exit\n");
+ break;
+ }
+ else if (ret < 0 && errno == EAGAIN)
+ {
+ usleep(10);
+ continue;
+ }
+ else if (ret < 0)
+ {
+ printf("server recv data failed ret %d, errno %d\n", ret, errno);
+ break;
+ }
+
+ snd = ret;
+ tmp = buf;
+ while (snd > 0)
+ {
+ if (args->nonblock)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = args->fd;
+ pfd.events = POLLOUT;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("server: poll failure: %d\n", errno);
+ break;
+ }
+ }
+
+ ret = send(args->fd, tmp, snd, 0);
+ if (ret > 0)
+ {
+ tmp += ret;
+ snd -= ret;
+ }
+ else if (ret == 0)
+ {
+ printf("server send data normal exit\n");
+ break;
+ }
+ else if (errno == EAGAIN)
+ {
+ usleep(10);
+ }
+ else
+ {
+ printf("server send data failed errno %d\n", errno);
+ break;
+ }
+ }
+ }
+
+ printf("server Complete ret %d, errno %d\n", ret, errno);
+ free(args);
+ return NULL;
+}
+
+static int rpsock_stream_server(int argc, char *argv[])
+{
+ struct sockaddr_rpmsg myaddr;
+ bool nonblock = false;
+ socklen_t addrlen;
+ int listensd;
+ int ret;
+
+ /* Create a new rpmsg domain socket */
+
+ if (strcmp(argv[2], "nonblock") == 0)
+ {
+ nonblock = true;
+ }
+
+ printf("server: create socket SOCK_STREAM nonblock %d\n", nonblock);
+
+ if (nonblock)
+ {
+ listensd = socket(PF_RPMSG, SOCK_STREAM | SOCK_NONBLOCK, 0);
+ }
+ else
+ {
+ listensd = socket(PF_RPMSG, SOCK_STREAM, 0);
+ }
+
+ if (listensd < 0)
+ {
+ printf("server: socket failure: %d\n", errno);
+ return -1;
+ }
+
+ /* Bind the socket to a local address */
+
+ myaddr.rp_family = AF_RPMSG;
+ strncpy(myaddr.rp_name, argv[3], RPMSG_SOCKET_NAME_SIZE);
+ if (argc == 5)
+ {
+ strncpy(myaddr.rp_cpu, argv[4], RPMSG_SOCKET_CPU_SIZE);
+ }
+ else
+ {
+ myaddr.rp_cpu[0] = '\0';
+ }
+
+ printf("server: bind cpu %s, name %s ...\n",
+ myaddr.rp_cpu, myaddr.rp_name);
+ ret = bind(listensd, (struct sockaddr *)&myaddr, sizeof(myaddr));
+ if (ret < 0)
+ {
+ printf("server: bind failure: %d\n", errno);
+ goto errout_with_listensd;
+ }
+
+ /* Listen for connections on the bound socket */
+
+ printf("server: listen ...\n");
+ ret = listen(listensd, 16);
+ if (ret < 0)
+ {
+ printf("server: listen failure %d\n", errno);
+ goto errout_with_listensd;
+ }
+
+ while (1)
+ {
+ struct rpsock_arg_s *args;
+ pthread_t thread;
+ struct pollfd pfd;
+ int new;
+
+ if (nonblock)
+ {
+ memset(&pfd, 0, sizeof(struct pollfd));
+ pfd.fd = listensd;
+ pfd.events = POLLIN;
+
+ ret = poll(&pfd, 1, -1);
+ if (ret < 0)
+ {
+ printf("server: poll failure: %d\n", errno);
+ goto errout_with_listensd;
+ }
+ }
+
+ printf("server: try accept ...\n");
+ new = accept(listensd, (struct sockaddr *)&myaddr, &addrlen);
+ if (new < 0)
+ break;
+
+ printf("server: Connection accepted -- %d\n", new);
+
+ args = malloc(sizeof(struct rpsock_arg_s));
+ assert(args);
+
+ args->fd = new;
+ args->nonblock = nonblock;
+ args->skippoll = false;
+
+ pthread_create(&thread, NULL, rpsock_thread,
+ (pthread_addr_t)args);
+ pthread_detach(thread);
+ }
+
+ printf("server: Terminating\n");
+ close(listensd);
+ return 0;
+
+errout_with_listensd:
+ close(listensd);
+
+ return -errno;
+}
+
+static int rpsock_dgram_server(int argc, char *argv[])
+{
+ struct sockaddr_rpmsg myaddr;
+ struct rpsock_arg_s *args;
+ bool nonblock = false;
+ int fd;
+ int ret;
+
+ /* Create a new rpmsg domain socket */
+
+ if (strcmp(argv[2], "nonblock") == 0)
+ {
+ nonblock = true;
+ }
+
+ printf("server: create socket SOCK_STREAM nonblock %d\n", nonblock);
+
+ if (nonblock)
+ {
+ fd = socket(PF_RPMSG, SOCK_DGRAM | SOCK_NONBLOCK, 0);
+ }
+ else
+ {
+ fd = socket(PF_RPMSG, SOCK_DGRAM, 0);
+ }
+
+ if (fd < 0)
+ {
+ printf("server: socket failure: %d\n", errno);
+ return -errno;
+ }
+
+ /* Bind the socket to a local address */
+
+ myaddr.rp_family = AF_RPMSG;
+ strncpy(myaddr.rp_name, argv[3], RPMSG_SOCKET_NAME_SIZE);
+ if (argc == 5)
+ {
+ strncpy(myaddr.rp_cpu, argv[4], RPMSG_SOCKET_CPU_SIZE);
+ }
+ else
+ {
+ myaddr.rp_cpu[0] = '\0';
+ }
+
+ printf("server: bind cpu %s, name %s ...\n",
+ myaddr.rp_cpu, myaddr.rp_name);
+ ret = bind(fd, (struct sockaddr *)&myaddr, sizeof(myaddr));
+ if (ret < 0)
+ {
+ printf("server: bind failure: %d\n", errno);
+ close(fd);
+ return ret;
+ }
+
+ args = malloc(sizeof(struct rpsock_arg_s));
+ assert(args);
+
+ args->fd = fd;
+ args->nonblock = nonblock;
+ args->skippoll = true;
+
+ rpsock_thread(args);
+
+ close(fd);
+ return 0;
+}
+
+int main(int argc, char *argv[])
+{
+ if (argc != 4 && argc != 5)
+ {
+ printf("Usage: rpsock_server stream/dgram"
+ " block/nonblock rp_name [rp_cpu]\n");
+ return -EINVAL;
+ }
+
+ if (!strcmp(argv[1], "stream"))
+ {
+ return rpsock_stream_server(argc, argv);
+ }
+ else if (!strcmp(argv[1], "dgram"))
+ {
+ return rpsock_dgram_server(argc, argv);
+ }
+
+ return -EINVAL;
+}