src/cmsmd5.c

   1 //---------------------------------------------------------------------------------
   2 //
   3 //  Little Color Management System
   4 //  Copyright (c) 1998-2010 Marti Maria Saguer
   5 //
   6 // Permission is hereby granted, free of charge, to any person obtaining
   7 // a copy of this software and associated documentation files (the "Software"),
   8 // to deal in the Software without restriction, including without limitation
   9 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10 // and/or sell copies of the Software, and to permit persons to whom the Software
  11 // is furnished to do so, subject to the following conditions:
  12 //
  13 // The above copyright notice and this permission notice shall be included in
  14 // all copies or substantial portions of the Software.
  15 //
  16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  17 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO
  18 // THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  20 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  21 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  22 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23 //
  24 //---------------------------------------------------------------------------------
  25
  26
  27 #include "lcms2_internal.h"
  28
  29 #ifdef CMS_USE_BIG_ENDIAN
  30
  31 static
  32 void byteReverse(cmsUInt8Number * buf, cmsUInt32Number longs)
  33 {
  34     do {
  35
  36         cmsUInt32Number t = _cmsAdjustEndianess32(*(cmsUInt32Number *) buf);
  37         *(cmsUInt32Number *) buf = t;
  38         buf += sizeof(cmsUInt32Number);
  39
  40     } while (--longs);
  41
  42 }
  43
  44 #else
  45 #define byteReverse(buf, len)
  46 #endif
  47
  48
  49 typedef struct {
  50
  51     cmsUInt32Number buf[4];
  52     cmsUInt32Number bits[2];
  53     cmsUInt8Number in[64];
  54     cmsContext ContextID;
  55
  56 } _cmsMD5;
  57
  58 #define F1(x, y, z) (z ^ (x & (y ^ z)))
  59 #define F2(x, y, z) F1(z, x, y)
  60 #define F3(x, y, z) (x ^ y ^ z)
  61 #define F4(x, y, z) (y ^ (x | ~z))
  62
  63 #define STEP(f, w, x, y, z, data, s) \
  64     ( w += f(x, y, z) + data,  w = w<<s | w>>(32-s),  w += x )
  65
  66
  67 static
  68 void MD5_Transform(cmsUInt32Number buf[4], cmsUInt32Number in[16])
  69
  70 {
  71     register cmsUInt32Number a, b, c, d;
  72
  73     a = buf[0];
  74     b = buf[1];
  75     c = buf[2];
  76     d = buf[3];
  77
  78     STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
  79     STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
  80     STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
  81     STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
  82     STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
  83     STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
  84     STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
  85     STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
  86     STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
  87     STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
  88     STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
  89     STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
  90     STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
  91     STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
  92     STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
  93     STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
  94
  95     STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
  96     STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
  97     STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
  98     STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
  99     STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
 100     STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
 101     STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
 102     STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
 103     STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
 104     STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
 105     STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
 106     STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
 107     STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
 108     STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
 109     STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
 110     STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
 111
 112     STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
 113     STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
 114     STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
 115     STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
 116     STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
 117     STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
 118     STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
 119     STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
 120     STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
 121     STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
 122     STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
 123     STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
 124     STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
 125     STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
 126     STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
 127     STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
 128
 129     STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
 130     STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
 131     STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
 132     STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
 133     STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
 134     STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
 135     STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
 136     STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
 137     STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
 138     STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
 139     STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
 140     STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
 141     STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
 142     STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
 143     STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
 144     STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
 145
 146     buf[0] += a;
 147     buf[1] += b;
 148     buf[2] += c;
 149     buf[3] += d;
 150 }
 151
 152
 153 // Create a MD5 object
 154 static
 155 cmsHANDLE  MD5alloc(cmsContext ContextID)
 156 {
 157     _cmsMD5* ctx = (_cmsMD5*) _cmsMallocZero(ContextID, sizeof(_cmsMD5));
 158     if (ctx == NULL) return NULL;
 159
 160     ctx ->ContextID = ContextID;
 161
 162     ctx->buf[0] = 0x67452301;
 163     ctx->buf[1] = 0xefcdab89;
 164     ctx->buf[2] = 0x98badcfe;
 165     ctx->buf[3] = 0x10325476;
 166
 167     ctx->bits[0] = 0;
 168     ctx->bits[1] = 0;
 169
 170     return (cmsHANDLE) ctx;
 171 }
 172
 173
 174 static
 175 void MD5add(cmsHANDLE Handle, cmsUInt8Number* buf, cmsUInt32Number len)
 176 {
 177     _cmsMD5* ctx = (_cmsMD5*) Handle;
 178     cmsUInt32Number t;
 179
 180     t = ctx->bits[0];
 181     if ((ctx->bits[0] = t + (len << 3)) < t)
 182         ctx->bits[1]++;
 183
 184     ctx->bits[1] += len >> 29;
 185
 186     t = (t >> 3) & 0x3f;
 187
 188     if (t) {
 189
 190         cmsUInt8Number *p = (cmsUInt8Number *) ctx->in + t;
 191
 192         t = 64 - t;
 193         if (len < t) {
 194             memmove(p, buf, len);
 195             return;
 196         }
 197
 198         memmove(p, buf, t);
 199         byteReverse(ctx->in, 16);
 200
 201         MD5_Transform(ctx->buf, (cmsUInt32Number *) ctx->in);
 202         buf += t;
 203         len -= t;
 204     }
 205
 206     while (len >= 64) {
 207         memmove(ctx->in, buf, 64);
 208         byteReverse(ctx->in, 16);
 209         MD5_Transform(ctx->buf, (cmsUInt32Number *) ctx->in);
 210         buf += 64;
 211         len -= 64;
 212     }
 213
 214     memmove(ctx->in, buf, len);
 215 }
 216
 217 // Destroy the object and return the checksum
 218 static
 219 void MD5finish(cmsProfileID* ProfileID,  cmsHANDLE Handle)
 220 {
 221     _cmsMD5* ctx = (_cmsMD5*) Handle;
 222     cmsUInt32Number count;
 223     cmsUInt8Number *p;
 224
 225     count = (ctx->bits[0] >> 3) & 0x3F;
 226
 227     p = ctx->in + count;
 228     *p++ = 0x80;
 229
 230     count = 64 - 1 - count;
 231
 232     if (count < 8) {
 233
 234         memset(p, 0, count);
 235         byteReverse(ctx->in, 16);
 236         MD5_Transform(ctx->buf, (cmsUInt32Number *) ctx->in);
 237
 238         memset(ctx->in, 0, 56);
 239     } else {
 240         memset(p, 0, count - 8);
 241     }
 242     byteReverse(ctx->in, 14);
 243
 244     ((cmsUInt32Number *) ctx->in)[14] = ctx->bits[0];
 245     ((cmsUInt32Number *) ctx->in)[15] = ctx->bits[1];
 246
 247     MD5_Transform(ctx->buf, (cmsUInt32Number *) ctx->in);
 248
 249     byteReverse((cmsUInt8Number *) ctx->buf, 4);
 250     memmove(ProfileID ->ID8, ctx->buf, 16);
 251
 252     _cmsFree(ctx ->ContextID, ctx);
 253 }
 254
 255
 256
 257 // Assuming io points to an ICC profile, compute and store MD5 checksum
 258 // In the header, rendering intentent, attributes and ID should be set to zero
 259 // before computing MD5 checksum (per 6.1.13 in ICC spec)
 260
 261 cmsBool CMSEXPORT cmsMD5computeID(cmsHPROFILE hProfile)
 262 {
 263     cmsContext   ContextID;
 264     cmsUInt32Number BytesNeeded;
 265     cmsUInt8Number* Mem = NULL;
 266     cmsHANDLE  MD5 = NULL;
 267     _cmsICCPROFILE* Icc = (_cmsICCPROFILE*) hProfile;
 268     _cmsICCPROFILE Keep;
 269
 270         _cmsAssert(hProfile != NULL);
 271
 272     ContextID = cmsGetProfileContextID(hProfile);
 273
 274     // Save a copy of the profile header
 275     memmove(&Keep, Icc, sizeof(_cmsICCPROFILE));
 276
 277     // Set RI, attributes and ID
 278     memset(&Icc ->attributes, 0, sizeof(Icc ->attributes));
 279     Icc ->RenderingIntent = 0;
 280     memset(&Icc ->ProfileID, 0, sizeof(Icc ->ProfileID));
 281
 282     // Compute needed storage
 283     if (!cmsSaveProfileToMem(hProfile, NULL, &BytesNeeded)) goto Error;
 284
 285     // Allocate memory
 286     Mem = (cmsUInt8Number*) _cmsMalloc(ContextID, BytesNeeded);
 287     if (Mem == NULL) goto Error;
 288
 289     // Save to temporary storage
 290     if (!cmsSaveProfileToMem(hProfile, Mem, &BytesNeeded)) goto Error;
 291
 292     // Create MD5 object
 293     MD5 = MD5alloc(ContextID);
 294     if (MD5 == NULL) goto Error;
 295
 296     // Add all bytes
 297     MD5add(MD5, Mem, BytesNeeded);
 298
 299     // Temp storage is no longer needed
 300     _cmsFree(ContextID, Mem);
 301
 302     // Restore header
 303     memmove(Icc, &Keep, sizeof(_cmsICCPROFILE));
 304
 305     // And store the ID
 306     MD5finish(&Icc ->ProfileID,  MD5);
 307     return TRUE;
 308
 309 Error:
 310
 311     // Free resources as something went wrong
 312     if (MD5 != NULL) _cmsFree(ContextID, MD5);
 313     if (Mem != NULL) _cmsFree(ContextID, Mem);
 314     memmove(Icc, &Keep, sizeof(_cmsICCPROFILE));
 315     return FALSE;
 316 }
 317