FilesExpand file tree

 develop

/

HashImpl.cs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

160 lines (143 loc) · 5.24 KB

 develop

/

HashImpl.cs

Copy path

Top

File metadata and controls

• Code
• Blame

160 lines (143 loc) · 5.24 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

/*----------------------------------------------------------

This Source Code Form is subject to the terms of the

Mozilla Public License, v.2.0. If a copy of the MPL

was not distributed with this file, You can obtain one

at http://mozilla.org/MPL/2.0/.

----------------------------------------------------------*/

using System;

using System.IO;

using System.Security.Cryptography;

using System.Text;

using OneScript.Contexts;

using OneScript.Exceptions;

using OneScript.StandardLibrary.Binary;

using OneScript.Values;

using ScriptEngine.Machine;

using ScriptEngine.Machine.Contexts;

namespace OneScript.StandardLibrary.Hash

{

[ContextClass("ХешированиеДанных", "DataHashing")]

public class HashImpl : AutoContext<HashImpl>, IDisposable

{

private HashAlgorithm _provider;

private HashFunctionEnum _enumValue;

private CombinedStream _toCalculate=new CombinedStream();

private bool _calculated;

private byte[] _hash;

public HashImpl(HashAlgorithm provider, HashFunctionEnum enumValue)

{

_provider = provider;

_enumValue = enumValue;

_calculated = false;

}

public byte[] InternalHash

{

get

{

if (!_calculated)

{

_hash = _provider.ComputeHash(_toCalculate);

_calculated = true;

}

return _hash;

}

}

[ContextProperty("ХешФункция", "HashFunction")]

public HashFunctionEnum Extension => _enumValue;

[ContextProperty("ХешСумма", "HashSum")]

public IValue Hash

{

get

{

if (_provider is Crc32)

{

var buffer = new byte[4];

Array.Copy(InternalHash, buffer, 4);

if (BitConverter.IsLittleEndian)

Array.Reverse(buffer);

var ret = BitConverter.ToUInt32(buffer, 0);

return ValueFactory.Create((decimal)ret);

}

return new BinaryDataContext(InternalHash);

}

}

[ContextProperty("ХешСуммаСтрокой", "HashSumOfString")]

public string HashString

{

get

{

var sb = new StringBuilder();

for (int i = 0; i < InternalHash.Length; i++)

sb.Append(InternalHash[i].ToString("X2"));

return sb.ToString();

}

}

[ContextMethod("Добавить", "Append")]

public void Append(BslValue toAdd, uint count = 0)

{

switch (toAdd)

{

case BslStringValue s:

AddStream(new MemoryStream(Encoding.UTF8.GetBytes((string)s)));

break;

case BslObjectValue obj when obj is IStreamWrapper wrapper:

var stream = wrapper.GetUnderlyingStream();

var readByte = (int)Math.Min(count == 0 ? stream.Length : count, stream.Length - stream.Position);

var buffer = new byte[readByte];

stream.Read(buffer, 0, readByte);

AddStream(new MemoryStream(buffer));

break;

case BslObjectValue obj when obj is BinaryDataContext binaryData:

AddStream(binaryData.GetStream());

break;

default:

throw RuntimeException.InvalidArgumentType(nameof(toAdd));

}

}

[ContextMethod("ДобавитьФайл", "AppendFile")]

public void AppendFile(string path)

{

if (!File.Exists(path))

throw RuntimeException.InvalidArgumentType();

AddStream(new FileStream(path, FileMode.Open, FileAccess.Read, FileShare.ReadWrite));

}

[ContextMethod("Очистить", "Clear")]

public void Clear()

{

_toCalculate.Close();

_toCalculate.Dispose();

_toCalculate = new CombinedStream();

_calculated = false;

}

[ScriptConstructor(Name = "По указанной хеш-функции")]

public static HashImpl Constructor(HashFunctionEnum providerEnum)

{

var objectProvider = GetProvider(providerEnum);

return new HashImpl(objectProvider, providerEnum);

}

private static HashAlgorithm GetProvider(HashFunctionEnum algo)

{

switch (algo)

{

case HashFunctionEnum.CRC32:

return new Crc32();

default:

var ret = HashAlgorithm.Create(algo.ToString());

if (ret == null)

throw RuntimeException.InvalidArgumentType();

return ret;

}

}

public void Dispose()

{

_toCalculate.Close();

_toCalculate.Dispose();

}

private void AddStream(Stream stream)

{

_toCalculate.AddStream(stream);

_toCalculate.Seek(0, SeekOrigin.Begin);

_calculated = false;

}

}

}