summaryrefslogtreecommitdiff
path: root/apt-pkg/contrib/sha2.cc
blob: dcdbef6e757a7877c7b00652ec10b86b77be8a03 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/*
 * Cryptographic API.							{{{
 *
 * SHA-512, as specified in
 * http://csrc.nist.gov/cryptval/shs/sha256-384-512.pdf
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option) 
 * any later version.
 *
 */									/*}}}*/

#ifdef __GNUG__
#pragma implementation "apt-pkg/sha2.h"
#endif

#include <apt-pkg/sha2.h>
#include <apt-pkg/strutl.h>




SHA512Summation::SHA512Summation()					/*{{{*/
{
   SHA512_Init(&ctx);
   Done = false;
}
         								/*}}}*/

SHA512SumValue SHA512Summation::Result()				/*{{{*/
{
   if (!Done) {
      SHA512_Final(Sum, &ctx);
      Done = true;
   }

   SHA512SumValue res;
   res.Set(Sum);
   return res;
}
									/*}}}*/
bool SHA512Summation::Add(const unsigned char *inbuf,unsigned long len) /*{{{*/
{
   if (Done) 
      return false;
   SHA512_Update(&ctx, inbuf, len);
   return true;
}
									/*}}}*/
// SHA512Summation::AddFD - Add content of file into the checksum      /*{{{*/
// ---------------------------------------------------------------------
/* */
bool SHA512Summation::AddFD(int Fd,unsigned long Size)
{
   unsigned char Buf[64 * 64];
   int Res = 0;
   int ToEOF = (Size == 0);
   while (Size != 0 || ToEOF)
   {
      unsigned n = sizeof(Buf);
      if (!ToEOF) n = min(Size,(unsigned long)n);
      Res = read(Fd,Buf,n);
      if (Res < 0 || (!ToEOF && (unsigned) Res != n)) // error, or short read
         return false;
      if (ToEOF && Res == 0) // EOF
         break;
      Size -= Res;
      Add(Buf,Res);
   }
   return true;
}
                                                                       /*}}}*/

SHA256Summation::SHA256Summation()					/*{{{*/
{
   SHA256_Init(&ctx);
   Done = false;
}
									/*}}}*/
bool SHA256Summation::Add(const unsigned char *inbuf,unsigned long len) /*{{{*/
{
   if (Done) 
      return false;
   SHA256_Update(&ctx, inbuf, len);
   return true;
}
									/*}}}*/
SHA256SumValue SHA256Summation::Result()				/*{{{*/
{
   if (!Done) {
      SHA256_Final(Sum, &ctx);
      Done = true;
   }

   SHA256SumValue res;
   res.Set(Sum);
   return res;
}
									/*}}}*/
// SHA256Summation::AddFD - Add content of file into the checksum      /*{{{*/
// ---------------------------------------------------------------------
/* */
bool SHA256Summation::AddFD(int Fd,unsigned long Size)
{
   unsigned char Buf[64 * 64];
   int Res = 0;
   int ToEOF = (Size == 0);
   while (Size != 0 || ToEOF)
   {
      unsigned n = sizeof(Buf);
      if (!ToEOF) n = min(Size,(unsigned long)n);
      Res = read(Fd,Buf,n);
      if (Res < 0 || (!ToEOF && (unsigned) Res != n)) // error, or short read
         return false;
      if (ToEOF && Res == 0) // EOF
         break;
      Size -= Res;
      Add(Buf,Res);
   }
   return true;
}
                                                                       /*}}}*/