Annotation of embedaddon/rsync/lib/mdfour.c, revision 1.1

1.1     ! misho       1: /* 
        !             2:  * Unix SMB/Netbios implementation.
        !             3:  * Version 1.9.
        !             4:  * An implementation of MD4 designed for use in the SMB authentication protocol.
        !             5:  *
        !             6:  * Copyright (C) 1997-1998 Andrew Tridgell
        !             7:  * Copyright (C) 2005-2008 Wayne Davison
        !             8:  * 
        !             9:  * This program is free software; you can redistribute it and/or modify
        !            10:  * it under the terms of the GNU General Public License as published by
        !            11:  * the Free Software Foundation; either version 3 of the License, or
        !            12:  * (at your option) any later version.
        !            13:  * 
        !            14:  * This program is distributed in the hope that it will be useful,
        !            15:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
        !            16:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        !            17:  * GNU General Public License for more details.
        !            18:  * 
        !            19:  * You should have received a copy of the GNU General Public License along
        !            20:  * with this program; if not, visit the http://fsf.org website.
        !            21:  */
        !            22: 
        !            23: #include "rsync.h"
        !            24: 
        !            25: /* NOTE: This code makes no attempt to be fast! 
        !            26:  *
        !            27:  * It assumes that a int is at least 32 bits long. */
        !            28: 
        !            29: static md_context *m;
        !            30: 
        !            31: #define MASK32 (0xffffffff)
        !            32: 
        !            33: #define F(X,Y,Z) ((((X)&(Y)) | ((~(X))&(Z))))
        !            34: #define G(X,Y,Z) ((((X)&(Y)) | ((X)&(Z)) | ((Y)&(Z))))
        !            35: #define H(X,Y,Z) (((X)^(Y)^(Z)))
        !            36: #define lshift(x,s) (((((x)<<(s))&MASK32) | (((x)>>(32-(s)))&MASK32)))
        !            37: 
        !            38: #define ROUND1(a,b,c,d,k,s) a = lshift((a + F(b,c,d) + M[k])&MASK32, s)
        !            39: #define ROUND2(a,b,c,d,k,s) a = lshift((a + G(b,c,d) + M[k] + 0x5A827999)&MASK32,s)
        !            40: #define ROUND3(a,b,c,d,k,s) a = lshift((a + H(b,c,d) + M[k] + 0x6ED9EBA1)&MASK32,s)
        !            41: 
        !            42: /* this applies md4 to 64 byte chunks */
        !            43: static void mdfour64(uint32 *M)
        !            44: {
        !            45:        uint32 AA, BB, CC, DD;
        !            46:        uint32 A,B,C,D;
        !            47: 
        !            48:        A = m->A; B = m->B; C = m->C; D = m->D; 
        !            49:        AA = A; BB = B; CC = C; DD = D;
        !            50: 
        !            51:        ROUND1(A,B,C,D,  0,  3);  ROUND1(D,A,B,C,  1,  7);  
        !            52:        ROUND1(C,D,A,B,  2, 11);  ROUND1(B,C,D,A,  3, 19);
        !            53:        ROUND1(A,B,C,D,  4,  3);  ROUND1(D,A,B,C,  5,  7);  
        !            54:        ROUND1(C,D,A,B,  6, 11);  ROUND1(B,C,D,A,  7, 19);
        !            55:        ROUND1(A,B,C,D,  8,  3);  ROUND1(D,A,B,C,  9,  7);  
        !            56:        ROUND1(C,D,A,B, 10, 11);  ROUND1(B,C,D,A, 11, 19);
        !            57:        ROUND1(A,B,C,D, 12,  3);  ROUND1(D,A,B,C, 13,  7);  
        !            58:        ROUND1(C,D,A,B, 14, 11);  ROUND1(B,C,D,A, 15, 19);      
        !            59: 
        !            60:        ROUND2(A,B,C,D,  0,  3);  ROUND2(D,A,B,C,  4,  5);  
        !            61:        ROUND2(C,D,A,B,  8,  9);  ROUND2(B,C,D,A, 12, 13);
        !            62:        ROUND2(A,B,C,D,  1,  3);  ROUND2(D,A,B,C,  5,  5);  
        !            63:        ROUND2(C,D,A,B,  9,  9);  ROUND2(B,C,D,A, 13, 13);
        !            64:        ROUND2(A,B,C,D,  2,  3);  ROUND2(D,A,B,C,  6,  5);  
        !            65:        ROUND2(C,D,A,B, 10,  9);  ROUND2(B,C,D,A, 14, 13);
        !            66:        ROUND2(A,B,C,D,  3,  3);  ROUND2(D,A,B,C,  7,  5);  
        !            67:        ROUND2(C,D,A,B, 11,  9);  ROUND2(B,C,D,A, 15, 13);
        !            68: 
        !            69:        ROUND3(A,B,C,D,  0,  3);  ROUND3(D,A,B,C,  8,  9);  
        !            70:        ROUND3(C,D,A,B,  4, 11);  ROUND3(B,C,D,A, 12, 15);
        !            71:        ROUND3(A,B,C,D,  2,  3);  ROUND3(D,A,B,C, 10,  9);  
        !            72:        ROUND3(C,D,A,B,  6, 11);  ROUND3(B,C,D,A, 14, 15);
        !            73:        ROUND3(A,B,C,D,  1,  3);  ROUND3(D,A,B,C,  9,  9);  
        !            74:        ROUND3(C,D,A,B,  5, 11);  ROUND3(B,C,D,A, 13, 15);
        !            75:        ROUND3(A,B,C,D,  3,  3);  ROUND3(D,A,B,C, 11,  9);  
        !            76:        ROUND3(C,D,A,B,  7, 11);  ROUND3(B,C,D,A, 15, 15);
        !            77: 
        !            78:        A += AA; B += BB; 
        !            79:        C += CC; D += DD;
        !            80:        
        !            81:        A &= MASK32; B &= MASK32; 
        !            82:        C &= MASK32; D &= MASK32;
        !            83: 
        !            84:        m->A = A; m->B = B; m->C = C; m->D = D;
        !            85: }
        !            86: 
        !            87: static void copy64(uint32 *M, const uchar *in)
        !            88: {
        !            89:        int i;
        !            90: 
        !            91:        for (i = 0; i < MD4_DIGEST_LEN; i++) {
        !            92:                M[i] = (in[i*4+3] << 24) | (in[i*4+2] << 16)
        !            93:                     | (in[i*4+1] << 8)  | (in[i*4+0] << 0);
        !            94:        }
        !            95: }
        !            96: 
        !            97: static void copy4(uchar *out,uint32 x)
        !            98: {
        !            99:        out[0] = x&0xFF;
        !           100:        out[1] = (x>>8)&0xFF;
        !           101:        out[2] = (x>>16)&0xFF;
        !           102:        out[3] = (x>>24)&0xFF;
        !           103: }
        !           104: 
        !           105: void mdfour_begin(md_context *md)
        !           106: {
        !           107:        md->A = 0x67452301;
        !           108:        md->B = 0xefcdab89;
        !           109:        md->C = 0x98badcfe;
        !           110:        md->D = 0x10325476;
        !           111:        md->totalN = 0;
        !           112:        md->totalN2 = 0;
        !           113: }
        !           114: 
        !           115: static void mdfour_tail(const uchar *in, uint32 length)
        !           116: {
        !           117:        uchar buf[128];
        !           118:        uint32 M[16];
        !           119:        extern int protocol_version;
        !           120: 
        !           121:        /*
        !           122:         * Count total number of bits, modulo 2^64
        !           123:         */
        !           124:        m->totalN += length << 3;
        !           125:        if (m->totalN < (length << 3))
        !           126:                m->totalN2++;
        !           127:        m->totalN2 += length >> 29;
        !           128: 
        !           129:        memset(buf, 0, 128);
        !           130:        if (length)
        !           131:                memcpy(buf, in, length);
        !           132:        buf[length] = 0x80;
        !           133: 
        !           134:        if (length <= 55) {
        !           135:                copy4(buf+56, m->totalN);
        !           136:                /*
        !           137:                 * Prior to protocol version 27 only the number of bits
        !           138:                 * modulo 2^32 was included.  MD4 requires the number
        !           139:                 * of bits modulo 2^64, which was fixed starting with
        !           140:                 * protocol version 27.
        !           141:                 */
        !           142:                if (protocol_version >= 27)
        !           143:                        copy4(buf+60, m->totalN2);
        !           144:                copy64(M, buf);
        !           145:                mdfour64(M);
        !           146:        } else {
        !           147:                copy4(buf+120, m->totalN); 
        !           148:                /*
        !           149:                 * Prior to protocol version 27 only the number of bits
        !           150:                 * modulo 2^32 was included.  MD4 requires the number
        !           151:                 * of bits modulo 2^64, which was fixed starting with
        !           152:                 * protocol version 27.
        !           153:                 */
        !           154:                if (protocol_version >= 27)
        !           155:                        copy4(buf+124, m->totalN2); 
        !           156:                copy64(M, buf);
        !           157:                mdfour64(M);
        !           158:                copy64(M, buf+64);
        !           159:                mdfour64(M);
        !           160:        }
        !           161: }
        !           162: 
        !           163: void mdfour_update(md_context *md, const uchar *in, uint32 length)
        !           164: {
        !           165:        uint32 M[16];
        !           166: 
        !           167:        m = md;
        !           168: 
        !           169:        if (length == 0)
        !           170:                mdfour_tail(in, length);
        !           171: 
        !           172:        while (length >= 64) {
        !           173:                copy64(M, in);
        !           174:                mdfour64(M);
        !           175:                in += 64;
        !           176:                length -= 64;
        !           177:                m->totalN += 64 << 3;
        !           178:                if (m->totalN < 64 << 3)
        !           179:                        m->totalN2++;
        !           180:        }
        !           181: 
        !           182:        if (length)
        !           183:                mdfour_tail(in, length);
        !           184: }
        !           185: 
        !           186: void mdfour_result(md_context *md, uchar digest[MD4_DIGEST_LEN])
        !           187: {
        !           188:        m = md;
        !           189: 
        !           190:        copy4(digest, m->A);
        !           191:        copy4(digest+4, m->B);
        !           192:        copy4(digest+8, m->C);
        !           193:        copy4(digest+12, m->D);
        !           194: }
        !           195: 
        !           196: void mdfour(uchar digest[MD4_DIGEST_LEN], uchar *in, int length)
        !           197: {
        !           198:        md_context md;
        !           199:        mdfour_begin(&md);
        !           200:        mdfour_update(&md, in, length);
        !           201:        mdfour_result(&md, digest);
        !           202: }
        !           203: 
        !           204: #ifdef TEST_MDFOUR
        !           205: int protocol_version = 28;
        !           206: 
        !           207: static void file_checksum1(char *fname)
        !           208: {
        !           209:        int fd, i, was_multiple_of_64 = 1;
        !           210:        md_context md;
        !           211:        uchar buf[64*1024], sum[MD4_DIGEST_LEN];
        !           212:        
        !           213:        fd = open(fname,O_RDONLY);
        !           214:        if (fd == -1) {
        !           215:                perror("fname");
        !           216:                exit(1);
        !           217:        }
        !           218:        
        !           219:        mdfour_begin(&md);
        !           220: 
        !           221:        while (1) {
        !           222:                int n = read(fd, buf, sizeof buf);
        !           223:                if (n <= 0)
        !           224:                        break;
        !           225:                was_multiple_of_64 = !(n % 64);
        !           226:                mdfour_update(&md, buf, n);
        !           227:        }
        !           228:        if (was_multiple_of_64 && protocol_version >= 27)
        !           229:                mdfour_update(&md, buf, 0);
        !           230: 
        !           231:        close(fd);
        !           232: 
        !           233:        mdfour_result(&md, sum);
        !           234: 
        !           235:        for (i = 0; i < MD4_DIGEST_LEN; i++)
        !           236:                printf("%02X", sum[i]);
        !           237:        printf("\n");
        !           238: }
        !           239: 
        !           240:  int main(int argc, char *argv[])
        !           241: {
        !           242:        while (--argc)
        !           243:                file_checksum1(*++argv);
        !           244:        return 0;
        !           245: }
        !           246: #endif

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>