Tests: ping_rpmsg: Update Test with Proper Socket Usage
[ipc/ipcdev.git] / linux / src / tests / ping_rpmsg.c
1 /*
2  * Copyright (c) 2012-2014, Texas Instruments Incorporated
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * *  Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * *  Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * *  Neither the name of Texas Instruments Incorporated nor the names of
17  *    its contributors may be used to endorse or promote products derived
18  *    from this software without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
22  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
24  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
27  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
28  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
29  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
30  * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  */
32 /*
33  *  ======== ping_rpmsg.c ========
34  *
35  *  Works with the ping_rpmsg BIOS sample over the rpmsg-proto socket.
36  */
38 #include <sys/select.h>
39 #include <sys/time.h>
40 #include <sys/types.h>
41 #include <sys/socket.h>
42 #include <errno.h>
43 #include <stdio.h>
44 #include <string.h>
45 #include <stdlib.h>
46 #include <unistd.h>
47 #include <time.h>
49 /* Ipc Socket Protocol Family */
50 #include <net/rpmsg.h>
52 #define NUM_LOOPS_DFLT 100
53 #define CORE_ID_DFLT   0 /* CORE ID should be (MultiProc ID - 1) */
55 long diff(struct timespec start, struct timespec end)
56 {
57     struct timespec temp;
59     if ((end.tv_nsec - start.tv_nsec) < 0) {
60         temp.tv_sec = end.tv_sec - start.tv_sec-1;
61         temp.tv_nsec = 1000000000UL + end.tv_nsec - start.tv_nsec;
62     } else {
63         temp.tv_sec = end.tv_sec - start.tv_sec;
64         temp.tv_nsec = end.tv_nsec - start.tv_nsec;
65     }
67     return (temp.tv_sec * 1000000UL + temp.tv_nsec / 1000);
68 }
70 int main (int argc, char ** argv)
71 {
72     unsigned int numLoops = NUM_LOOPS_DFLT;
73     short coreId = CORE_ID_DFLT;
74     int send_sock, err, recv_sock;
75     struct sockaddr_rpmsg src_addr, dst_addr;
76     socklen_t len;
77     const char *msg = "Ping!";
78     char buf[512];
79     struct timespec   start,end;
80     long              elapsed=0,delta;
81     int i;
82     fd_set rfds;
84     /* Parse Args: */
85     switch (argc) {
86         case 1:
87            /* use defaults */
88            break;
89         case 2:
90            numLoops = atoi(argv[1]);
91            break;
92         case 3:
93            numLoops   = atoi(argv[1]);
94            coreId     = atoi(argv[2]);
95            break;
96         default:
97            printf("Usage: %s [<numLoops>] [<CoreId>]\n", argv[0]);
98            printf("\tDefaults: numLoops: %d; CoreId: %d\n",
99                    NUM_LOOPS_DFLT, CORE_ID_DFLT);
100            exit(0);
101     }
103     /* create an RPMSG socket */
104     send_sock = socket(AF_RPMSG, SOCK_SEQPACKET, 0);
105     if (send_sock < 0) {
106         printf("socket failed for send_sock: %s (%d)\n", strerror(errno), errno);
107         return -1;
108     }
110     /* connect to remote service */
111     memset(&dst_addr, 0, sizeof(dst_addr));
112     dst_addr.family = AF_RPMSG;
113     dst_addr.vproc_id = coreId;
114     dst_addr.addr = 51; // use 51 for ping_tasks;
115     //dst_addr.addr = 61; // use 61 for messageQ transport;
117     printf("Connecting to address 0x%x on vprocId %d\n",
118             dst_addr.addr, dst_addr.vproc_id);
120     len = sizeof(struct sockaddr_rpmsg);
121     err = connect(send_sock, (struct sockaddr *)&dst_addr, len);
122     if (err < 0) {
123         printf("connect failed: %s (%d)\n", strerror(errno), errno);
124         return -1;
125     }
127     /* let's see what local address we got */
128     err = getsockname(send_sock, (struct sockaddr *)&src_addr, &len);
129     if (err < 0) {
130         printf("getsockname failed for send_sock: %s (%d)\n", strerror(errno), errno);
131         return -1;
132     }
134     printf("Our address: send_sock: socket family: %d, proc id = %d, addr = %d\n",
135                  src_addr.family, src_addr.vproc_id, src_addr.addr);
137     /* bind a local endpoint for receiving responses */
138     /* create an RPMSG socket */
139     recv_sock = socket(AF_RPMSG, SOCK_SEQPACKET, 0);
140     if (recv_sock < 0) {
141         printf("socket failed for recv_sock: %s (%d)\n", strerror(errno), errno);
142         return -1;
143     }
145     /* bind to the source address */
146     memset(&src_addr, 0, sizeof(src_addr));
147     src_addr.family = AF_RPMSG;
148     src_addr.vproc_id = coreId;
149     src_addr.addr  = 51; // use 51 for ping_tasks;
151     printf("Binding to address 0x%x on vprodId %d\n",
152             src_addr.addr, src_addr.vproc_id);
154     len = sizeof(struct sockaddr_rpmsg);
155     err = bind(recv_sock, (struct sockaddr *)&src_addr, len);
156     if (err < 0) {
157         printf("bind failed: %s (%d)\n", strerror(errno), errno);
158         return -1;
159     }
161     /* let's see what local address we got */
162     err = getsockname(recv_sock, (struct sockaddr *)&src_addr, &len);
163     if (err < 0) {
164         printf("getsockname failed for recv_sock: %s (%d)\n", strerror(errno), errno);
165         return -1;
166     }
168     printf("Our address: recv_sock: socket family: %d, proc id = %d, addr = %d\n",
169                  src_addr.family, src_addr.vproc_id, src_addr.addr);
171     FD_ZERO(&rfds);
172     FD_SET(recv_sock, &rfds);
174     printf("Sending \"%s\" in a loop.\n", msg);
176     for (i = 0; i < numLoops; i++) {
177         clock_gettime(CLOCK_REALTIME, &start);
179         err = send(send_sock, msg, strlen(msg) + 1, 0);
180         if (err < 0) {
181             printf("sendto failed: %s (%d)\n", strerror(errno), errno);
182             return -1;
183         }
185         err = select(recv_sock + 1, &rfds, NULL, NULL, NULL);
187         /* if error, try again */
188         if (err < 0) {
189             printf("Warning: select failed, trying again\n");
190             continue;
191         }
193         memset(&src_addr, 0, sizeof(src_addr));
195         len = sizeof(src_addr);
197         err = recvfrom(recv_sock, buf, sizeof(buf), 0,
198                        (struct sockaddr *)&src_addr, &len);
200         if (err < 0) {
201             printf("recvfrom failed: %s (%d)\n", strerror(errno), errno);
202             return -1;
203         }
204         if (len != sizeof(src_addr)) {
205             printf("recvfrom: got bad addr len (%d)\n", len);
206             return -1;
207         }
209         clock_gettime(CLOCK_REALTIME, &end);
210         delta = diff(start,end);
211         elapsed += delta;
213         printf("%d: Received msg: %s, from: %d\n", i, buf, src_addr.addr);
215         /*
216         printf ("Message time: %ld usecs\n", delta);
217         printf("Received a msg from address 0x%x on processor %d\n",
218                            src_addr.addr, src_addr.vproc_id);
219         printf("Message content: \"%s\".\n", buf);
220         */
221     }
222     printf ("Avg time: %ld usecs over %d iterations\n", elapsed / i, i);
224     close(send_sock);
225     close(recv_sock);
227     return 0;