reactos/drivers/network/tcpip/tcpip/info.c

   1 /*
   2  * COPYRIGHT:   See COPYING in the top level directory
   3  * PROJECT:     ReactOS TCP/IP protocol driver
   4  * FILE:        tcpip/info.c
   5  * PURPOSE:     TDI query and set information routines
   6  * PROGRAMMERS: Casper S. Hornstrup (chorns@users.sourceforge.net)
   7  * REVISIONS:
   8  *   CSH 01/08-2000 Created
   9  */
  10
  11 #include "precomp.h"
  12
  13 #include <debug.h>
  14
  15 VOID AddEntity(ULONG EntityType, PVOID Context, ULONG Flags)
  16 {
  17     KIRQL OldIrql;
  18     ULONG i, Instance = 0;
  19     BOOLEAN ChoseIndex = FALSE;
  20
  21     TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
  22
  23     while (!ChoseIndex)
  24     {
  25          ChoseIndex = TRUE;
  26          for (i = 0; i < EntityCount; i++)
  27          {
  28              if (EntityList[i].tei_entity == EntityType &&
  29                  EntityList[i].tei_instance == Instance)
  30              {
  31                  Instance++;
  32                  ChoseIndex = FALSE;
  33              }
  34          }
  35     }
  36
  37     EntityList[EntityCount].tei_entity = EntityType;
  38     EntityList[EntityCount].tei_instance = Instance;
  39     EntityList[EntityCount].context = Context;
  40     EntityList[EntityCount].flags = Flags;
  41     EntityCount++;
  42
  43     TcpipReleaseSpinLock(&EntityListLock, OldIrql);
  44 }
  45
  46 VOID RemoveEntityByContext(PVOID Context)
  47 {
  48    ULONG i;
  49    KIRQL OldIrql;
  50
  51    TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
  52
  53    for (i = 0; i < EntityCount; i++) {
  54         if( EntityList[i].context == Context ) {
  55             if( i != EntityCount - 1 ) {
  56                 memcpy( &EntityList[i],
  57                         &EntityList[--EntityCount],
  58                         sizeof(EntityList[i]) );
  59             } else {
  60                 EntityCount--;
  61             }
  62         }
  63     }
  64
  65     TcpipReleaseSpinLock(&EntityListLock, OldIrql);
  66 }
  67
  68 PVOID GetContext(TDIEntityID ID)
  69 {
  70     UINT i;
  71     KIRQL OldIrql;
  72     PVOID Context;
  73
  74     TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
  75
  76     for (i = 0; i < EntityCount; i++)
  77     {
  78         if (EntityList[i].tei_entity == ID.tei_entity &&
  79             EntityList[i].tei_instance == ID.tei_instance)
  80             break;
  81     }
  82
  83     if (i == EntityCount)
  84     {
  85         TcpipReleaseSpinLock(&EntityListLock, OldIrql);
  86         DbgPrint("WARNING: Unable to get context for %d %d\n", ID.tei_entity, ID.tei_instance);
  87         return NULL;
  88     }
  89
  90     Context = EntityList[i].context;
  91
  92     TcpipReleaseSpinLock(&EntityListLock, OldIrql);
  93
  94     return Context;
  95 }
  96
  97 TDI_STATUS InfoCopyOut( PCHAR DataOut, UINT SizeOut,
  98                         PNDIS_BUFFER ClientBuf, PUINT ClientBufSize ) {
  99     UINT RememberedCBSize = *ClientBufSize;
 100     *ClientBufSize = SizeOut;
 101
 102     /* The driver returns success even when it couldn't fit every available
 103      * byte. */
 104     if( RememberedCBSize < SizeOut || !ClientBuf )
 105         return TDI_SUCCESS;
 106     else {
 107         CopyBufferToBufferChain( ClientBuf, 0, (PCHAR)DataOut, SizeOut );
 108         return TDI_SUCCESS;
 109     }
 110 }
 111
 112 TDI_STATUS InfoTdiQueryEntityType(TDIEntityID ID,
 113                                   PNDIS_BUFFER Buffer,
 114                                   PUINT BufferSize)
 115 {
 116     KIRQL OldIrql;
 117     UINT i, Flags = 0;
 118
 119     TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
 120
 121     for (i = 0; i < EntityCount; i++)
 122     {
 123         if (EntityList[i].tei_entity == ID.tei_entity &&
 124             EntityList[i].tei_instance == ID.tei_instance)
 125             break;
 126     }
 127
 128     if (i == EntityCount)
 129     {
 130         TcpipReleaseSpinLock(&EntityListLock, OldIrql);
 131         return TDI_INVALID_PARAMETER;
 132     }
 133
 134     Flags = EntityList[i].flags;
 135
 136     InfoCopyOut((PCHAR)&Flags,
 137                 sizeof(ULONG),
 138                 Buffer,
 139                 BufferSize);
 140
 141     TcpipReleaseSpinLock(&EntityListLock, OldIrql);
 142
 143     return TDI_SUCCESS;
 144 }
 145
 146 TDI_STATUS InfoTdiQueryListEntities(PNDIS_BUFFER Buffer,
 147                                     PUINT BufferSize)
 148 {
 149     UINT Count, Size, BufSize = *BufferSize;
 150     KIRQL OldIrql;
 151
 152     TI_DbgPrint(DEBUG_INFO,("About to copy %d TDIEntityIDs to user\n",
 153                            EntityCount));
 154
 155     TcpipAcquireSpinLock(&EntityListLock, &OldIrql);
 156
 157     Size = EntityCount * sizeof(TDIEntityID);
 158     *BufferSize = Size;
 159
 160     TI_DbgPrint(DEBUG_INFO,("BufSize: %d, NeededSize: %d\n", BufSize, Size));
 161
 162     if (BufSize < Size || !Buffer)
 163     {
 164         TcpipReleaseSpinLock( &EntityListLock, OldIrql );
 165         /* The buffer is too small to contain requested data, but we return
 166          * success anyway, as we did everything we wanted. */
 167         return TDI_SUCCESS;
 168     }
 169
 170     /* Return entity list -- Copy only the TDIEntityID parts. */
 171     for( Count = 0; Count < EntityCount; Count++ ) {
 172         CopyBufferToBufferChain(Buffer,
 173                                 Count * sizeof(TDIEntityID),
 174                                 (PCHAR)&EntityList[Count],
 175                                 sizeof(TDIEntityID));
 176     }
 177
 178     TcpipReleaseSpinLock(&EntityListLock, OldIrql);
 179
 180     return TDI_SUCCESS;
 181 }
 182
 183 TDI_STATUS InfoTdiQueryInformationEx(
 184   PTDI_REQUEST Request,
 185   TDIObjectID *ID,
 186   PNDIS_BUFFER Buffer,
 187   PUINT BufferSize,
 188   PVOID Context)
 189 /*
 190  * FUNCTION: Returns extended information
 191  * ARGUMENTS:
 192  *   Request    = Pointer to TDI request structure for the request
 193  *   ID         = TDI object ID
 194  *   Buffer     = Pointer to buffer with data to use
 195  *   BufferSize = Pointer to buffer with size of Buffer. On return
 196  *                this is filled with number of bytes returned
 197  *   Context    = Pointer to context buffer
 198  * RETURNS:
 199  *   Status of operation
 200  */
 201 {
 202     PVOID EntityListContext;
 203
 204     TI_DbgPrint(DEBUG_INFO,
 205                 ("InfoEx Req: %x %x %x!%04x:%d\n",
 206                  ID->toi_class,
 207                  ID->toi_type,
 208                  ID->toi_id,
 209                  ID->toi_entity.tei_entity,
 210                  ID->toi_entity.tei_instance));
 211
 212     switch (ID->toi_class)
 213     {
 214         case INFO_CLASS_GENERIC:
 215            switch (ID->toi_id)
 216            {
 217               case ENTITY_LIST_ID:
 218                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 219                      return TDI_INVALID_PARAMETER;
 220
 221                  return InfoTdiQueryListEntities(Buffer, BufferSize);
 222
 223               case ENTITY_TYPE_ID:
 224                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 225                      return TDI_INVALID_PARAMETER;
 226
 227                  return InfoTdiQueryEntityType(ID->toi_entity, Buffer, BufferSize);
 228
 229               default:
 230                  return TDI_INVALID_REQUEST;
 231            }
 232
 233         case INFO_CLASS_PROTOCOL:
 234            if (ID->toi_type == INFO_TYPE_ADDRESS_OBJECT)
 235            {
 236                if ((EntityListContext = GetContext(ID->toi_entity)))
 237                     return GetAddressFileInfo(ID, EntityListContext, Buffer, BufferSize);
 238                else
 239                     return TDI_INVALID_PARAMETER;
 240            }
 241
 242            switch (ID->toi_id)
 243            {
 244               case IF_MIB_STATS_ID:
 245                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 246                      return TDI_INVALID_PARAMETER;
 247
 248                  if (ID->toi_entity.tei_entity == IF_ENTITY)
 249                      if ((EntityListContext = GetContext(ID->toi_entity)))
 250                          return InfoTdiQueryGetInterfaceMIB(ID->toi_entity, EntityListContext, Buffer, BufferSize);
 251                      else
 252                          return TDI_INVALID_PARAMETER;
 253                  else if (ID->toi_entity.tei_entity == CL_NL_ENTITY ||
 254                           ID->toi_entity.tei_entity == CO_NL_ENTITY)
 255                      if ((EntityListContext = GetContext(ID->toi_entity)))
 256                          return InfoTdiQueryGetIPSnmpInfo(ID->toi_entity, EntityListContext, Buffer, BufferSize);
 257                      else
 258                          return TDI_INVALID_PARAMETER;
 259                  else
 260                      return TDI_INVALID_PARAMETER;
 261
 262               case IP_MIB_ADDRTABLE_ENTRY_ID:
 263                  if (ID->toi_entity.tei_entity != CL_NL_ENTITY &&
 264                      ID->toi_entity.tei_entity != CO_NL_ENTITY)
 265                      return TDI_INVALID_PARAMETER;
 266
 267                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 268                      return TDI_INVALID_PARAMETER;
 269
 270                  return InfoTdiQueryGetAddrTable(ID->toi_entity, Buffer, BufferSize);
 271
 272               case IP_MIB_ARPTABLE_ENTRY_ID:
 273                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 274                      return TDI_INVALID_PARAMETER;
 275
 276                  if (ID->toi_entity.tei_entity == AT_ENTITY)
 277                      if ((EntityListContext = GetContext(ID->toi_entity)))
 278                          return InfoTdiQueryGetArptableMIB(ID->toi_entity, EntityListContext,
 279                                                            Buffer, BufferSize);
 280                      else
 281                          return TDI_INVALID_PARAMETER;
 282                  else if (ID->toi_entity.tei_entity == CO_NL_ENTITY ||
 283                           ID->toi_entity.tei_entity == CL_NL_ENTITY)
 284                      if ((EntityListContext = GetContext(ID->toi_entity)))
 285                          return InfoTdiQueryGetRouteTable(EntityListContext, Buffer, BufferSize);
 286                      else
 287                          return TDI_INVALID_PARAMETER;
 288                  else
 289                      return TDI_INVALID_PARAMETER;
 290
 291 #if 0
 292               case IP_INTFC_INFO_ID:
 293                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 294                      return TDI_INVALID_PARAMETER;
 295
 296                  return InfoTdiQueryGetIFInfo(Context, Buffer, BufferSize);
 297 #endif
 298
 299               default:
 300                  return TDI_INVALID_REQUEST;
 301            }
 302
 303         default:
 304            return TDI_INVALID_REQUEST;
 305     }
 306 }
 307
 308 TDI_STATUS InfoTdiSetInformationEx
 309 (PTDI_REQUEST Request,
 310  TDIObjectID *ID,
 311  PVOID Buffer,
 312  UINT BufferSize)
 313 /*
 314  * FUNCTION: Sets extended information
 315  * ARGUMENTS:
 316  *   Request    = Pointer to TDI request structure for the request
 317  *   ID         = Pointer to TDI object ID
 318  *   Buffer     = Pointer to buffer with data to use
 319  *   BufferSize = Size of Buffer
 320  * RETURNS:
 321  *   Status of operation
 322  */
 323 {
 324     PVOID EntityListContext;
 325
 326     switch (ID->toi_class)
 327     {
 328        case INFO_CLASS_PROTOCOL:
 329           if (ID->toi_type == INFO_TYPE_ADDRESS_OBJECT)
 330           {
 331               if ((EntityListContext = GetContext(ID->toi_entity)))
 332                    return SetAddressFileInfo(ID, EntityListContext, Buffer, BufferSize);
 333               else
 334                    return TDI_INVALID_PARAMETER;
 335           }
 336
 337           switch (ID->toi_id)
 338           {
 339               case IP_MIB_ARPTABLE_ENTRY_ID:
 340                  if (ID->toi_type != INFO_TYPE_PROVIDER)
 341                      return TDI_INVALID_PARAMETER;
 342
 343                  if (ID->toi_entity.tei_entity == AT_ENTITY)
 344                      if ((EntityListContext = GetContext(ID->toi_entity)))
 345                          return InfoTdiSetArptableMIB(EntityListContext,
 346                                                       Buffer, BufferSize);
 347                      else
 348                          return TDI_INVALID_PARAMETER;
 349                  else if (ID->toi_entity.tei_entity == CL_NL_ENTITY ||
 350                           ID->toi_entity.tei_entity == CO_NL_ENTITY)
 351                      if ((EntityListContext = GetContext(ID->toi_entity)))
 352                         return InfoTdiSetRoute(EntityListContext, Buffer, BufferSize);
 353                      else
 354                         return TDI_INVALID_PARAMETER;
 355                  else
 356                      return TDI_INVALID_PARAMETER;
 357
 358               default:
 359                 return TDI_INVALID_REQUEST;
 360           }
 361
 362        default:
 363           return TDI_INVALID_REQUEST;
 364     }
 365 }