Have all tests use EXIT() to exit rather than exit(), since the latter doesn't
[openssl.git] / crypto / rc4 / rc4test.c
index aa4053e64496aa80814d0ef16a5ec1eb8de66ef7..b9d8f209753667b59e483bd90dad2f46eb9c4e49 100644 (file)
@@ -1,5 +1,5 @@
 /* crypto/rc4/rc4test.c */
 /* crypto/rc4/rc4test.c */
-/* Copyright (C) 1995-1997 Eric Young (eay@cryptsoft.com)
+/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  * All rights reserved.
  *
  * This package is an SSL implementation written
  * All rights reserved.
  *
  * This package is an SSL implementation written
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include "rc4.h"
 
 
-unsigned char keys[6][11]={
+#include "../e_os.h"
+
+#ifdef OPENSSL_NO_RC4
+int main(int argc, char *argv[])
+{
+    printf("No RC4 support\n");
+    return(0);
+}
+#else
+#include <openssl/rc4.h>
+
+static unsigned char keys[7][30]={
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
        {8,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
        {4,0xef,0x01,0x23,0x45},
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
        {8,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
        {4,0xef,0x01,0x23,0x45},
        {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
+       {4,0xef,0x01,0x23,0x45},
        };
 
        };
 
-unsigned char data[6][11]={
-       {8,0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef},
-       {8,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
-       {8,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
-       {10,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
+static unsigned char data_len[7]={8,8,8,20,28,10};
+static unsigned char data[7][30]={
+       {0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xff},
+       {0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xff},
+       {0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xff},
+       {0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+          0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+          0x00,0x00,0x00,0x00,0xff},
+       {0x12,0x34,0x56,0x78,0x9A,0xBC,0xDE,0xF0,
+          0x12,0x34,0x56,0x78,0x9A,0xBC,0xDE,0xF0,
+          0x12,0x34,0x56,0x78,0x9A,0xBC,0xDE,0xF0,
+          0x12,0x34,0x56,0x78,0xff},
+       {0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xff},
        {0},
        };
 
        {0},
        };
 
-unsigned char output[6][11]={
-       {0x75,0xb7,0x87,0x80,0x99,0xe0,0xc5,0x96},
-       {0x74,0x94,0xc2,0xe7,0x10,0x4b,0x08,0x79},
-       {0xde,0x18,0x89,0x41,0xa3,0x37,0x5d,0x3a},
-       {0xd6,0xa1,0x41,0xa7,0xec,0x3c,0x38,0xdf,0xbd,0x61},
+static unsigned char output[7][30]={
+       {0x75,0xb7,0x87,0x80,0x99,0xe0,0xc5,0x96,0x00},
+       {0x74,0x94,0xc2,0xe7,0x10,0x4b,0x08,0x79,0x00},
+       {0xde,0x18,0x89,0x41,0xa3,0x37,0x5d,0x3a,0x00},
+       {0xd6,0xa1,0x41,0xa7,0xec,0x3c,0x38,0xdf,
+        0xbd,0x61,0x5a,0x11,0x62,0xe1,0xc7,0xba,
+        0x36,0xb6,0x78,0x58,0x00},
+       {0x66,0xa0,0x94,0x9f,0x8a,0xf7,0xd6,0x89,
+        0x1f,0x7f,0x83,0x2b,0xa8,0x33,0xc0,0x0c,
+        0x89,0x2e,0xbe,0x30,0x14,0x3c,0xe2,0x87,
+        0x40,0x01,0x1e,0xcf,0x00},
+       {0xd6,0xa1,0x41,0xa7,0xec,0x3c,0x38,0xdf,0xbd,0x61,0x00},
        {0},
        };
 
        {0},
        };
 
-int main(argc,argv)
-int argc;
-char *argv[];
+int main(int argc, char *argv[])
        {
        int i,err=0;
        {
        int i,err=0;
-       unsigned int j;
+       int j;
        unsigned char *p;
        RC4_KEY key;
        unsigned char buf[512],obuf[512];
 
        for (i=0; i<512; i++) buf[i]=0x01;
 
        unsigned char *p;
        RC4_KEY key;
        unsigned char buf[512],obuf[512];
 
        for (i=0; i<512; i++) buf[i]=0x01;
 
-       for (i=0; i<4; i++)
+       for (i=0; i<6; i++)
                {
                RC4_set_key(&key,keys[i][0],&(keys[i][1]));
                {
                RC4_set_key(&key,keys[i][0],&(keys[i][1]));
-               RC4(&key,data[i][0],&(data[i][1]),obuf);
-               if (memcmp(obuf,output[i],data[i][0]) != 0)
+               memset(obuf,0x00,sizeof(obuf));
+               RC4(&key,data_len[i],&(data[i][0]),obuf);
+               if (memcmp(obuf,output[i],data_len[i]+1) != 0)
                        {
                        printf("error calculating RC4\n");
                        printf("output:");
                        {
                        printf("error calculating RC4\n");
                        printf("output:");
-                       for (j=0; j<data[i][0]; j++)
+                       for (j=0; j<data_len[i]+1; j++)
                                printf(" %02x",obuf[j]);
                        printf("\n");
                        printf("expect:");
                        p= &(output[i][0]);
                                printf(" %02x",obuf[j]);
                        printf("\n");
                        printf("expect:");
                        p= &(output[i][0]);
-                       for (j=0; j<data[i][0]; j++)
+                       for (j=0; j<data_len[i]+1; j++)
                                printf(" %02x",*(p++));
                        printf("\n");
                        err++;
                                printf(" %02x",*(p++));
                        printf("\n");
                        err++;
@@ -118,7 +143,61 @@ char *argv[];
                else
                        printf("test %d ok\n",i);
                }
                else
                        printf("test %d ok\n",i);
                }
-       exit(err);
+       printf("test end processing ");
+       for (i=0; i<data_len[3]; i++)
+               {
+               RC4_set_key(&key,keys[3][0],&(keys[3][1]));
+               memset(obuf,0x00,sizeof(obuf));
+               RC4(&key,i,&(data[3][0]),obuf);
+               if ((memcmp(obuf,output[3],i) != 0) || (obuf[i] != 0))
+                       {
+                       printf("error in RC4 length processing\n");
+                       printf("output:");
+                       for (j=0; j<i+1; j++)
+                               printf(" %02x",obuf[j]);
+                       printf("\n");
+                       printf("expect:");
+                       p= &(output[3][0]);
+                       for (j=0; j<i; j++)
+                               printf(" %02x",*(p++));
+                       printf(" 00\n");
+                       err++;
+                       }
+               else
+                       {
+                       printf(".");
+                       fflush(stdout);
+                       }
+               }
+       printf("done\n");
+       printf("test multi-call ");
+       for (i=0; i<data_len[3]; i++)
+               {
+               RC4_set_key(&key,keys[3][0],&(keys[3][1]));
+               memset(obuf,0x00,sizeof(obuf));
+               RC4(&key,i,&(data[3][0]),obuf);
+               RC4(&key,data_len[3]-i,&(data[3][i]),&(obuf[i]));
+               if (memcmp(obuf,output[3],data_len[3]+1) != 0)
+                       {
+                       printf("error in RC4 multi-call processing\n");
+                       printf("output:");
+                       for (j=0; j<data_len[3]+1; j++)
+                               printf(" %02x",obuf[j]);
+                       printf("\n");
+                       printf("expect:");
+                       p= &(output[3][0]);
+                       for (j=0; j<data_len[3]+1; j++)
+                               printf(" %02x",*(p++));
+                       err++;
+                       }
+               else
+                       {
+                       printf(".");
+                       fflush(stdout);
+                       }
+               }
+       printf("done\n");
+       EXIT(err);
        return(0);
        }
        return(0);
        }
-
+#endif