Author: elecharny
Date: Mon May  2 14:23:08 2005
New Revision: 165698

URL: http://svn.apache.org/viewcvs?rev=165698&view=rev
Log:
Added a test and factorized some common code in a function.

Modified:
    directory/sandbox/trunk/asn1-new-codec/src/test/org/apache/asn1/spnego/codec/SpnegoTest.java

Modified: directory/sandbox/trunk/asn1-new-codec/src/test/org/apache/asn1/spnego/codec/SpnegoTest.java
URL: http://svn.apache.org/viewcvs/directory/sandbox/trunk/asn1-new-codec/src/test/org/apache/asn1/spnego/codec/SpnegoTest.java?rev=165698&r1=165697&r2=165698&view=diff
==============================================================================
--- directory/sandbox/trunk/asn1-new-codec/src/test/org/apache/asn1/spnego/codec/SpnegoTest.java (original)
+++ directory/sandbox/trunk/asn1-new-codec/src/test/org/apache/asn1/spnego/codec/SpnegoTest.java Mon May  2 14:23:08 2005
@@ -23,6 +23,7 @@
 import org.apache.asn1.ber.containers.IAsn1Container;
 import org.apache.asn1.ldap.codec.BindRequestTest;
 import org.apache.asn1.primitives.OID;
+import org.apache.asn1.primitives.OctetString;
 import org.apache.asn1.spnego.pojo.SpnegoNegTokenInitPOJO;
 import org.apache.asn1.spnego.pojo.SpnegoPOJO;
 import org.apache.asn1.util.pools.PoolException;
@@ -43,6 +44,21 @@
     {
         PropertyConfigurator.configure( "conf/log4j.conf" );
     }
