Changeset 772 for branches/smbpdr-1.0/smb.c

Timestamp:

May 2, 2013, 5:39:58 PM (12 years ago)

Author:

Alex Taylor

Message:

Fix to SplPdQuery allowing the required buffer size to be queried.

File:

• branches/smbpdr-1.0/smb.c (modified) (1 diff)

branches/smbpdr-1.0/smb.c

@@ -955 +955 @@
 
         case BIDI_Q_PORTDRV:
+            // If no buffer was provided, just return the required size
+            if ( !pOutData && pcbOutData ) {
+                *pcbOutData = sizeof( PORTSETTINGS );
+                return ERROR_MORE_DATA;
+            }
             // Make sure the data is valid
-            if ( !pOutData || *pcbOutData < sizeof( PORTSETTINGS ))
+            if ( !pOutData || !pcbOutData || ( *pcbOutData < sizeof( PORTSETTINGS )))
                 return( ERROR_INVALID_PARAMETER );
             pSettings = (PPORTSETTINGS) pOutData;