code.grnet.gr Git - pithos-ms-client/blob - trunk/Pithos.Core/Agents/BlockExtensions.cs

   1 // -----------------------------------------------------------------------
   2 // <copyright file="FileInfoExtensions.cs" company="GRNET">
   3 // Copyright 2011-2012 GRNET S.A. All rights reserved.
   4 //
   5 // Redistribution and use in source and binary forms, with or
   6 // without modification, are permitted provided that the following
   7 // conditions are met:
   8 //
   9 //   1. Redistributions of source code must retain the above
  10 //      copyright notice, this list of conditions and the following
  11 //      disclaimer.
  12 //
  13 //   2. Redistributions in binary form must reproduce the above
  14 //      copyright notice, this list of conditions and the following
  15 //      disclaimer in the documentation and/or other materials
  16 //      provided with the distribution.
  17 //
  18 // THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  19 // OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20 // WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  21 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  22 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  25 // USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  26 // AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27 // LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  28 // ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29 // POSSIBILITY OF SUCH DAMAGE.
  30 //
  31 // The views and conclusions contained in the software and
  32 // documentation are those of the authors and should not be
  33 // interpreted as representing official policies, either expressed
  34 // or implied, of GRNET S.A.
  35 // </copyright>
  36 // -----------------------------------------------------------------------
  37
  38
  39 using System;
  40 using System.Collections.Generic;
  41 using System.Diagnostics.Contracts;
  42 using System.Linq;
  43 using System.Text;
  44 using System.IO;
  45 using System.Text.RegularExpressions;
  46 using System.Threading.Tasks;
  47 using Pithos.Network;
  48
  49 namespace Pithos.Core.Agents
  50 {
  51     static class BlockExtensions
  52     {
  53
  54
  55         public static int Read(this FileInfo fileInfo,byte[] buffer,int offset,int count)
  56         {
  57             //Open the stream only long enough to read a block
  58             using (var stream = fileInfo.OpenRead())
  59             {
  60                 stream.Seek(offset, SeekOrigin.Begin);
  61                 return  stream.Read(buffer, 0, count);
  62             }
  63         }
  64
  65        public static string CalculateHash(this FileSystemInfo info,int blockSize,string algorithm)
  66         {
  67             if (info==null)
  68                 throw new ArgumentNullException("info");
  69             if (String.IsNullOrWhiteSpace(info.FullName))
  70                 throw new ArgumentException("info");
  71             if (blockSize<=0)
  72                 throw new ArgumentOutOfRangeException("blockSize",blockSize,"blockSize must be greater than 0");
  73             if (String.IsNullOrWhiteSpace(algorithm))
  74                 throw new ArgumentNullException("algorithm");
  75             Contract.EndContractBlock();
  76
  77            //The hash for directories is an empty string
  78            if (info is DirectoryInfo)
  79                 return String.Empty;
  80            //The hash for non-existent files is an empty string
  81            if (!info.Exists)
  82                return String.Empty;
  83
  84            var fileInfo = (FileInfo)info;
  85            if (fileInfo.Length <= blockSize)
  86                 return Signature.CalculateMD5(info.FullName);
  87             else
  88                 return Signature.CalculateTreeHash(info.FullName, blockSize, algorithm).TopHash.ToHashString();
  89
  90         }
  91     }
  92 }