Add Socket Library applications.
[mirror_edk2.git] / AppPkg / Applications / Sockets / RecvDgram / RecvDgram.c
1 /** @file
2 Receive a datagram
3
4 Copyright (c) 2011, Intel Corporation
5 All rights reserved. This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12
13 **/
14
15 #include <errno.h>
16 #include <string.h>
17 #include <Uefi.h>
18 #include <unistd.h>
19
20 #include <Library/DebugLib.h>
21 #include <Library/UefiLib.h>
22
23 #include <netinet/in.h>
24
25 #include <sys/socket.h>
26 #include <sys/time.h>
27
28 UINT8 mBuffer [ 65536 ];
29
30 /**
31 Receive a datagram
32
33 @param [in] Argc The number of arguments
34 @param [in] Argv The argument value array
35
36 @retval 0 The application exited normally.
37 @retval Other An error occurred.
38 **/
39 int
40 main (
41 IN int Argc,
42 IN char **Argv
43 )
44 {
45 struct sockaddr_in Address;
46 socklen_t AddressLength;
47 ssize_t LengthInBytes;
48 int s;
49 int Status;
50 struct timeval Timeout;
51
52
53 DEBUG (( DEBUG_INFO,
54 "%a starting\r\n",
55 Argv[0]));
56
57 //
58 // Get the socket
59 //
60 s = socket ( AF_INET, SOCK_DGRAM, 0 );
61 if ( -1 == s ) {
62 Print ( L"ERROR - Unable to open the socket, errno: %d\r\n", errno );
63 }
64 else {
65 Timeout.tv_sec = 5;
66 Timeout.tv_usec = 0;
67 Status = setsockopt ( s,
68 SOL_SOCKET,
69 SO_RCVTIMEO,
70 &Timeout,
71 sizeof ( Timeout ));
72 if ( -1 == Status ) {
73 Print ( L"ERROR - Unable to set the receive timeout, errno: %d\r\n", errno );
74 }
75 else {
76 AddressLength = sizeof ( Address );
77 LengthInBytes = recvfrom ( s,
78 &mBuffer[0],
79 sizeof ( mBuffer[0]),
80 0,
81 (struct sockaddr *)&Address,
82 &AddressLength );
83 if ( -1 == LengthInBytes ) {
84 if ( ETIMEDOUT == errno ) {
85 Print ( L"No datagram received\r\n" );
86 }
87 else {
88 Print ( L"ERROR - No datagram received, errno: %d\r\n", errno );
89 }
90 }
91 else {
92 Print ( L"Received %d bytes from %d.%d.%d.%d:%d\r\n",
93 LengthInBytes,
94 (UINT8)Address.sin_addr.s_addr,
95 (UINT8)( Address.sin_addr.s_addr >> 8 ),
96 (UINT8)( Address.sin_addr.s_addr >> 16 ),
97 (UINT8)( Address.sin_addr.s_addr >> 24 ),
98 htons ( Address.sin_port ));
99 }
100 }
101
102 //
103 // Done with the socket
104 //
105 close ( s );
106 }
107
108 //
109 // All done
110 //
111 DEBUG (( DEBUG_INFO,
112 "%a exiting, errno: %d\r\n",
113 Argv[0],
114 errno ));
115 return errno;
116 }