+    
+    /**
+     * Check the default value of Req Flags
+     * @param spnego The spnego POJO to test
+     */
+    private void checkDefaultReqFlags(SpnegoPOJO spnego)
+    {
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
+        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+    }
 
     /**
      * Test the decoding of an empty Spnego NegTokenInit.
@@ -107,13 +123,7 @@
         Assert.assertEquals(0, nbOids);
         
         // Check the reqFlags
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+        checkDefaultReqFlags(spnego);
 
         // Check the mech token
         Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechToken());
@@ -193,13 +203,7 @@
         Assert.assertEquals("1.2.840.48018.1.2.2", oids[0].getOID());
         
         // Check the reqFlags
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+        checkDefaultReqFlags(spnego);
 
         // Check the mech token
         Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechToken());
@@ -292,13 +296,7 @@
         }
         
         // Check the reqFlags
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+        checkDefaultReqFlags(spnego);
 
         // Check the mech token
         Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechToken());
@@ -388,13 +386,7 @@
         }
         
         // Check the reqFlags
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+        checkDefaultReqFlags(spnego);
 
         // Check the mech token
         Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechToken());
@@ -408,6 +400,253 @@
     }
 
     /**
+     * Test the decoding of a Spnego NegTokenInit with an empty mech token .
+     */
+    public void testDecodeSpnegoNegTokenInitEmptyMechToken()
+    {
+        Asn1Decoder spnegoDecoder = new SpnegoDecoder();
+
+        ByteBuffer  stream      = ByteBuffer.allocate( 0x08 );
+        stream.put(
+            new byte[]
+            {
+                (byte)0xa0, 0x06, 	// SPNEGO --> CHOICE {
+                					//     	negTokenInit  [0]  NegTokenInit
+                0x30, 0x04, 		// NegTokenInit ::= SEQUENCE {
+                (byte)0xA2, 0x02,	// mechToken     [2]  OCTET STRING  OPTIONAL ,
+                0x04, 0x00  		// empty mech token 
+            } );
+
+        stream.flip();
+
+        // Allocate a Spnego Container
+        IAsn1Container spnegoContainer = null;
+
+        try
+        {
+            spnegoContainer = ( IAsn1Container ) spnegoDecoder.allocate(
+                    SpnegoPoolEnum.SPNEGO_CONTAINER_POOL );
+        }
+        catch ( PoolException pe )
+        {
+            Assert.fail("Cannot allocat a SpnegoContainer : " + pe.getMessage());
+        }
+
+        spnegoContainer.setPoolManager( spnegoDecoder.getPoolManager() );
+
+        try
+        {
+            spnegoDecoder.decode( stream, spnegoContainer );
+        }
+        catch ( DecoderException de )
+        {
+            de.printStackTrace();
+            Assert.fail( de.getMessage() );
+        }
+    	
+        SpnegoPOJO spnego = ( ( SpnegoContainer ) spnegoContainer ).getSpnego();
+
+        // Checks the Oids
+        OID oids[] = ((SpnegoNegTokenInitPOJO)spnego).getMechTypeList();
+        int nbOids = 0;
+        
+        for ( int i = 0 ; i < oids.length ; i++ )
+        {
+            if (oids[i] == null)
+            {
+                break;
+            }
+            
+            nbOids ++;
+        }
+        
+        Assert.assertEquals(0, nbOids);
+        
+        // Check the reqFlags
+        checkDefaultReqFlags(spnego);
+
+        // Check the mech token
+        OctetString mechToken = ((SpnegoNegTokenInitPOJO)spnego).getMechToken();
+        
+        Assert.assertEquals(OctetString.EMPTY_STRING, mechToken);
+        Assert.assertEquals( "" , mechToken.toString() );
+
+        // Check the mech list MIC
+        Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechListMIC());
+
+        // Free the BindRequest Container. It will be put back in the IPool
+        // after being reset.
+        spnegoContainer.free();
+    }
+
+    /**
+     * Test the decoding of a Spnego NegTokenInit with a mech token.
+     */
+    public void testDecodeSpnegoNegTokenInitMechToken()
+    {
+        Asn1Decoder spnegoDecoder = new SpnegoDecoder();
+
+        ByteBuffer  stream      = ByteBuffer.allocate( 0x0C );
+        stream.put(
+            new byte[]
+            {
+                (byte)0xa0, 0x0A, 		// SPNEGO --> CHOICE {
+                						//     	negTokenInit  [0]  NegTokenInit
+                0x30, 0x08, 			// NegTokenInit ::= SEQUENCE {
+                (byte)0xA2, 0x06,		// mechToken     [2]  OCTET STRING  OPTIONAL ,
+                0x04, 0x04, 'a', 'b', 'c', 'd' 	// mechToken = 'abcd'
+            } );
+
+        stream.flip();
+
+        // Allocate a Spnego Container
+        IAsn1Container spnegoContainer = null;
+
+        try
+        {
+            spnegoContainer = ( IAsn1Container ) spnegoDecoder.allocate(
+                    SpnegoPoolEnum.SPNEGO_CONTAINER_POOL );
+        }
+        catch ( PoolException pe )
+        {
+            Assert.fail("Cannot allocat a SpnegoContainer : " + pe.getMessage());
+        }
+
+        spnegoContainer.setPoolManager( spnegoDecoder.getPoolManager() );
+
+        try
+        {
+            spnegoDecoder.decode( stream, spnegoContainer );
+        }
+        catch ( DecoderException de )
+        {
+            de.printStackTrace();
+            Assert.fail( de.getMessage() );
+        }
+    	
+        SpnegoPOJO spnego = ( ( SpnegoContainer ) spnegoContainer ).getSpnego();
+
+        // Checks the Oids
+        OID oids[] = ((SpnegoNegTokenInitPOJO)spnego).getMechTypeList();
+        int nbOids = 0;
+        
+        for ( int i = 0 ; i < oids.length ; i++ )
+        {
+            if (oids[i] == null)
+            {
+                break;
+            }
+            
+            nbOids ++;
+        }
+        
+        Assert.assertEquals(0, nbOids);
+        
+        // Check the reqFlags
+        checkDefaultReqFlags(spnego);
+
+        // Check the mech token
+        Assert.assertEquals("[61][62][63][64]", ((SpnegoNegTokenInitPOJO)spnego).getMechToken().toString());
+
+        // Check the mech list MIC
+        Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechListMIC());
+
+        // Free the BindRequest Container. It will be put back in the IPool
+        // after being reset.
+        spnegoContainer.free();
+    }
+
+    /**
+     * Test the decoding of a Spnego NegTokenInit with a 2048 bytes long mech token .
+     */
+    public void testDecodeSpnegoNegTokenInit2048MechToken()
+    {
+        Asn1Decoder spnegoDecoder = new SpnegoDecoder();
+
+        ByteBuffer  stream      = ByteBuffer.allocate( 0x810 );
+        stream.put(
+            new byte[]
+            {
+                (byte)0xa0, (byte)0x82, (byte)0x08, 0x0C, 	// SPNEGO --> CHOICE {
+                											//     	negTokenInit  [0]  NegTokenInit
+                0x30, (byte)0x82, (byte)0x08, 0x08, 		// NegTokenInit ::= SEQUENCE {
+                (byte)0xA2, (byte)0x82, (byte)0x08, 0x04,	// mechToken     [2]  OCTET STRING  OPTIONAL ,
+                0x04, (byte)0x82, (byte)0x08, 0x00  		// 2048 octets 
+            } );
+
+        // Stores 2048 bytes in the stream
+        for ( int i = 0 ; i < 2048; i++)
+        {
+            stream.put( (byte)((i % 256) & 0x00FF) );
+        }
+        
+        stream.flip();
+
+        // Allocate a Spnego Container
+        IAsn1Container spnegoContainer = null;
+
+        try
+        {
+            spnegoContainer = ( IAsn1Container ) spnegoDecoder.allocate(
+                    SpnegoPoolEnum.SPNEGO_CONTAINER_POOL );
+        }
+        catch ( PoolException pe )
+        {
+            Assert.fail("Cannot allocat a SpnegoContainer : " + pe.getMessage());
+        }
+
+        spnegoContainer.setPoolManager( spnegoDecoder.getPoolManager() );
+
+        try
+        {
+            spnegoDecoder.decode( stream, spnegoContainer );
+        }
+        catch ( DecoderException de )
+        {
+            de.printStackTrace();
+            Assert.fail( de.getMessage() );
+        }
+    	
+        SpnegoPOJO spnego = ( ( SpnegoContainer ) spnegoContainer ).getSpnego();
+
+        // Checks the Oids
+        OID oids[] = ((SpnegoNegTokenInitPOJO)spnego).getMechTypeList();
+        int nbOids = 0;
+        
+        for ( int i = 0 ; i < oids.length ; i++ )
+        {
+            if (oids[i] == null)
+            {
+                break;
+            }
+            
+            nbOids ++;
+        }
+        
+        Assert.assertEquals(0, nbOids);
+        
+        // Check the reqFlags
+        checkDefaultReqFlags(spnego);
+
+        // Check the mech token
+        OctetString mechToken = ((SpnegoNegTokenInitPOJO)spnego).getMechToken();
+        
+        Assert.assertEquals(2048, mechToken.getValue().length);
+        
+        for ( int i = 0 ; i < 2048 ; i++ )
+        {
+            Assert.assertEquals( (byte)((i % 256) & 0x00FF) , mechToken.getValue()[i] );
+        }
+
+        // Check the mech list MIC
+        Assert.assertEquals(null, ((SpnegoNegTokenInitPOJO)spnego).getMechListMIC());
+
+        // Free the BindRequest Container. It will be put back in the IPool
+        // after being reset.
+        spnegoContainer.free();
+    }
+
+    /**
      * Test the decoding of a Spnego NegTokenInit with 2 mech types and
      * one mech token.
      */
@@ -479,13 +718,7 @@
         Assert.assertEquals("1.2.840.113554.1.2.2", oids[1].getOID());
         
         // Check the reqFlags
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isAnonFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isConfFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isDelegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isIntegFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isMutualFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isReplayFlag());
-        Assert.assertEquals(false, ((SpnegoNegTokenInitPOJO)spnego).isSequenceFlag());
+        checkDefaultReqFlags(spnego);
 
         // Check the mech token
         Assert.assertEquals("[61][62][63][64]", ((SpnegoNegTokenInitPOJO)spnego).getMechToken().toString());