[TCPIP] Implement returning TCP connections
[reactos.git] / drivers / network / tcpip / tcpip / ninfo.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS TCP/IP protocol driver
4 * FILE: tcpip/ninfo.c
5 * PURPOSE: Network information
6 * PROGRAMMERS: Art Yerkes
7 * REVISIONS:
8 * CSH 01/08-2000 Created
9 */
10
11 #include "precomp.h"
12
13 #define IP_ROUTE_TYPE_ADD 3
14 #define IP_ROUTE_TYPE_DEL 2
15
16 /* Get IPRouteEntry s for each of the routes in the system */
17 TDI_STATUS InfoTdiQueryGetRouteTable( PIP_INTERFACE IF, PNDIS_BUFFER Buffer, PUINT BufferSize ) {
18 TDI_STATUS Status;
19 KIRQL OldIrql;
20 UINT RtCount = CountFIBs(IF);
21 UINT Size = sizeof( IPROUTE_ENTRY ) * RtCount;
22 PFIB_ENTRY RCache, RCacheCur;
23 PIPROUTE_ENTRY RouteEntries, RtCurrent;
24 UINT i;
25
26 TI_DbgPrint(DEBUG_INFO, ("Called, routes = %d\n",
27 RtCount));
28
29 if (RtCount == 0)
30 return InfoCopyOut(NULL, 0, NULL, BufferSize);
31
32 RouteEntries = ExAllocatePoolWithTag( NonPagedPool, Size, ROUTE_ENTRY_TAG );
33 RtCurrent = RouteEntries;
34
35 RCache = ExAllocatePoolWithTag( NonPagedPool, sizeof( FIB_ENTRY ) * RtCount, FIB_TAG );
36 RCacheCur = RCache;
37
38 if( !RCache || !RouteEntries ) {
39 if( RCache ) ExFreePoolWithTag( RCache, FIB_TAG );
40 if( RouteEntries ) ExFreePoolWithTag( RouteEntries, ROUTE_ENTRY_TAG );
41 return TDI_NO_RESOURCES;
42 }
43
44 RtlZeroMemory( RouteEntries, Size );
45
46 RtCount = CopyFIBs( IF, RCache );
47
48 while( RtCurrent < RouteEntries + RtCount ) {
49 ASSERT(RCacheCur->Router);
50
51 RtlCopyMemory( &RtCurrent->Dest,
52 &RCacheCur->NetworkAddress.Address,
53 sizeof(RtCurrent->Dest) );
54 RtlCopyMemory( &RtCurrent->Mask,
55 &RCacheCur->Netmask.Address,
56 sizeof(RtCurrent->Mask) );
57 RtlCopyMemory( &RtCurrent->Gw,
58 &RCacheCur->Router->Address.Address,
59 sizeof(RtCurrent->Gw) );
60
61 RtCurrent->Metric1 = RCacheCur->Metric;
62 RtCurrent->Type = TDI_ADDRESS_TYPE_IP;
63
64 TI_DbgPrint
65 (DEBUG_INFO,
66 ("%d: NA %08x NM %08x GW %08x MT %x\n",
67 RtCurrent - RouteEntries,
68 RtCurrent->Dest,
69 RtCurrent->Mask,
70 RtCurrent->Gw,
71 RtCurrent->Metric1 ));
72
73 TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
74 for (i = 0; i < EntityCount; i++)
75 if (EntityList[i].context == IF)
76 break;
77
78 if (i < EntityCount)
79 RtCurrent->Index = EntityList[i].tei_instance;
80 else
81 RtCurrent->Index = 0;
82
83 TcpipReleaseSpinLock(&EntityListLock, OldIrql);
84
85 RtCurrent++; RCacheCur++;
86 }
87
88 Status = InfoCopyOut( (PCHAR)RouteEntries, Size, Buffer, BufferSize );
89
90 ExFreePoolWithTag( RouteEntries, ROUTE_ENTRY_TAG );
91 ExFreePoolWithTag( RCache, FIB_TAG );
92
93 TI_DbgPrint(DEBUG_INFO, ("Returning %08x\n", Status));
94
95 return Status;
96 }
97
98 TDI_STATUS InfoTdiQueryGetAddrTable(TDIEntityID ID,
99 PNDIS_BUFFER Buffer,
100 PUINT BufferSize)
101 {
102 KIRQL OldIrql;
103 PIPADDR_ENTRY IPEntry;
104 PIP_INTERFACE CurrentIF;
105 UINT i;
106
107 TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
108
109
110 TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
111
112 for (i = 0; i < EntityCount; i++)
113 {
114 if (EntityList[i].tei_entity == ID.tei_entity &&
115 EntityList[i].tei_instance == ID.tei_instance)
116 break;
117 }
118
119 if (i == EntityCount)
120 {
121 TcpipReleaseSpinLock(&EntityListLock, OldIrql);
122 return TDI_INVALID_PARAMETER;
123 }
124
125 IPEntry = ExAllocatePoolWithTag(NonPagedPool, sizeof(IPADDR_ENTRY), IP_ADDRESS_TAG);
126 if (!IPEntry)
127 {
128 TcpipReleaseSpinLock(&EntityListLock, OldIrql);
129 return TDI_NO_RESOURCES;
130 }
131
132 CurrentIF = EntityList[i].context;
133
134 IPEntry->Index = CurrentIF->Index;
135 GetInterfaceIPv4Address(CurrentIF,
136 ADE_UNICAST,
137 &IPEntry->Addr);
138 GetInterfaceIPv4Address(CurrentIF,
139 ADE_ADDRMASK,
140 &IPEntry->Mask);
141 GetInterfaceIPv4Address(CurrentIF,
142 ADE_BROADCAST,
143 &IPEntry->BcastAddr);
144
145 TcpipReleaseSpinLock(&EntityListLock, OldIrql);
146
147 InfoCopyOut((PCHAR)IPEntry, sizeof(IPADDR_ENTRY),
148 Buffer, BufferSize);
149
150 ExFreePoolWithTag(IPEntry, IP_ADDRESS_TAG);
151
152 return TDI_SUCCESS;
153 }
154
155 TDI_STATUS InfoTdiQueryGetIPSnmpInfo( TDIEntityID ID,
156 PIP_INTERFACE IF,
157 PNDIS_BUFFER Buffer,
158 PUINT BufferSize ) {
159 IPSNMPInfo SnmpInfo;
160 UINT IfCount = CountInterfaces();
161 UINT RouteCount = CountFIBs(IF);
162 TDI_STATUS Status = TDI_INVALID_REQUEST;
163
164 TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
165
166 RtlZeroMemory(&SnmpInfo, sizeof(SnmpInfo));
167
168 SnmpInfo.ipsi_numif = IfCount;
169 SnmpInfo.ipsi_numaddr = 1;
170 SnmpInfo.ipsi_numroutes = RouteCount;
171
172 Status = InfoCopyOut( (PCHAR)&SnmpInfo, sizeof(SnmpInfo),
173 Buffer, BufferSize );
174
175 TI_DbgPrint(DEBUG_INFO, ("Returning %08x\n", Status));
176
177 return Status;
178 }
179
180 TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
181 PNDIS_BUFFER Buffer,
182 PUINT BufferSize)
183 {
184 MIB_TCPROW TcpRow;
185 PADDRESS_FILE EndPoint;
186 TDI_STATUS Status = TDI_INVALID_REQUEST;
187
188 TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
189
190 EndPoint = NULL;
191 if (AddrFile->Connection != NULL)
192 EndPoint = AddrFile->Connection->AddressFile;
193 else if (AddrFile->Listener != NULL)
194 EndPoint = AddrFile->Listener->AddressFile;
195
196 TcpRow.State = 0; /* FIXME */
197 TcpRow.dwLocalAddr = AddrFile->Address.Address.IPv4Address;
198 TcpRow.dwLocalPort = AddrFile->Port;
199
200 if (EndPoint != NULL)
201 {
202 TcpRow.dwRemoteAddr = EndPoint->Address.Address.IPv4Address;
203 TcpRow.dwRemotePort = EndPoint->Port;
204 }
205 else
206 {
207 TcpRow.dwRemoteAddr = 0;
208 TcpRow.dwRemotePort = 0;
209 }
210
211 Status = InfoCopyOut( (PCHAR)&TcpRow, sizeof(TcpRow),
212 Buffer, BufferSize );
213
214 TI_DbgPrint(DEBUG_INFO, ("Returning %08x\n", Status));
215
216 return Status;
217 }
218
219 TDI_STATUS InfoTdiSetRoute(PIP_INTERFACE IF, PVOID Buffer, UINT BufferSize)
220 {
221 IP_ADDRESS Address, Netmask, Router;
222 PIPROUTE_ENTRY Route = Buffer;
223
224 AddrInitIPv4( &Address, Route->Dest );
225 AddrInitIPv4( &Netmask, Route->Mask );
226 AddrInitIPv4( &Router, Route->Gw );
227
228 if (!Buffer || BufferSize < sizeof(IPROUTE_ENTRY))
229 return TDI_INVALID_PARAMETER;
230
231 if (IF == Loopback)
232 {
233 DbgPrint("Failing attempt to add route to loopback adapter\n");
234 return TDI_INVALID_PARAMETER;
235 }
236
237 if( Route->Type == IP_ROUTE_TYPE_ADD ) { /* Add the route */
238 TI_DbgPrint(DEBUG_INFO,("Adding route (%s)\n", A2S(&Address)));
239 if (!RouterCreateRoute( &Address, &Netmask, &Router,
240 IF, Route->Metric1))
241 return TDI_NO_RESOURCES;
242
243 return TDI_SUCCESS;
244 } else if( Route->Type == IP_ROUTE_TYPE_DEL ) {
245 TI_DbgPrint(DEBUG_INFO,("Removing route (%s)\n", A2S(&Address)));
246 if (NT_SUCCESS(RouterRemoveRoute( &Address, &Router )))
247 return TDI_SUCCESS;
248 else
249 return TDI_INVALID_PARAMETER;
250 }
251
252 return TDI_INVALID_REQUEST;
253 }