shamodule.c revision e93237dfcc4ee4feee62adafb4e7899487ca864b
129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* SHA module */ 229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* This module provides an interface to NIST's Secure Hash Algorithm */ 429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* See below for information about the original code this module was 629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum based upon. Additional work performed by: 729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 83adefccdcb26ca547e26184018b55e4243a568b1Andrew M. Kuchling Andrew Kuchling (amk@amk.ca) 929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum Greg Stein (gstein@lyra.org) 10f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 11f80578548d46dbe6dad87b8b8f1ac0002bf6aef8Gregory P. Smith Copyright (C) 2005 Gregory P. Smith (greg@krypto.org) 12f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith Licensed to PSF under a Contributor Agreement. 13f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 1429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum*/ 1529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 1629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* SHA objects */ 1729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 1829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#include "Python.h" 19f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith#include "structmember.h" 2029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 2129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 2229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Endianness testing and definitions */ 2329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define TestEndianness(variable) {int i=1; variable=PCT_BIG_ENDIAN;\ 2429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum if (*((char*)&i)==1) variable=PCT_LITTLE_ENDIAN;} 2529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 2629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define PCT_LITTLE_ENDIAN 1 2729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define PCT_BIG_ENDIAN 0 2829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 2929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Some useful types */ 3029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumtypedef unsigned char SHA_BYTE; 3229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#if SIZEOF_INT == 4 3429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumtypedef unsigned int SHA_INT32; /* 32-bit integer */ 3529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#else 3629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* not defined. compilation will die. */ 3729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#endif 3829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* The SHA block size and message digest sizes, in bytes */ 4029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define SHA_BLOCKSIZE 64 4229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define SHA_DIGESTSIZE 20 4329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* The structure for storing SHS info */ 4529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumtypedef struct { 472c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake PyObject_HEAD 482c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHA_INT32 digest[5]; /* Message digest */ 492c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHA_INT32 count_lo, count_hi; /* 64-bit bit count */ 502c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHA_BYTE data[SHA_BLOCKSIZE]; /* SHA data buffer */ 512c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake int Endianness; 522c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake int local; /* unprocessed amount in data */ 5329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} SHAobject; 5429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* When run on a little-endian CPU we need to perform byte reversal on an 5629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum array of longwords. */ 5729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 582c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakestatic void longReverse(SHA_INT32 *buffer, int byteCount, int Endianness) 5929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 6029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_INT32 value; 6129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 6229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum if ( Endianness == PCT_BIG_ENDIAN ) 6329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum return; 6429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 6529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum byteCount /= sizeof(*buffer); 662c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake while (byteCount--) { 6729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum value = *buffer; 6829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum value = ( ( value & 0xFF00FF00L ) >> 8 ) | \ 6929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum ( ( value & 0x00FF00FFL ) << 8 ); 7029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum *buffer++ = ( value << 16 ) | ( value >> 16 ); 7129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 7229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 7329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 742c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakestatic void SHAcopy(SHAobject *src, SHAobject *dest) 7529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 762c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake dest->Endianness = src->Endianness; 772c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake dest->local = src->local; 782c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake dest->count_lo = src->count_lo; 792c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake dest->count_hi = src->count_hi; 802c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake memcpy(dest->digest, src->digest, sizeof(src->digest)); 812c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake memcpy(dest->data, src->data, sizeof(src->data)); 8229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 8329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 8429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 8529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* ------------------------------------------------------------------------ 8629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * 8729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * This code for the SHA algorithm was noted as public domain. The original 8829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * headers are pasted below. 8929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * 9029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * Several changes have been made to make it more compatible with the 9129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * Python environment and desired interface. 9229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * 9329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum */ 9429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 9529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* NIST Secure Hash Algorithm */ 9629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* heavily modified by Uwe Hollerbach <uh@alumni.caltech edu> */ 9729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* from Peter C. Gutmann's implementation as found in */ 9829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Applied Cryptography by Bruce Schneier */ 9929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Further modifications to include the "UNRAVEL" stuff, below */ 10029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 10129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* This code is in the public domain */ 10229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 10329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* UNRAVEL should be fastest & biggest */ 10429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* UNROLL_LOOPS should be just as big, but slightly slower */ 10529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* both undefined should be smallest and slowest */ 10629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 10729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define UNRAVEL 10829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* #define UNROLL_LOOPS */ 10929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 11029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* The SHA f()-functions. The f1 and f3 functions can be optimized to 11129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum save one boolean operation each - thanks to Rich Schroeppel, 11229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum rcs@cs.arizona.edu for discovering this */ 11329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 11429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/*#define f1(x,y,z) ((x & y) | (~x & z)) // Rounds 0-19 */ 11529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define f1(x,y,z) (z ^ (x & (y ^ z))) /* Rounds 0-19 */ 11629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define f2(x,y,z) (x ^ y ^ z) /* Rounds 20-39 */ 11729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/*#define f3(x,y,z) ((x & y) | (x & z) | (y & z)) // Rounds 40-59 */ 11829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define f3(x,y,z) ((x & y) | (z & (x | y))) /* Rounds 40-59 */ 11929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define f4(x,y,z) (x ^ y ^ z) /* Rounds 60-79 */ 12029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 12129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* SHA constants */ 12229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 12329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define CONST1 0x5a827999L /* Rounds 0-19 */ 12429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define CONST2 0x6ed9eba1L /* Rounds 20-39 */ 12529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define CONST3 0x8f1bbcdcL /* Rounds 40-59 */ 12629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define CONST4 0xca62c1d6L /* Rounds 60-79 */ 12729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 12829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* 32-bit rotate */ 12929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 13029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define R32(x,n) ((x << n) | (x >> (32 - n))) 13129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 13229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* the generic case, for when the overall rotation is not unraveled */ 13329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 13429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FG(n) \ 13529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum T = R32(A,5) + f##n(B,C,D) + E + *WP++ + CONST##n; \ 13629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum E = D; D = C; C = R32(B,30); B = A; A = T 13729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 13829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* specific cases, for when the overall rotation is unraveled */ 13929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 14029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FA(n) \ 14129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum T = R32(A,5) + f##n(B,C,D) + E + *WP++ + CONST##n; B = R32(B,30) 14229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 14329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FB(n) \ 14429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum E = R32(T,5) + f##n(A,B,C) + D + *WP++ + CONST##n; A = R32(A,30) 14529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 14629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FC(n) \ 14729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum D = R32(E,5) + f##n(T,A,B) + C + *WP++ + CONST##n; T = R32(T,30) 14829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 14929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FD(n) \ 15029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum C = R32(D,5) + f##n(E,T,A) + B + *WP++ + CONST##n; E = R32(E,30) 15129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 15229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FE(n) \ 15329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum B = R32(C,5) + f##n(D,E,T) + A + *WP++ + CONST##n; D = R32(D,30) 15429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 15529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#define FT(n) \ 15629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum A = R32(B,5) + f##n(C,D,E) + T + *WP++ + CONST##n; C = R32(C,30) 15729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 15829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* do SHA transformation */ 15929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 16029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic void 1612c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakesha_transform(SHAobject *sha_info) 16229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 16329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum int i; 16429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_INT32 T, A, B, C, D, E, W[80], *WP; 16529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 16629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum memcpy(W, sha_info->data, sizeof(sha_info->data)); 167ff1ccbfc2172c564c6ca93113e1a1cf9a271f6efGuido van Rossum longReverse(W, (int)sizeof(sha_info->data), sha_info->Endianness); 16829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 16929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum for (i = 16; i < 80; ++i) { 17029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum W[i] = W[i-3] ^ W[i-8] ^ W[i-14] ^ W[i-16]; 17129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 17229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum /* extra rotation fix */ 17329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum W[i] = R32(W[i], 1); 17429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 17529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum A = sha_info->digest[0]; 17629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum B = sha_info->digest[1]; 17729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum C = sha_info->digest[2]; 17829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum D = sha_info->digest[3]; 17929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum E = sha_info->digest[4]; 18029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum WP = W; 18129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#ifdef UNRAVEL 18229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FA(1); FB(1); FC(1); FD(1); FE(1); FT(1); FA(1); FB(1); FC(1); FD(1); 18329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FE(1); FT(1); FA(1); FB(1); FC(1); FD(1); FE(1); FT(1); FA(1); FB(1); 18429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FC(2); FD(2); FE(2); FT(2); FA(2); FB(2); FC(2); FD(2); FE(2); FT(2); 18529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FA(2); FB(2); FC(2); FD(2); FE(2); FT(2); FA(2); FB(2); FC(2); FD(2); 18629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FE(3); FT(3); FA(3); FB(3); FC(3); FD(3); FE(3); FT(3); FA(3); FB(3); 18729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FC(3); FD(3); FE(3); FT(3); FA(3); FB(3); FC(3); FD(3); FE(3); FT(3); 18829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FA(4); FB(4); FC(4); FD(4); FE(4); FT(4); FA(4); FB(4); FC(4); FD(4); 18929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FE(4); FT(4); FA(4); FB(4); FC(4); FD(4); FE(4); FT(4); FA(4); FB(4); 19029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[0] += E; 19129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[1] += T; 19229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[2] += A; 19329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[3] += B; 19429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[4] += C; 19529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#else /* !UNRAVEL */ 19629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#ifdef UNROLL_LOOPS 19729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); 19829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); FG(1); 19929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); 20029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); FG(2); 20129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); 20229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); FG(3); 20329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); 20429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); FG(4); 20529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#else /* !UNROLL_LOOPS */ 20629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum for (i = 0; i < 20; ++i) { FG(1); } 20729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum for (i = 20; i < 40; ++i) { FG(2); } 20829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum for (i = 40; i < 60; ++i) { FG(3); } 20929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum for (i = 60; i < 80; ++i) { FG(4); } 21029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#endif /* !UNROLL_LOOPS */ 21129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[0] += A; 21229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[1] += B; 21329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[2] += C; 21429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[3] += D; 21529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[4] += E; 21629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum#endif /* !UNRAVEL */ 21729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 21829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 21929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* initialize the SHA digest */ 22029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 22129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic void 2222c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakesha_init(SHAobject *sha_info) 22329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 22429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum TestEndianness(sha_info->Endianness) 22529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 22629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[0] = 0x67452301L; 22729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[1] = 0xefcdab89L; 22829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[2] = 0x98badcfeL; 22929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[3] = 0x10325476L; 23029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->digest[4] = 0xc3d2e1f0L; 23129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->count_lo = 0L; 23229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->count_hi = 0L; 23329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->local = 0; 23429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 23529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 23629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* update the SHA digest */ 23729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 23829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic void 2392c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakesha_update(SHAobject *sha_info, SHA_BYTE *buffer, int count) 24029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 24129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum int i; 24229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_INT32 clo; 24329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 24429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum clo = sha_info->count_lo + ((SHA_INT32) count << 3); 24529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum if (clo < sha_info->count_lo) { 2462c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake ++sha_info->count_hi; 24729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 24829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->count_lo = clo; 24929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->count_hi += (SHA_INT32) count >> 29; 25029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum if (sha_info->local) { 2512c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake i = SHA_BLOCKSIZE - sha_info->local; 2522c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (i > count) { 2532c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake i = count; 2542c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 2552c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake memcpy(((SHA_BYTE *) sha_info->data) + sha_info->local, buffer, i); 2562c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake count -= i; 2572c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake buffer += i; 2582c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_info->local += i; 2592c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (sha_info->local == SHA_BLOCKSIZE) { 2602c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_transform(sha_info); 2612c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 2622c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake else { 2632c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return; 2642c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 26529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 26629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum while (count >= SHA_BLOCKSIZE) { 2672c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake memcpy(sha_info->data, buffer, SHA_BLOCKSIZE); 2682c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake buffer += SHA_BLOCKSIZE; 2692c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake count -= SHA_BLOCKSIZE; 2702c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_transform(sha_info); 27129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 27229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum memcpy(sha_info->data, buffer, count); 27329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->local = count; 27429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 27529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 27629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* finish computing the SHA digest */ 27729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 27829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic void 2792c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drakesha_final(unsigned char digest[20], SHAobject *sha_info) 28029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 28129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum int count; 28229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_INT32 lo_bit_count, hi_bit_count; 28329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 28429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum lo_bit_count = sha_info->count_lo; 28529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum hi_bit_count = sha_info->count_hi; 28629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum count = (int) ((lo_bit_count >> 3) & 0x3f); 28729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum ((SHA_BYTE *) sha_info->data)[count++] = 0x80; 2882c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (count > SHA_BLOCKSIZE - 8) { 28929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum memset(((SHA_BYTE *) sha_info->data) + count, 0, 29029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_BLOCKSIZE - count); 29129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_transform(sha_info); 29229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum memset((SHA_BYTE *) sha_info->data, 0, SHA_BLOCKSIZE - 8); 29329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 2942c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake else { 29529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum memset(((SHA_BYTE *) sha_info->data) + count, 0, 29629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum SHA_BLOCKSIZE - 8 - count); 29729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum } 29829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 29929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum /* GJS: note that we add the hi/lo in big-endian. sha_transform will 30029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum swap these values into host-order. */ 30129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[56] = (hi_bit_count >> 24) & 0xff; 30229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[57] = (hi_bit_count >> 16) & 0xff; 30329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[58] = (hi_bit_count >> 8) & 0xff; 30429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[59] = (hi_bit_count >> 0) & 0xff; 30529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[60] = (lo_bit_count >> 24) & 0xff; 30629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[61] = (lo_bit_count >> 16) & 0xff; 30729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[62] = (lo_bit_count >> 8) & 0xff; 30829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_info->data[63] = (lo_bit_count >> 0) & 0xff; 30929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum sha_transform(sha_info); 31029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 0] = (unsigned char) ((sha_info->digest[0] >> 24) & 0xff); 31129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 1] = (unsigned char) ((sha_info->digest[0] >> 16) & 0xff); 31229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 2] = (unsigned char) ((sha_info->digest[0] >> 8) & 0xff); 31329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 3] = (unsigned char) ((sha_info->digest[0] ) & 0xff); 31429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 4] = (unsigned char) ((sha_info->digest[1] >> 24) & 0xff); 31529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 5] = (unsigned char) ((sha_info->digest[1] >> 16) & 0xff); 31629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 6] = (unsigned char) ((sha_info->digest[1] >> 8) & 0xff); 31729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 7] = (unsigned char) ((sha_info->digest[1] ) & 0xff); 31829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 8] = (unsigned char) ((sha_info->digest[2] >> 24) & 0xff); 31929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[ 9] = (unsigned char) ((sha_info->digest[2] >> 16) & 0xff); 32029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[10] = (unsigned char) ((sha_info->digest[2] >> 8) & 0xff); 32129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[11] = (unsigned char) ((sha_info->digest[2] ) & 0xff); 32229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[12] = (unsigned char) ((sha_info->digest[3] >> 24) & 0xff); 32329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[13] = (unsigned char) ((sha_info->digest[3] >> 16) & 0xff); 32429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[14] = (unsigned char) ((sha_info->digest[3] >> 8) & 0xff); 32529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[15] = (unsigned char) ((sha_info->digest[3] ) & 0xff); 32629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[16] = (unsigned char) ((sha_info->digest[4] >> 24) & 0xff); 32729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[17] = (unsigned char) ((sha_info->digest[4] >> 16) & 0xff); 32829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[18] = (unsigned char) ((sha_info->digest[4] >> 8) & 0xff); 32929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum digest[19] = (unsigned char) ((sha_info->digest[4] ) & 0xff); 33029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 33129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 33229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* 33329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * End of copied SHA code. 33429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * 33529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum * ------------------------------------------------------------------------ 33629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum */ 33729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 338938ace69a0e112424a2f426a4881d1fd1fc922d2Jeremy Hyltonstatic PyTypeObject SHAtype; 33929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 34029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 34129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic SHAobject * 342f3f33dcf03eaed3c4e720178f9d69205a66d6a91Thomas WoutersnewSHAobject(void) 34329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 3442c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return (SHAobject *)PyObject_New(SHAobject, &SHAtype); 34529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 34629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 34729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Internal methods for a hashing object */ 34829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 34929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic void 3502c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred DrakeSHA_dealloc(PyObject *ptr) 35129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 3522c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake PyObject_Del(ptr); 35329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 35429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 35529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 35629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* External methods for a hashing object */ 35729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 35814f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. LöwisPyDoc_STRVAR(SHA_copy__doc__, "Return a copy of the hashing object."); 35929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 36029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 36196a8c3954cbdb186bc567a490dad8987508ce268Georg BrandlSHA_copy(SHAobject *self, PyObject *unused) 36229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 3632c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAobject *newobj; 36429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3652c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if ( (newobj = newSHAobject())==NULL) 3662c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return NULL; 36729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3682c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAcopy(self, newobj); 3692c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return (PyObject *)newobj; 37029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 37129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 37214f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. LöwisPyDoc_STRVAR(SHA_digest__doc__, 37314f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwis"Return the digest value as a string of binary data."); 37429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 37529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 37696a8c3954cbdb186bc567a490dad8987508ce268Georg BrandlSHA_digest(SHAobject *self, PyObject *unused) 37729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 3782c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake unsigned char digest[SHA_DIGESTSIZE]; 3792c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAobject temp; 38029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 3812c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAcopy(self, &temp); 3822c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_final(digest, &temp); 3832c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return PyString_FromStringAndSize((const char *)digest, sizeof(digest)); 38429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 38529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 38614f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. LöwisPyDoc_STRVAR(SHA_hexdigest__doc__, 38714f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwis"Return the digest value as a string of hexadecimal digits."); 38829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 38929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 39096a8c3954cbdb186bc567a490dad8987508ce268Georg BrandlSHA_hexdigest(SHAobject *self, PyObject *unused) 39129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 3922c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake unsigned char digest[SHA_DIGESTSIZE]; 3932c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAobject temp; 3942c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake PyObject *retval; 3952c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake char *hex_digest; 3962c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake int i, j; 3972c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 3982c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake /* Get the raw (binary) digest value */ 3992c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAcopy(self, &temp); 4002c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_final(digest, &temp); 4012c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 4022c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake /* Create a new string */ 4032c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake retval = PyString_FromStringAndSize(NULL, sizeof(digest) * 2); 40457b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw if (!retval) 40557b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw return NULL; 4062c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake hex_digest = PyString_AsString(retval); 40757b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw if (!hex_digest) { 40857b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw Py_DECREF(retval); 40957b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw return NULL; 41057b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw } 4112c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 4122c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake /* Make hex version of the digest */ 4132c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake for(i=j=0; i<sizeof(digest); i++) { 4142c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake char c; 41557b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw c = (digest[i] >> 4) & 0xf; 41657b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw c = (c>9) ? c+'a'-10 : c + '0'; 4172c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake hex_digest[j++] = c; 41857b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw c = (digest[i] & 0xf); 41957b808d21a7651b303bf22264c4bc47824ab3b51Barry Warsaw c = (c>9) ? c+'a'-10 : c + '0'; 4202c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake hex_digest[j++] = c; 4212c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 4222c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return retval; 42329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 42429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 42514f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. LöwisPyDoc_STRVAR(SHA_update__doc__, 42614f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwis"Update this hashing object's state with the provided string."); 42729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 42829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 4292c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred DrakeSHA_update(SHAobject *self, PyObject *args) 43029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 4312c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake unsigned char *cp; 4322c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake int len; 43329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4342c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (!PyArg_ParseTuple(args, "s#:update", &cp, &len)) 4352c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return NULL; 43629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4372c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_update(self, cp, len); 43829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 4392c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake Py_INCREF(Py_None); 4402c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return Py_None; 44129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 44229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 44329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyMethodDef SHA_methods[] = { 44496a8c3954cbdb186bc567a490dad8987508ce268Georg Brandl {"copy", (PyCFunction)SHA_copy, METH_NOARGS, SHA_copy__doc__}, 44596a8c3954cbdb186bc567a490dad8987508ce268Georg Brandl {"digest", (PyCFunction)SHA_digest, METH_NOARGS, SHA_digest__doc__}, 44696a8c3954cbdb186bc567a490dad8987508ce268Georg Brandl {"hexdigest", (PyCFunction)SHA_hexdigest, METH_NOARGS, SHA_hexdigest__doc__}, 4472c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake {"update", (PyCFunction)SHA_update, METH_VARARGS, SHA_update__doc__}, 4482c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake {NULL, NULL} /* sentinel */ 44929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum}; 45029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 45129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 452f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. SmithSHA_get_block_size(PyObject *self, void *closure) 45329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 454f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith return PyInt_FromLong(SHA_BLOCKSIZE); 455f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith} 4562c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 457f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smithstatic PyObject * 458f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. SmithSHA_get_digest_size(PyObject *self, void *closure) 459f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith{ 460f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith return PyInt_FromLong(SHA_DIGESTSIZE); 46129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 46229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 463f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smithstatic PyObject * 464f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. SmithSHA_get_name(PyObject *self, void *closure) 465f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith{ 466f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith return PyString_FromStringAndSize("SHA1", 4); 467f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith} 468f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 469f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smithstatic PyGetSetDef SHA_getseters[] = { 470f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith {"digest_size", 471f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith (getter)SHA_get_digest_size, NULL, 472f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL, 473f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL}, 474f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith {"block_size", 475f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith (getter)SHA_get_block_size, NULL, 476f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL, 477f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL}, 478f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith {"name", 479f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith (getter)SHA_get_name, NULL, 480f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL, 481f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL}, 482f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith /* the old md5 and sha modules support 'digest_size' as in PEP 247. 483f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith * the old sha module also supported 'digestsize'. ugh. */ 484f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith {"digestsize", 485f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith (getter)SHA_get_digest_size, NULL, 486f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL, 487f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith NULL}, 488f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith {NULL} /* Sentinel */ 489f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith}; 490f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 49129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyTypeObject SHAtype = { 4926819210b9e4e5719a6f7f9c1725f8fa70a8936f6Martin v. Löwis PyVarObject_HEAD_INIT(NULL, 0) 493f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith "_sha.sha", /*tp_name*/ 4942c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sizeof(SHAobject), /*tp_size*/ 4952c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 0, /*tp_itemsize*/ 4962c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake /* methods */ 4972c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHA_dealloc, /*tp_dealloc*/ 4982c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake 0, /*tp_print*/ 499f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_getattr*/ 500f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_setattr*/ 501f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_compare*/ 502f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_repr*/ 503f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_as_number*/ 504f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_as_sequence*/ 505f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_as_mapping*/ 506f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_hash*/ 507f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_call*/ 508f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_str*/ 509f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_getattro*/ 510f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_setattro*/ 511f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_as_buffer*/ 512f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith Py_TPFLAGS_DEFAULT, /*tp_flags*/ 513f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_doc*/ 514f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_traverse*/ 515f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_clear*/ 516f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_richcompare*/ 517f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_weaklistoffset*/ 518f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_iter*/ 519f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /*tp_iternext*/ 520f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith SHA_methods, /* tp_methods */ 521f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith 0, /* tp_members */ 522f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith SHA_getseters, /* tp_getset */ 52329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum}; 52429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 52529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 52629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* The single module-level function: new() */ 52729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 52814f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. LöwisPyDoc_STRVAR(SHA_new__doc__, 52914f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwis"Return a new SHA hashing object. An optional string argument\n\ 53014f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwismay be provided; if present, this string will be automatically\n\ 53114f8b4cfcb98de74b9c6e9316539be9e2a5cd31fMartin v. Löwishashed."); 53229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 53329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic PyObject * 5342c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred DrakeSHA_new(PyObject *self, PyObject *args, PyObject *kwdict) 53529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 53615e62742fad688b026ba80bf17d1345c4cbd423bMartin v. Löwis static char *kwlist[] = {"string", NULL}; 5372c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake SHAobject *new; 5382c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake unsigned char *cp = NULL; 5392c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake int len; 54029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5412c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (!PyArg_ParseTupleAndKeywords(args, kwdict, "|s#:new", kwlist, 5422c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake &cp, &len)) { 5432c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return NULL; 5442c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 54529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 54655087f0c351d6de453a5c95293792051d899f16bJeremy Hylton if ((new = newSHAobject()) == NULL) 54755087f0c351d6de453a5c95293792051d899f16bJeremy Hylton return NULL; 54855087f0c351d6de453a5c95293792051d899f16bJeremy Hylton 5492c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_init(new); 55029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5512c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (PyErr_Occurred()) { 5522c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake Py_DECREF(new); 5532c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return NULL; 5542c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake } 5552c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake if (cp) 5562c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake sha_update(new, cp, len); 55729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5582c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake return (PyObject *)new; 55929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 56029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 56129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 56229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* List of functions exported by this module */ 56329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 56429d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossumstatic struct PyMethodDef SHA_functions[] = { 5652c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake {"new", (PyCFunction)SHA_new, METH_VARARGS|METH_KEYWORDS, SHA_new__doc__}, 5662c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake {NULL, NULL} /* Sentinel */ 56729d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum}; 56829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 56929d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 57029d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum/* Initialize this module. */ 57129d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5728b14ac98986c585d301eb8d2782602c43a7e54ddFred Drake#define insint(n,v) { PyModule_AddIntConstant(m,n,v); } 57329d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 574fe51c6d66e0fbf6a142036bee2c448bd7fe8fefcMark HammondPyMODINIT_FUNC 575f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smithinit_sha(void) 57629d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum{ 577d63e504d333fc325ed4d7951fcfc2afb3d5aa910Fred Drake PyObject *m; 57829d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 579e93237dfcc4ee4feee62adafb4e7899487ca864bChristian Heimes Py_TYPE(&SHAtype) = &PyType_Type; 580f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith if (PyType_Ready(&SHAtype) < 0) 581f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith return; 582f21a5f773964d34c7b6deb7e3d753fae2b9c70e2Gregory P. Smith m = Py_InitModule("_sha", SHA_functions); 5831ac754fa10f5d199d19911e21185d0970cb3073fNeal Norwitz if (m == NULL) 5841ac754fa10f5d199d19911e21185d0970cb3073fNeal Norwitz return; 58529d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum 5862c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake /* Add some symbolic constants to the module */ 5872c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake insint("blocksize", 1); /* For future use, in case some hash 5882c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake functions require an integral number of 5892c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake blocks */ 5902c4a3dceafeadd5821e36a6e1b7fc6b7d81338f0Fred Drake insint("digestsize", 20); 59175fec2c8edd4cc673abf44d032ede17a63657ec1Andrew M. Kuchling insint("digest_size", 20); 59229d2acc17096bd63d882a87591d336ea5682eab9Guido van Rossum} 593