ReaderWriter.cs
上传用户:szltgg
上传日期:2019-05-16
资源大小:604k
文件大小:5k
源码类别:

Telnet服务器

开发平台:

C#

  1. /*
  2.  Copyright (c) 2005 Poderosa Project, All Rights Reserved.
  3.  This file is a part of the Granados SSH Client Library that is subject to
  4.  the license included in the distributed package.
  5.  You may not use this file except in compliance with the license.
  6.  $Id: ReaderWriter.cs,v 1.2 2005/04/20 08:58:56 okajima Exp $
  7. */
  8. using System;
  9. using System.Text;
  10. using System.IO;
  11. using Granados.PKI;
  12. namespace Granados.SSHC {
  13. ////////////////////////////////////////////////////////////
  14. /// read/write primitive types
  15. /// 
  16. internal abstract class SSHDataReader {
  17. protected byte[] _data;
  18. protected int _offset;
  19. public SSHDataReader(byte[] image) {
  20. _data = image;
  21. _offset = 0;
  22. }
  23. public byte[] Image {
  24. get {
  25. return _data;
  26. }
  27. }
  28. public int Offset {
  29. get {
  30. return _offset;
  31. }
  32. }
  33. public int ReadInt32() {
  34. if(_offset+3>=_data.Length) throw new IOException(Strings.GetString("UnexpectedEOF"));
  35. int ret = (((int)_data[_offset])<<24) + (((int)_data[_offset+1])<<16) + (((int)_data[_offset+2])<<8) + _data[_offset+3];
  36. _offset += 4;
  37. return ret;
  38. }
  39. public byte ReadByte() {
  40. if(_offset>=_data.Length) throw new IOException(Strings.GetString("UnexpectedEOF"));
  41. return _data[_offset++];
  42. }
  43. public bool ReadBool() {
  44. if(_offset>=_data.Length) throw new IOException(Strings.GetString("UnexpectedEOF"));
  45. return _data[_offset++]==0? false : true;
  46. }
  47. /**
  48. * multi-precise integer
  49. */
  50. public abstract BigInteger ReadMPInt();
  51. public byte[] ReadString() {
  52. int length = ReadInt32();
  53. return Read(length);
  54. }
  55. public byte[] Read(int length) {
  56. byte[] image = new byte[length];
  57. for(int i=0; i<image.Length; i++) {
  58. if(_offset==_data.Length) throw new IOException(Strings.GetString("UnexpectedEOF"));
  59. image[i] = _data[_offset++];
  60. }
  61. return image;
  62. }
  63. public byte[] ReadAll() {
  64. byte[] t = new byte[_data.Length - _offset];
  65. Array.Copy(_data, _offset, t, 0, t.Length);
  66. return t;
  67. }
  68. public int Rest {
  69. get {
  70. return _data.Length - _offset;
  71. }
  72. }
  73. }
  74. internal abstract class SSHDataWriter : IKeyWriter {
  75. protected MemoryStream _strm; 
  76. public SSHDataWriter() {
  77. _strm = new MemoryStream(512);
  78. }
  79. public byte[] ToByteArray() { return _strm.ToArray(); }
  80. public long Length {
  81. get {
  82. return _strm.Length;
  83. }
  84. }
  85. public void Write(byte[] data) { _strm.Write(data, 0, data.Length); }
  86. public void Write(byte[] data, int offset, int count) { _strm.Write(data, offset, count); }
  87. public void Write(byte data)   { _strm.WriteByte(data); }
  88. public void Write(bool data)   { _strm.WriteByte(data? (byte)1 : (byte)0); }
  89. public void Write(int data) {
  90. uint udata = (uint)data;
  91. uint a = udata & 0xFF000000;
  92. a >>= 24;
  93. _strm.WriteByte((byte)a);
  94. a = udata & 0x00FF0000;
  95. a >>= 16;
  96. _strm.WriteByte((byte)a);
  97. a = udata & 0x0000FF00;
  98. a >>= 8;
  99. _strm.WriteByte((byte)a);
  100. a = udata & 0x000000FF;
  101. _strm.WriteByte((byte)a);
  102. }
  103. public abstract void Write(BigInteger data);
  104. public void Write(string data) {
  105. Write(data.Length);
  106. if(data.Length>0) Write(Encoding.ASCII.GetBytes(data));
  107. }
  108. public void WriteAsString(byte[] data) {
  109. Write(data.Length);
  110. if(data.Length>0) Write(data);
  111. }
  112. public void WriteAsString(byte[] data, int offset, int length) {
  113. Write(length);
  114. if(length>0) Write(data, offset, length);
  115. }
  116. }
  117. }
  118. namespace Granados.SSHCV1 {
  119. internal class SSH1DataReader : Granados.SSHC.SSHDataReader {
  120. public SSH1DataReader(byte[] image) : base(image) {}
  121. public override BigInteger ReadMPInt() {
  122. //first 2 bytes describes the bit count
  123. int bits = (((int)_data[_offset])<<8) + _data[_offset+1];
  124. _offset += 2;
  125. return new BigInteger(Read((bits+7) / 8));
  126. }
  127. }
  128. internal class SSH1DataWriter : Granados.SSHC.SSHDataWriter {
  129. public override void Write(BigInteger data) {
  130. byte[] image = data.getBytes();
  131. int off = (image[0]==0? 1 : 0);
  132. int len = (image.Length-off) * 8; 
  133. int a = len & 0x0000FF00;
  134. a >>= 8;
  135. _strm.WriteByte((byte)a);
  136. a = len & 0x000000FF;
  137. _strm.WriteByte((byte)a);
  138. _strm.Write(image,off,image.Length-off);
  139. }
  140. }
  141. }
  142. namespace Granados.SSHCV2 {
  143. internal class SSH2DataReader : Granados.SSHC.SSHDataReader {
  144. public SSH2DataReader(byte[] image) : base(image) {}
  145. //SSH2 Key File Only
  146. public BigInteger ReadBigIntWithBits() {
  147. int bits = ReadInt32();
  148. int bytes = (bits + 7) / 8;
  149. return new BigInteger(Read(bytes));
  150. }
  151. public override BigInteger ReadMPInt() {
  152. return new BigInteger(ReadString());
  153. }
  154. public PacketType ReadPacketType() {
  155. return (PacketType)ReadByte();
  156. }
  157. }
  158. internal class SSH2DataWriter : Granados.SSHC.SSHDataWriter {
  159. //writes mpint in SSH2 format
  160. public override void Write(BigInteger data) {
  161. byte[] t = data.getBytes();
  162. int len = t.Length;
  163. if(t[0] >= 0x80) {
  164. Write(++len);
  165. Write((byte)0);
  166. }
  167. else
  168. Write(len);
  169. Write(t);
  170. }
  171. public void WriteBigIntWithBits(BigInteger bi) {
  172. Write(bi.bitCount());
  173. Write(bi.getBytes());
  174. }
  175. public void WritePacketType(PacketType pt) {
  176. Write((byte)pt);
  177. }
  178. }
  179. }