Statistics
| Branch: | Revision:

root / trunk / Pithos.Core / Agents / BlockExtensions.cs @ c945b450

History | View | Annotate | Download (5.2 kB)

1
#region
2
/* -----------------------------------------------------------------------
3
 * <copyright file="BlockExtensions.cs" company="GRNet">
4
 * 
5
 * Copyright 2011-2012 GRNET S.A. All rights reserved.
6
 *
7
 * Redistribution and use in source and binary forms, with or
8
 * without modification, are permitted provided that the following
9
 * conditions are met:
10
 *
11
 *   1. Redistributions of source code must retain the above
12
 *      copyright notice, this list of conditions and the following
13
 *      disclaimer.
14
 *
15
 *   2. Redistributions in binary form must reproduce the above
16
 *      copyright notice, this list of conditions and the following
17
 *      disclaimer in the documentation and/or other materials
18
 *      provided with the distribution.
19
 *
20
 *
21
 * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
22
 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
23
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
25
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
28
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32
 * POSSIBILITY OF SUCH DAMAGE.
33
 *
34
 * The views and conclusions contained in the software and
35
 * documentation are those of the authors and should not be
36
 * interpreted as representing official policies, either expressed
37
 * or implied, of GRNET S.A.
38
 * </copyright>
39
 * -----------------------------------------------------------------------
40
 */
41
#endregion
42
using System;
43
using System.Collections.Generic;
44
using System.Diagnostics.Contracts;
45
using System.Linq;
46
using System.Reflection;
47
using System.Security.Cryptography;
48
using System.Text;
49
using System.IO;
50
using System.Text.RegularExpressions;
51
using System.Threading.Tasks;
52
using Pithos.Network;
53
using log4net;
54

    
55
namespace Pithos.Core.Agents
56
{
57
    static class BlockExtensions
58
    {
59

    
60
        private static readonly ILog Log = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
61

    
62
        public static int Read(this FileInfo fileInfo,byte[] buffer,long offset,int count)
63
        {
64
            if (offset < 0)
65
                throw new ArgumentOutOfRangeException("offset", offset, "The file offset can't be negative");
66
            Contract.EndContractBlock();
67
            //Open the stream only long enough to read a block
68
            using (var stream = fileInfo.OpenRead())
69
            {
70
                stream.Seek(offset, SeekOrigin.Begin);
71
                return  stream.Read(buffer, 0, count);                
72
            }
73
        }
74

    
75
       public static string CalculateHash(this FileSystemInfo info,int blockSize,string algorithm)
76
        {
77
            if (info==null)
78
                throw new ArgumentNullException("info");
79
            if (String.IsNullOrWhiteSpace(info.FullName))
80
                throw new ArgumentException("info");
81
            if (blockSize<=0)
82
                throw new ArgumentOutOfRangeException("blockSize",blockSize,"blockSize must be greater than 0");
83
            if (String.IsNullOrWhiteSpace(algorithm))
84
                throw new ArgumentNullException("algorithm");
85
            Contract.EndContractBlock();
86
           info.Refresh();
87
           //The hash for directories is an empty string
88
           if (info is DirectoryInfo)
89
                return String.Empty;
90
           //The hash for non-existent files is an empty string
91
           if (!info.Exists)
92
               return String.Empty;
93

    
94
           return Signature.CalculateTreeHash(info.FullName, blockSize, algorithm).TopHash.ToHashString();
95

    
96
        }
97

    
98
       /// <summary>
99
       ///Calculates a simple hash for an entire file
100
       /// </summary>
101
       /// <param name="info">The file to hash</param>
102
       /// <param name="hasher">The hash algorithm to use</param>
103
       /// <returns>A hash value for the entire file. An empty string if the file does not exist.</returns>
104
       public static string ComputeShortHash(this FileInfo info, HashAlgorithm hasher)
105
       {
106
           if(info == null)
107
               throw new ArgumentNullException("info");
108
           if(hasher== null)
109
               throw new ArgumentNullException("hasher");
110
           Contract.EndContractBlock();
111
           info.Refresh();
112
           if (!info.Exists)
113
               return String.Empty;
114

    
115
           if (Log.IsDebugEnabled)
116
                Log.DebugFormat("Short Hashing [{0}] ",info.FullName);
117

    
118
           using (var stream = info.Open(FileMode.Open, FileAccess.Read, FileShare.Read))
119
           {
120
               var hash = hasher.ComputeHash(stream);
121
                var hashString = hash.ToHashString();
122
               return hashString;
123
           }
124
       }
125

    
126
        public static string ComputeShortHash(this FileInfo info)
127
       {
128
           if(info == null)
129
               throw new ArgumentNullException("info");
130
            Contract.EndContractBlock();
131

    
132
           using (var hasher=HashAlgorithm.Create("sha1"))
133
           {               
134
               return ComputeShortHash(info,hasher);
135
           }
136
       }
137

    
138
    }
139
}