Fix some errors detected by the GCC 4.4 compiler.
[mirror_edk2.git] / AppPkg / Applications / Sockets / RawIp4Tx / RawIp4Tx.c
1 /** @file
2 Raw IP4 transmit application
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 "RawIp4Tx.h"
16
17 UINT8 mBuffer[1024];
18
19 /**
20 Transmit raw IP4 packets to the remote system.
21
22 @param [in] ArgC Argument count
23 @param [in] ArgV Argument value array
24
25 @retval 0 Successfully operation
26 **/
27
28 int
29 RawIp4Tx (
30 IN int ArgC,
31 IN char **ArgV
32 )
33 {
34 UINT32 BytesSent;
35 ssize_t BytesTransmitted;
36 struct sockaddr_in LocalPort;
37 UINT32 RemoteAddress[4];
38 struct sockaddr_in RemotePort;
39 int RetVal;
40 UINT32 TotalSent;
41 SOCKET s;
42
43 //
44 // Create the socket
45 //
46 s = socket ( AF_INET, SOCK_RAW, RAW_PROTOCOL );
47 if ( -1 == s ) {
48 RetVal = GET_ERRNO;
49 printf ( "ERROR - socket error, errno: %d\r\n", RetVal );
50 }
51 else {
52 //
53 // Use for/break; instead of goto
54 //
55 for ( ; ; ) {
56 //
57 // Validate the arguments
58 //
59 if (( 2 > ArgC )
60 || ( 4 != sscanf ( ArgV[1],
61 "%d.%d.%d.%d",
62 &RemoteAddress[0],
63 &RemoteAddress[1],
64 &RemoteAddress[2],
65 &RemoteAddress[3]))
66 || ( 224 < RemoteAddress[0])
67 || ( 255 < RemoteAddress[1])
68 || ( 255 < RemoteAddress[2])
69 || ( 255 < RemoteAddress[3])
70 || (( 0 == RemoteAddress[0])
71 && ( 0 == RemoteAddress[1])
72 && ( 0 == RemoteAddress[2])
73 && ( 0 == RemoteAddress[3]))) {
74 printf ( "%s <remote IP address>\r\n", ArgV[0]);
75 RetVal = EINVAL;
76 break;
77 }
78
79 //
80 // Bind the socket to a local port
81 //
82 memset ( &LocalPort, 0, sizeof ( LocalPort ));
83 SIN_LEN ( LocalPort ) = sizeof ( LocalPort );
84 SIN_FAMILY ( LocalPort ) = AF_INET;
85 SIN_ADDR ( LocalPort ) = 0;
86 SIN_PORT ( LocalPort ) = 0;
87 RetVal = bind ( s,
88 (struct sockaddr *)&LocalPort,
89 sizeof ( LocalPort ));
90 if ( -1 == RetVal ) {
91 RetVal = GET_ERRNO;
92 printf ( "ERROR - bind error, errno: %d\r\n", RetVal );
93 break;
94 }
95
96 //
97 // Specify the remote port
98 //
99 memset ( &RemotePort, 0, sizeof ( RemotePort ));
100 SIN_LEN ( RemotePort ) = sizeof ( RemotePort );
101 SIN_FAMILY ( RemotePort ) = AF_INET;
102 SIN_ADDR ( RemotePort ) = ( RemoteAddress[3] << 24 )
103 | ( RemoteAddress[2] << 16 )
104 | ( RemoteAddress[1] << 8 )
105 | RemoteAddress[0];
106 SIN_PORT ( RemotePort ) = 0;
107
108 //
109 // Initialize the messages
110 //
111 memset ( &mBuffer[0], 0, sizeof ( mBuffer ));
112
113 //
114 // Send the data before the out-of-band message
115 //
116 TotalSent = 0;
117 BytesSent = 0;
118 do {
119 BytesTransmitted = sendto ( s,
120 &mBuffer[BytesSent],
121 sizeof ( mBuffer ) - BytesSent,
122 0,
123 (struct sockaddr *)&RemotePort,
124 sizeof ( RemotePort ));
125 if ( -1 == BytesTransmitted ) {
126 RetVal = GET_ERRNO;
127 printf ( "ERROR - send before error, errno: %d\r\n", RetVal );
128 break;
129 }
130 BytesSent += (UINT32)BytesTransmitted;
131 RetVal = 0;
132 } while ( sizeof ( mBuffer ) > BytesSent );
133 if ( 0 != RetVal ) {
134 break;
135 }
136 TotalSent += BytesSent;
137
138 //
139 // Test completed successfully
140 //
141 if ( 0 == RetVal ) {
142 printf ( "Bytes sent: %8d\r\n", TotalSent );
143 }
144 break;
145 }
146
147 //
148 // Close the socket
149 //
150 CLOSE_SOCKET ( s );
151 }
152
153 //
154 // Return the operation status
155 //
156 return RetVal;
157 }