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

Telnet服务器

开发平台:

C#

  1. /*
  2. * Copyright (c) 2005 Poderosa Project, All Rights Reserved.
  3. * $Id: TerminalConnection.cs,v 1.2 2005/04/20 08:45:47 okajima Exp $
  4. */
  5. using System;
  6. using System.Diagnostics;
  7. using System.Net.Sockets;
  8. using System.Net;
  9. using System.Text;
  10. using System.Threading;
  11. using System.Windows.Forms;
  12. using System.IO;
  13. using Poderosa.ConnectionParam;
  14. using Poderosa.Connection;
  15. using Poderosa.Log;
  16. using Poderosa.SSH;
  17. using Poderosa.Toolkit;
  18. using Poderosa.Terminal;
  19. using Poderosa.Text;
  20. using Granados.SSHC;
  21. namespace Poderosa.Communication
  22. {
  23. public interface IDataReceiver {
  24. void DataArrived(byte[] buf, int offset, int count);
  25. void DisconnectedFromServer();
  26. void ErrorOccurred(string msg);
  27. }
  28. internal abstract class AbstractGuevaraSocket {
  29. protected IDataReceiver _callback;
  30. internal abstract void Transmit(byte[] data, int offset, int length);
  31. internal abstract void Flush();
  32. internal abstract void Close();
  33. internal abstract bool DataAvailable { get; }
  34. internal abstract void RepeatAsyncRead(IDataReceiver receiver);
  35. }
  36. internal class PlainGuevaraSocket : AbstractGuevaraSocket {
  37. private Socket _socket;
  38. private byte[] _buf;
  39. internal PlainGuevaraSocket(Socket s) {
  40. _socket = s;
  41. _buf = new byte[0x1000];
  42. }
  43. internal override bool DataAvailable {
  44. get {
  45. return _socket.Available>0;
  46. }
  47. }
  48. internal override void Transmit(byte[] data, int offset, int length) {
  49. _socket.Send(data, offset, length, SocketFlags.None);
  50. }
  51. internal override void Flush() {
  52. }
  53. internal override void Close() {
  54. _socket.Close();
  55. }
  56. internal override void RepeatAsyncRead(IDataReceiver receiver) {
  57. _callback = receiver;
  58. _socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  59. }
  60. #if true
  61. private void RepeatCallback(IAsyncResult result) {
  62. try {
  63. int n = _socket.EndReceive(result);
  64. //GUtil.WriteDebugLog(String.Format("t={0}, n={1} av={2}", DateTime.Now.ToString(), n, _socket.Available));
  65. //Debug.WriteLine(String.Format("r={0}, n={1} ", result.IsCompleted, n));
  66. if(n > 0) {
  67. _callback.DataArrived(_buf, 0, n);
  68. _socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  69. }
  70. else if(n < 0) {
  71. //WindowsME偵偍偄偰偼丄偲偒偳偒偙偙偱-1偑曉偭偰偒偰偄傞偙偲偑敪妎偟偨丅
  72. _socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  73. }
  74. else
  75. _callback.DisconnectedFromServer();
  76. }
  77. catch(Exception ex) {
  78. if((ex is SocketException) && ((SocketException)ex).ErrorCode==995) {
  79. //GUtil.WriteDebugLog(String.Format("t={0} error995", DateTime.Now.ToString()));
  80. _socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  81. }
  82. else
  83. _callback.ErrorOccurred(ex.Message);
  84. }
  85. }
  86. #else //庴怣捈屻偐傜儊僀儞僗儗僢僪偵張棟傪堏偡僶乕僕儑儞
  87. private int _readLen;
  88. private bool _nextReadRequired;
  89. private void RepeatCallback(IAsyncResult result) {
  90. _readLen = _socket.EndReceive(result);
  91. _nextReadRequired = false;
  92. GEnv.Frame.AsForm().Invoke(new MethodInvoker(ReadBody), null);
  93. if(_nextReadRequired)
  94. _socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  95. }
  96. private void ReadBody() {
  97. try {
  98. Debug.Assert(!GEnv.Frame.AsForm().InvokeRequired);
  99. if(_readLen > 0) {
  100. _callback.DataArrived(_buf, 0, _readLen);
  101. _nextReadRequired = true;
  102. }
  103. else if(_readLen < 0) {
  104. //WindowsME偵偍偄偰偼丄偲偒偳偒偙偙偱-1偑曉偭偰偒偰偄傞偙偲偑敪妎偟偨丅
  105. //_socket.BeginReceive(_buf, 0, _buf.Length, SocketFlags.None, new AsyncCallback(RepeatCallback), null);
  106. _nextReadRequired = true;
  107. }
  108. else
  109. _callback.DisconnectedFromServer();
  110. }
  111. catch(Exception ex) {
  112. if((ex is SocketException) && ((SocketException)ex).ErrorCode==995) {
  113. //GUtil.WriteDebugLog(String.Format("t={0} error995", DateTime.Now.ToString()));
  114. _nextReadRequired = true;
  115. }
  116. else
  117. _callback.ErrorOccurred(ex.Message);
  118. }
  119. }
  120. #endif
  121. }
  122. internal class ChannelGuevaraSocket : AbstractGuevaraSocket, ISSHConnectionEventReceiver, ISSHChannelEventReceiver {
  123. private SSHChannel _channel;
  124. private bool _ready;
  125. private AutoResetEvent _event;
  126. internal ChannelGuevaraSocket() {
  127. _ready = false;
  128. }
  129. internal SSHChannel SSHChennal {
  130. get {
  131. return _channel;
  132. }
  133. set {
  134. _channel = value;
  135. }
  136. }
  137. internal override void RepeatAsyncRead(IDataReceiver cb) {
  138. Debug.Assert(_callback==null); //侾夞偟偐屇傫偱偼偩傔
  139. _callback = cb;
  140. if(_event!=null) {
  141. _event.Set();
  142. _event.Close();
  143. _event = null;
  144. }
  145. }
  146. internal override bool DataAvailable {
  147. get {
  148. return _channel.Connection.Available;
  149. }
  150. }
  151. internal override void Transmit(byte[] data, int offset, int length) {
  152. if(!_ready || _channel==null) throw new IOException("channel not ready");
  153. _channel.Transmit(data, offset, length);
  154. }
  155. internal override void Flush() {
  156. }
  157. internal override void Close() {
  158. _channel.Close();
  159. if(_channel.Connection.ChannelCount<=1)
  160. _channel.Connection.Close();
  161. }
  162. public void OnData(byte[] data, int offset, int length) {
  163. EnsureHandler();
  164. _callback.DataArrived(data, offset, length);
  165. }
  166. public void OnChannelEOF() {
  167. EnsureHandler();
  168. _callback.DisconnectedFromServer();
  169. }
  170. public void OnChannelError(Exception ex, string msg) {
  171. EnsureHandler();
  172. if(!_ready)
  173. msg = GEnv.Strings.GetString("Message.ChannelPoderosaSocket.FailedToPortforward") + msg;
  174. _callback.ErrorOccurred(msg);
  175. }
  176. public void OnChannelClosed() {
  177. EnsureHandler();
  178. _callback.DisconnectedFromServer();
  179. }
  180. public void OnChannelReady() {
  181. _ready = true;
  182. }
  183. public void OnAuthenticationPrompt(string[] msg) {
  184. for(int i=0; i<msg.Length; i++) {
  185. if(i==0) msg[i] += "rn";
  186. byte[] buf = Encoding.ASCII.GetBytes(msg[i]);
  187. OnData(buf, 0, buf.Length);
  188. }
  189. }
  190. public void OnExtendedData(int type, byte[] data) {
  191. }
  192. public void OnConnectionClosed() {
  193. EnsureHandler();
  194. _callback.DisconnectedFromServer();
  195. }
  196. public void OnDebugMessage(bool display, byte[] data) {
  197. Debug.WriteLine(String.Format("SSH debug {0}[{1}]", data.Length, data[0] ));
  198. }
  199. public void OnIgnoreMessage(byte[] data) {
  200. Debug.WriteLine(String.Format("SSH ignore {0}[{1}]", data.Length, data[0] ));
  201. }
  202. public void OnUnknownMessage(byte type, byte[] data) {
  203. Debug.WriteLine(String.Format("Unexpected SSH packet type {0}", type));
  204. }
  205. public void OnError(Exception ex, string msg) {
  206. EnsureHandler();
  207. _callback.ErrorOccurred(msg);
  208. }
  209. public void OnMiscPacket(byte type, byte[] data, int offset, int length) {
  210. }
  211. public Granados.SSHC.PortForwardingCheckResult CheckPortForwardingRequest(string host, int port, string originator, int originator_port) {
  212. return new Granados.SSHC.PortForwardingCheckResult();
  213. }
  214. public void EstablishPortforwarding(ISSHChannelEventReceiver receiver, SSHChannel channel) {
  215. }
  216. private void EnsureHandler() {
  217. if(_callback!=null) return;
  218. _event = new AutoResetEvent(false);
  219. _event.WaitOne();
  220. }
  221. }
  222. public abstract class TerminalConnection {
  223. internal TerminalParam _param;
  224. internal ITerminalTextLogger _loggerT;
  225. internal ITerminalBinaryLogger _loggerB;
  226. internal int _width;
  227. internal int _height;
  228. internal LogType _logType;
  229. internal string _logPath;
  230. //儘僌偺堦帪揑側僆僼偺偨傔偺僗僀僢僠
  231. internal bool _logsuspended;
  232. //憲庴怣偟偨僨乕僞僒僀僘偺忣曬
  233. internal int _sentPacketCount;
  234. internal int _sentDataSize;
  235. internal int _receivedPacketCount;
  236. internal int _receivedDataSize;
  237. internal IPAddress _serverAddress;
  238. internal string _serverName;
  239. //偡偱偵僋儘乕僘偝傟偨偐偳偆偐偺僼儔僌
  240. internal bool _closed;
  241. protected TerminalConnection(TerminalParam p, int width, int height) {
  242. _param = p;
  243. _width = width;
  244. _height = height;
  245. ResetLog(p.LogType, p.LogPath, p.LogAppend);
  246. _logsuspended = false;
  247. }
  248. public IPAddress ServerAddress {
  249. get {
  250. return _serverAddress;
  251. }
  252. }
  253. public string ServerName {
  254. get {
  255. return _serverName;
  256. }
  257. }
  258. public abstract string ProtocolDescription {
  259. get;
  260. }
  261. public int SentPacketCount {
  262. get {
  263. return _sentPacketCount;
  264. }
  265. }
  266. public int SentDataSize {
  267. get {
  268. return _sentDataSize;
  269. }
  270. }
  271. public int ReceivedPacketCount {
  272. get {
  273. return _receivedPacketCount;
  274. }
  275. }
  276. public int ReceivedDataSize {
  277. get {
  278. return _receivedDataSize;
  279. }
  280. }
  281. public TerminalParam Param {
  282. get {
  283. return _param;
  284. }
  285. }
  286. public int TerminalHeight {
  287. get {
  288. return _height;
  289. }
  290. }
  291. public int TerminalWidth {
  292. get {
  293. return _width;
  294. }
  295. }
  296. public bool LogSuspended {
  297. get {
  298. return _logsuspended;
  299. }
  300. set {
  301. _logsuspended = value;
  302. }
  303. }
  304. public LogType LogType {
  305. get {
  306. return _logType;
  307. }
  308. }
  309. public string LogPath {
  310. get {
  311. return _logPath;
  312. }
  313. }
  314. public void SetServerInfo(string host, IPAddress address) {
  315. _serverName = host;
  316. _serverAddress = address;
  317. }
  318. public void CommentLog(string comment) {
  319. _loggerT.Comment(comment);
  320. _loggerB.Comment(comment);
  321. }
  322. public ITerminalTextLogger TextLogger {
  323. get {
  324. return _loggerT;
  325. }
  326. }
  327. public ITerminalBinaryLogger BinaryLogger {
  328. get {
  329. return _loggerB;
  330. }
  331. }
  332. public bool IsClosed {
  333. get {
  334. return _closed;
  335. }
  336. }
  337. public abstract string[] ConnectionParameter { get; }
  338. public abstract bool Available { get; }
  339. public virtual void Resize(int width, int height) {
  340. Debug.Assert(width>0 && height>0);
  341. _width = width;
  342. _height = height;
  343. }
  344. public abstract ConnectionTag Reproduce(); //摨偠偲偙傠傊偺愙懕傪傕偆侾杮
  345. //廔椆張棟
  346. internal virtual void Close() {
  347. if(_loggerT!=null) {
  348. _loggerT.Flush();
  349. _loggerT.Close();
  350. _loggerT = null;
  351. }
  352. if(_loggerB!=null) {
  353. _loggerB.Flush();
  354. _loggerB.Close();
  355. _loggerB = null;
  356. }
  357. _closed = true;
  358. }
  359. //愙懕傪暵偠傞慜偺嵟屻偺傾僋僔儑儞丅偙偪傜偐傜愗傞偲偒偵偼偙傟傪傛傇傋偒偩偑丄僒乕僶偐傜愗偭偰偒偨偲偒偼屇傇昁梫側偟
  360. internal virtual void Disconnect() {
  361. }
  362. internal abstract void RepeatAsyncRead(IDataReceiver cb);
  363. public void WriteChars(char[] data) {
  364. byte[] b = _param.EncodingProfile.GetBytes(data);
  365. Write(b);
  366. }
  367. public void WriteChar(char data) {
  368. byte[] b = _param.EncodingProfile.GetBytes(data);
  369. Write(b);
  370. }
  371. public abstract void Write(byte[] data);
  372. public abstract void Write(byte[] data, int offset, int length);
  373. //僨僼僅儖僩幚憰偼嬻
  374. public virtual void AreYouThere() {
  375. throw new NotSupportedException("[AYT] unsupported");
  376. }
  377. public virtual void SendBreak() {
  378. throw new NotSupportedException("[Break] unsupported");
  379. }
  380. public virtual void SendKeepAliveData() {
  381. }
  382. internal void AddSentDataStats(int bytecount) {
  383. _sentPacketCount++;
  384. _sentDataSize += bytecount;
  385. }
  386. internal void AddReceivedDataStats(int bytecount) {
  387. _receivedPacketCount++;
  388. _receivedDataSize += bytecount;
  389. }
  390. public void ResetLog(LogType t, string path, bool append) {
  391. _logType = t;
  392. _logPath = path;
  393. if(_loggerT!=null) _loggerT.Close();
  394. if(_loggerB!=null) _loggerB.Close();
  395. switch(t) {
  396. case LogType.None:
  397. _loggerT = new NullTextLogger();
  398. _loggerB = new NullBinaryLogger();
  399. break;
  400. case LogType.Default:
  401. _loggerT = new DefaultLogger(new StreamWriter(path, append, Encoding.Default));
  402. _loggerB = new NullBinaryLogger();
  403. break;
  404. case LogType.Binary:
  405. _loggerT = new NullTextLogger();
  406. _loggerB = new BinaryLogger(new FileStream(path, append? FileMode.Append : FileMode.Create));
  407. break;
  408. case LogType.Xml:
  409. _loggerT = new XmlLogger(new StreamWriter(path, append, Encoding.UTF8), _param); //DebugLog偼UTF8
  410. _loggerB = new NullBinaryLogger();
  411. break;
  412. }
  413. _loggerT = new InternalLoggerT(_loggerT, this);
  414. _loggerB = new InternalLoggerB(_loggerB, this);
  415. _loggerT.TerminalResized(_width, _height);
  416. }
  417. private class InternalLoggerT : ITerminalTextLogger {
  418. private TerminalConnection _parent;
  419. private ITerminalTextLogger _logger;
  420. public InternalLoggerT(ITerminalTextLogger l, TerminalConnection p) {
  421. _parent = p;
  422. _logger = l;
  423. }
  424. public void Append(char ch) {
  425. if(!_parent.LogSuspended) _logger.Append(ch);
  426. }
  427. public void Append(char[] ch) {
  428. if(!_parent.LogSuspended) _logger.Append(ch);
  429. }
  430. public void Append(char[] ch, int offset, int length) {
  431. if(!_parent.LogSuspended) _logger.Append(ch,offset,length);
  432. }
  433. public void BeginEscapeSequence() {
  434. if(!_parent.LogSuspended) _logger.BeginEscapeSequence();
  435. }
  436. public void AbortEscapeSequence() {
  437. if(!_parent.LogSuspended) _logger.AbortEscapeSequence();
  438. }
  439. public void CommitEscapeSequence() {
  440. if(!_parent.LogSuspended) _logger.CommitEscapeSequence();
  441. }
  442. public void Comment(string comment) {
  443. if(!_parent.LogSuspended) _logger.Comment(comment);
  444. }
  445. public void Flush() {
  446. _logger.Flush();
  447. }
  448. public void Close() {
  449. _logger.Close();
  450. }
  451. public bool IsActive {
  452. get {
  453. return _logger.IsActive;
  454. }
  455. }
  456. public void PacketDelimiter() {
  457. if(!_parent.LogSuspended) _logger.PacketDelimiter();
  458. }
  459. public void TerminalResized(int width, int height) {
  460. if(!_parent.LogSuspended) _logger.TerminalResized(width, height);
  461. }
  462. public void WriteLine(GLine line) {
  463. if(!_parent.LogSuspended) _logger.WriteLine(line);
  464. }
  465. }
  466. private class InternalLoggerB : ITerminalBinaryLogger {
  467. private TerminalConnection _parent;
  468. private ITerminalBinaryLogger _logger;
  469. public InternalLoggerB(ITerminalBinaryLogger l, TerminalConnection p) {
  470. _parent = p;
  471. _logger = l;
  472. }
  473. public void Append(byte[] data, int offset, int length) {
  474. if(!_parent.LogSuspended) _logger.Append(data,offset,length);
  475. }
  476. public void Comment(string comment) {
  477. if(!_parent.LogSuspended) _logger.Comment(comment);
  478. }
  479. public void Flush() {
  480. _logger.Flush();
  481. }
  482. public void Close() {
  483. _logger.Close();
  484. }
  485. public bool IsActive {
  486. get {
  487. return _logger.IsActive;
  488. }
  489. }
  490. }
  491. }
  492. internal abstract class TCPTerminalConnection : TerminalConnection {
  493. protected bool _usingSocks;
  494. protected TCPTerminalConnection(TerminalParam p, int w, int h) : base(p, w, h) {
  495. _usingSocks = false;
  496. }
  497. public TCPTerminalParam TCPParam {
  498. get {
  499. return _param as TCPTerminalParam;
  500. }
  501. }
  502. public override string ProtocolDescription {
  503. get {
  504. string s = _param.MethodName;
  505. if(_usingSocks) s += GEnv.Strings.GetString("Caption.TCPTerminalConnection.UsingSOCKS");
  506. return s;
  507. }
  508. }
  509. //愝掕偼嵟弶偩偗峴偆
  510. public bool UsingSocks {
  511. get {
  512. return _usingSocks;
  513. }
  514. set {
  515. _usingSocks = value;
  516. }
  517. }
  518. }
  519. internal class SSHTerminalConnection : TCPTerminalConnection, ISSHConnectionEventReceiver, ISSHChannelEventReceiver {
  520. private SSHConnection _connection;
  521. private SSHChannel _channel;
  522. private MemoryStream _passwordBuffer;
  523. private bool _waitingSendBreakReply;
  524. public SSHTerminalConnection(TCPTerminalParam p, int width, int height) : base(p, width, height) {
  525. }
  526. public override ConnectionTag Reproduce() {
  527. SSHTerminalParam sshp = (SSHTerminalParam)_param.Clone();
  528. if(GEnv.Options.DefaultLogType!=LogType.None) {
  529. sshp.LogType = GEnv.Options.DefaultLogType;
  530. sshp.LogPath = GUtil.CreateLogFileName(sshp.ShortDescription);
  531. }
  532. else
  533. sshp.LogType = LogType.None;
  534. if(sshp.Method==ConnectionMethod.SSH2 && !this.IsClosed) { //SSH2偺偲偒偼僐僱僋僔儑儞傪嫟桳偟偰儅儖僠僠儍僱儖傪巊梡
  535. System.Drawing.Size sz = GEnv.Frame.TerminalSizeForNextConnection;
  536. SSHTerminalConnection newcon = new SSHTerminalConnection(sshp, sz.Width, sz.Height);
  537. newcon.FixConnection(_connection);
  538. newcon.OpenShell();
  539. newcon.SetServerInfo(_serverName, _serverAddress);
  540. return new ConnectionTag(newcon);
  541. }
  542. else { 
  543. bool pp_found = (sshp.Passphrase!=null && sshp.Passphrase.Length>0);
  544. if(sshp.Method==ConnectionMethod.SSH1 && !pp_found)
  545. throw new ApplicationException(GEnv.Strings.GetString("Message.SSHTerminalConnection.ReproduceErrorOnSSH1"));
  546. if(sshp.Method==ConnectionMethod.SSH2 && !pp_found && this.IsClosed)
  547. throw new ApplicationException(GEnv.Strings.GetString("Message.SSHTerminalConnection.ReproduceErrorOnSSH2"));
  548. HostKeyChecker checker = new HostKeyChecker(GEnv.Frame, sshp);
  549. return CommunicationUtil.CreateNewConnection(sshp, new HostKeyCheckCallback(checker.CheckHostKeyCallback));
  550. }
  551. }
  552. public void FixConnection(SSHConnection con) {
  553. _connection = con;
  554. }
  555. public SSHConnectionInfo ConnectionInfo {
  556. get {
  557. return _connection.ConnectionInfo;
  558. }
  559. }
  560. public void OpenShell() {
  561. _channel = _connection.OpenShell(this);
  562. }
  563. internal override void Disconnect() {
  564. Close();
  565. }
  566. internal override void Close() {
  567. if(_closed) return; //俀搙埲忋僋儘乕僘偟偰傕暃嶌梡側偟 
  568. base.Close();
  569. try {
  570. if(_channel!=null) _channel.Close();
  571. if(_connection.ChannelCount==0) {
  572. _connection.Close();
  573. }
  574. }
  575. catch(Exception ex) {
  576. GUtil.Warning(GEnv.Frame, GEnv.Strings.GetString("Message.SSHTerminalConnection.CloseError")+ex.Message);
  577. }
  578. }
  579. public override string[] ConnectionParameter {
  580. get {
  581. string[] d = new string[2];
  582. d[0] = String.Format("ServerVersionString: {0}", _connection.ConnectionInfo.ServerVersionString);
  583. d[1] = String.Format("EncryptionAlgorithm: {0}",_connection.ConnectionInfo.AlgorithmForReception.ToString());
  584. return d;
  585. }
  586. }
  587. public override bool Available {
  588. get {
  589. return !_closed && _connection.Available;
  590. }
  591. }
  592. public override void Write(byte[] buf) {
  593. if(_connection.AuthenticationResult==AuthenticationResult.Prompt)
  594. InputAuthenticationResponse(buf, 0, buf.Length);
  595. else {
  596. AddSentDataStats(buf.Length);
  597. _channel.Transmit(buf);
  598. }
  599. }
  600. public override void Write(byte[] buf, int offset, int length) {
  601. if(_connection.AuthenticationResult==AuthenticationResult.Prompt)
  602. InputAuthenticationResponse(buf, offset, length);
  603. else {
  604. AddSentDataStats(length);
  605. _channel.Transmit(buf, offset, length);
  606. }
  607. }
  608. //authentication process for keyboard-interactive
  609. private void InputAuthenticationResponse(byte[] buf, int offset, int length) {
  610. for(int i=offset; i<offset+length; i++) {
  611. byte b = buf[i];
  612. if(_passwordBuffer==null) _passwordBuffer = new MemoryStream();
  613. if(b==13 || b==10) { //CR/LF
  614. byte[] pwd = _passwordBuffer.ToArray();
  615. if(pwd.Length>0) {
  616. _passwordBuffer.Close();
  617. string[] response = new string[1];
  618. response[0] = Encoding.ASCII.GetString(pwd);
  619. OnData(Encoding.ASCII.GetBytes("rn"), 0, 2); //昞帵忋夵峴偟側偄偲奿岲埆偄
  620. if(((Granados.SSHCV2.SSH2Connection)_connection).DoKeyboardInteractiveAuth(response)==AuthenticationResult.Success)
  621. _channel = _connection.OpenShell(this);
  622. _passwordBuffer = null;
  623. }
  624. }
  625. else if(b==3 || b==27) { //Ctrl+C, ESC
  626. GEnv.GetConnectionCommandTarget(this).Disconnect();
  627. return;
  628. }
  629. else
  630. _passwordBuffer.WriteByte(b);
  631. }
  632. }
  633. //旕摨婜偵庴怣偡傞丅
  634. private IDataReceiver _callback;
  635. private MemoryStream _buffer;
  636. internal override void RepeatAsyncRead(IDataReceiver cb) {
  637. _callback = cb;
  638. if(_buffer!=null) {
  639. lock(this) {
  640. _buffer.Close();
  641. byte[] t = _buffer.ToArray();
  642. if(t.Length>0) _callback.DataArrived(t, 0, t.Length);
  643. _buffer = null;
  644. }
  645. }
  646. }
  647. public override void Resize(int width, int height) {
  648. if(_connection.AuthenticationResult!=AuthenticationResult.Success) return;
  649. base.Resize(width, height);
  650. if(!_closed)
  651. _channel.ResizeTerminal(width, height, 0, 0);
  652. }
  653. public override void SendBreak() {
  654. if(_connection.AuthenticationResult!=AuthenticationResult.Success) return;
  655. if(((TCPTerminalParam)_param).Method==ConnectionMethod.SSH1)
  656. base.SendBreak(); //偙傟偱嫅斲偵側傞
  657. else {
  658. _waitingSendBreakReply = true;
  659. ((Granados.SSHCV2.SSH2Channel)_channel).SendBreak(500);
  660. }
  661. }
  662. public override void SendKeepAliveData() {
  663. _connection.SendIgnorableData("keep alive");
  664. }
  665. public void OnChannelClosed() {
  666. if(!_closed)
  667. _callback.DisconnectedFromServer();
  668. }
  669. public void OnChannelEOF() {
  670. if(!_closed)
  671. _callback.DisconnectedFromServer();
  672. }
  673. public void OnData(byte[] data, int offset, int length) {
  674. if(_callback==null) { //RepeatAsyncRead偑屇偽傟傞慜偺僨乕僞傪廤傔偰偍偔
  675. lock(this) {
  676. if(_buffer==null) _buffer = new MemoryStream(0x100);
  677. _buffer.Write(data, offset, length);
  678. }
  679. }
  680. else
  681. _callback.DataArrived(data, offset, length);
  682. }
  683. public void OnAuthenticationPrompt(string[] msg) {
  684. for(int i=0; i<msg.Length; i++) {
  685. if(i!=0) msg[i] += "rn";
  686. byte[] buf = Encoding.ASCII.GetBytes(msg[i]);
  687. OnData(buf, 0, buf.Length);
  688. }
  689. }
  690. public void OnExtendedData(int type, byte[] data) {
  691. }
  692. public void OnMiscPacket(byte type, byte[] data, int offset, int length) {
  693. if(_waitingSendBreakReply) {
  694. _waitingSendBreakReply = false;
  695. if(type==(byte)Granados.SSHCV2.PacketType.SSH_MSG_CHANNEL_FAILURE)
  696. GEnv.InterThreadUIService.Warning(GEnv.Strings.GetString("Message.SSHTerminalconnection.BreakError"));
  697. }
  698. }
  699. public void OnConnectionClosed() {
  700. if(!_closed) {
  701. EnsureCallbackHandler();
  702. _callback.DisconnectedFromServer();
  703. }
  704. }
  705. public void OnDebugMessage(bool display, byte[] data) {
  706. Debug.WriteLine(String.Format("SSH debug {0}[{1}]", data.Length, data[0] ));
  707. }
  708. public void OnIgnoreMessage(byte[] data) {
  709. Debug.WriteLine(String.Format("SSH ignore {0}[{1}]", data.Length, data[0] ));
  710. }
  711. public void OnUnknownMessage(byte type, byte[] data) {
  712. Debug.WriteLine(String.Format("Unexpected SSH packet type {0}", type));
  713. }
  714. public void OnChannelReady() { //!!Transmit傪嫋壜偡傞捠抦偑昁梫丠
  715. }
  716. public void OnChannelError(Exception ex, string msg) {
  717. if(!_closed) {
  718. EnsureCallbackHandler();
  719. _callback.ErrorOccurred(msg);
  720. }
  721. }
  722. public void OnError(Exception ex, string msg) {
  723. if(!_closed) {
  724. EnsureCallbackHandler();
  725. _callback.ErrorOccurred(msg);
  726. }
  727. }
  728. public Granados.SSHC.PortForwardingCheckResult CheckPortForwardingRequest(string host, int port, string originator, int originator_port) {
  729. return new Granados.SSHC.PortForwardingCheckResult();
  730. }
  731. public void EstablishPortforwarding(ISSHChannelEventReceiver receiver, SSHChannel channel) {
  732. }
  733. private void EnsureCallbackHandler() {
  734. int n = 0;
  735. while(_callback==null && n++<10) //傢偢偐側帪娫嵎偱僴儞僪儔偑僙僢僩偝傟側偄偙偲傕偁傞
  736. Thread.Sleep(100);
  737. }
  738. }
  739. internal class TelnetTerminalConnection : TCPTerminalConnection, IDataReceiver {
  740. private AbstractGuevaraSocket _socket;
  741. private TelnetNegotiator _negotiator;
  742. public TelnetTerminalConnection(TerminalParam p, TelnetNegotiator neg, AbstractGuevaraSocket s, int width, int height) : base(p, width, height) {
  743. _socket = s;
  744. _negotiator = neg;
  745. }
  746. internal override void Close() {
  747. if(_closed) return; //俀搙埲忋僋儘乕僘偟偰傕暃嶌梡側偟 
  748. base.Close();
  749. try {
  750. _socket.Close();
  751. }
  752. catch(Exception ex) {
  753. GUtil.Warning(GEnv.Frame, GEnv.Strings.GetString("Message.SSHTerminalConnection.CloseError")+ex.Message);
  754. }
  755. }
  756. internal override void Disconnect() {
  757. }
  758. public override ConnectionTag Reproduce() {
  759. TerminalParam np = (TerminalParam)_param.Clone();
  760. if(GEnv.Options.DefaultLogType!=LogType.None) {
  761. np.LogType = GEnv.Options.DefaultLogType;
  762. np.LogPath = GUtil.CreateLogFileName(np.ShortDescription);
  763. }
  764. else
  765. np.LogType = LogType.None;
  766. return CommunicationUtil.CreateNewConnection(np);
  767. }
  768. public override string[] ConnectionParameter {
  769. get {
  770. string[] d = new string[1];
  771. d[0] = String.Format("");
  772. return d;
  773. }
  774. }
  775. public override bool Available {
  776. get {
  777. return !_closed && _socket.DataAvailable;
  778. }
  779. }
  780. private IDataReceiver _callback;
  781. internal override void RepeatAsyncRead(IDataReceiver cb) {
  782. if(_callback!=null) throw new InvalidOperationException("duplicated AsyncRead() is attempted");
  783. _callback = cb;
  784. _socket.RepeatAsyncRead(this);
  785. }
  786. public void DataArrived(byte[] buf, int offset, int count) {
  787. ProcessBuffer(buf, offset, offset+count);
  788. _negotiator.Flush(_socket);
  789. }
  790. public void DisconnectedFromServer() {
  791. if(!_closed)
  792. _callback.DisconnectedFromServer();
  793. }
  794. public void ErrorOccurred(string msg) {
  795. if(!_closed)
  796. _callback.ErrorOccurred(msg);
  797. }
  798. //CR NUL -> CR
  799. private void ProcessBuffer(byte[] buf, int offset, int limit) {
  800. while(offset < limit) {
  801. while(offset < limit && _negotiator.InProcessing) {
  802. if(_negotiator.Process(buf[offset++])==TelnetNegotiator.ProcessResult.REAL_0xFF)
  803. _callback.DataArrived(buf, offset-1, 1);
  804. }
  805. int delim = offset;
  806. while(delim < limit) {
  807. byte b = buf[delim];
  808. if(b==0xFF) {
  809. _negotiator.StartNegotiate();
  810. break;
  811. }
  812. if(b==0 && delim-1>=0 && buf[delim-1]==0x0D) break; //CR NUL僒億乕僩
  813. delim++;
  814. }
  815. if(delim>offset) _callback.DataArrived(buf, offset, delim-offset); //delim偺庤慜傑偱張棟
  816. offset = delim+1;
  817. }
  818. }
  819. public override void Resize(int width, int height) {
  820. base.Resize(width, height);
  821. if(!_closed) {
  822. TelnetOptionWriter wr = new TelnetOptionWriter();
  823. wr.WriteTerminalSize(width, height);
  824. wr.WriteTo(_socket);
  825. }
  826. }
  827. public override void Write(byte[] buf) {
  828. Write(buf, 0, buf.Length);
  829. }
  830. public override void Write(byte[] buf, int offset, int length) {
  831. AddSentDataStats(length);
  832. for(int i=0; i<length; i++) {
  833. byte t = buf[offset+i];
  834. if(t==0xFF || t==0x0D) { //0xFF傑偨偼CRLF埲奜偺CR傪尒偮偗偨傜
  835. WriteEscaping(buf, offset, length);
  836. return;
  837. }
  838. }
  839. _socket.Transmit(buf, offset, length); //戝掞偺応崌偼偙偆偄偆僨乕僞偼擖偭偰偄側偄偺偱丄崅懍壔偺偨傔偦偺傑傑憲傝弌偡
  840. }
  841. private void WriteEscaping(byte[] buf, int offset, int length) {
  842. byte[] newbuf = new byte[length*2];
  843. int newoffset = 0;
  844. for(int i=0; i<length; i++) {
  845. byte t = buf[offset+i];
  846. if(t==0xFF) {
  847. newbuf[newoffset++] = 0xFF;
  848. newbuf[newoffset++] = 0xFF; //俀屄
  849. }
  850. else if(t==0x0D/* && (i==length-1 || buf[offset+i+1]!=0x0A)*/) {
  851. newbuf[newoffset++] = 0x0D;
  852. newbuf[newoffset++] = 0x00;
  853. }
  854. else
  855. newbuf[newoffset++] = t;
  856. }
  857. _socket.Transmit(newbuf, 0, newoffset);
  858. }
  859. public override void AreYouThere() {
  860. byte[] data = new byte[2];
  861. data[0] = (byte)TelnetCode.IAC;
  862. data[1] = (byte)TelnetCode.AreYouThere;
  863. _socket.Transmit(data, 0, data.Length);
  864. }
  865. public override void SendBreak() {
  866. byte[] data = new byte[2];
  867. data[0] = (byte)TelnetCode.IAC;
  868. data[1] = (byte)TelnetCode.Break;
  869. _socket.Transmit(data, 0, data.Length);
  870. }
  871. public override void SendKeepAliveData() {
  872. byte[] data = new byte[2];
  873. data[0] = (byte)TelnetCode.IAC;
  874. data[1] = (byte)TelnetCode.NOP;
  875. _socket.Transmit(data, 0, data.Length);
  876. }
  877. }
  878. public class SerialTerminalConnection : TerminalConnection {
  879. private byte[] _buf;
  880. //僔儕傾儖偺旕摨婜捠怣傪偪傖傫偲傗傠偆偲偡傞偲.NET儔僀僽儔儕偱偼晄廫暘側偺偱傎傏API捈撉傒
  881. private IntPtr _fileHandle;
  882. private Win32.OVERLAPPED _readOL;
  883. private Win32.OVERLAPPED _writeOL;
  884. public SerialTerminalConnection(SerialTerminalParam p, IntPtr fh, int width, int height) : base(p, width, height) {
  885. _fileHandle = fh;
  886. _buf = new byte[0x1000];
  887. _readOL.hEvent  = Win32.CreateEvent(IntPtr.Zero, 0, 0, null);
  888. _writeOL.hEvent = Win32.CreateEvent(IntPtr.Zero, 0, 0, null);
  889. }
  890. internal override void Close() {
  891. if(_closed) return; //俀搙埲忋僋儘乕僘偟偰傕暃嶌梡側偟 
  892. base.Close();
  893. Win32.CloseHandle(_readOL.hEvent);
  894. Win32.CloseHandle(_writeOL.hEvent);
  895. Win32.CloseHandle(_fileHandle);
  896. _readOL.hEvent = _writeOL.hEvent = _fileHandle = IntPtr.Zero;
  897. //Debug.WriteLine("COM connection termingating...");
  898. }
  899. public override ConnectionTag Reproduce() {
  900. throw new NotSupportedException(GEnv.Strings.GetString("Message.SerialTerminalConnection.ReproduceError"));
  901. }
  902. internal override void Disconnect() {
  903. }
  904. public override string ProtocolDescription {
  905. get {
  906. return "-";
  907. }
  908. }
  909. public override string[] ConnectionParameter {
  910. get {
  911. string[] d = new string[6];
  912. SerialTerminalParam p = (SerialTerminalParam)_param;
  913. d[0] = String.Format("Port         : COM{0}", p.Port);
  914. d[1] = String.Format("Baud rate    : {0}", p.BaudRate);
  915. d[2] = String.Format("Data bits    : {0}", p.ByteSize);
  916. d[3] = String.Format("Stop bits    : {0}", EnumDescAttribute.For(typeof(StopBits)).GetDescription(p.StopBits));
  917. d[4] = String.Format("Parity       : {0}", EnumDescAttribute.For(typeof(Parity)).GetDescription(p.Parity));
  918. d[5] = String.Format("Flow Control : {0}", EnumDescAttribute.For(typeof(FlowControl)).GetDescription(p.FlowControl));
  919. return d;
  920. }
  921. }
  922. public override bool Available {
  923. get {
  924. return false; //僔儕傾儖偩偲傛偔傢偐傜側偄偺偱false傪曉偟偰偍偔
  925. }
  926. }
  927. private IDataReceiver _callback;
  928. internal override void RepeatAsyncRead(IDataReceiver cb) {
  929. if(_callback!=null) throw new InvalidOperationException("duplicated AsyncRead() is attempted");
  930. _callback = cb;
  931. GUtil.CreateThread(new ThreadStart(AsyncEntry)).Start();
  932. //_stream.BeginRead(_buf, 0, _buf.Length, new AsyncCallback(RepeatCallback), null);
  933. }
  934. private void AsyncEntry() {
  935. Win32.OVERLAPPED ol = new Win32.OVERLAPPED();
  936. try {
  937. //弶婜壔
  938. bool success = false;
  939. int len = 0, flags = 0;
  940. ol.hEvent = Win32.CreateEvent(IntPtr.Zero, 0, 0, null);
  941. success = Win32.ClearCommError(_fileHandle, out flags, IntPtr.Zero);
  942. //偙偺SetCommMask傪幚峴偟側偄偲WaitCommEvent偑幐攕偟偰偟傑偆
  943. success = Win32.SetCommMask(_fileHandle, 0);
  944. success = Win32.SetCommMask(_fileHandle, 1); //EV_RXCHAR
  945. byte[] buf = new byte[128];
  946. while(true) {
  947. success = Win32.WaitCommEvent(_fileHandle, out flags, ref ol); //偙偙偼晛捠false偑曉傞
  948. if(!success && Win32.GetLastError()!=Win32.ERROR_IO_PENDING)
  949. throw new Exception("WaitCommEvent failed " + Win32.GetLastError());
  950. //偙偙偱僨乕僞偑棃傞傑偱僽儘僢僋偡傞丅偨偩丄僪僉儏儊儞僩偵傛傟偽WaitCommEvent偵懳墳偡傞GetOverlappedResult偼len偺抣偼柍堄枴偲偺偙偲
  951. success = Win32.GetOverlappedResult(_fileHandle, ref ol, ref len, true);
  952. if(!success) break; //偨偲偊偽愙懕傪愗傞側偳偱false偑曉偭偰偔傞
  953. do {
  954. len = 0;
  955. success = Win32.ReadFile(_fileHandle, buf, buf.Length, ref len, ref _readOL);
  956. //偙偺WaitForSingleObject偑昁梫偐偳偆偐偑傛偔傢偐傜側偄
  957. //if(Win32.WaitForSingleObject(_readOL.hEvent, 5000)!=Win32.WAIT_OBJECT_0)
  958. // throw new Exception("WaitForSingleObject timed out");
  959. success = Win32.GetOverlappedResult(_fileHandle, ref _readOL, ref len, true);
  960. if(len>0) _callback.DataArrived(buf, 0, len);
  961. } while(len > 0);
  962. }
  963. }
  964. catch(Exception ex) {
  965. if(!_closed) {
  966. _callback.ErrorOccurred(ex.Message);
  967. }
  968. }
  969. finally {
  970. Win32.CloseHandle(ol.hEvent);
  971. //Debug.WriteLine("COM thread terminating...");
  972. }
  973. }
  974. public override void Write(byte[] buf) {
  975. Write(buf, 0, buf.Length);
  976. }
  977. public override void Write(byte[] data, int offset, int length) {
  978. SerialTerminalParam sp = (SerialTerminalParam)_param;
  979. if(sp.TransmitDelayPerChar==0) {
  980. if(sp.TransmitDelayPerLine==0)
  981. WriteMain(data, offset, length); //嵟傕扨弮
  982. else { //夵峴偺傒僂僃僀僩憓擖
  983. byte nl = (byte)(sp.TransmitNL==NewLine.CR? 13 : 10); 
  984. int limit = offset+length;
  985. int c = offset;
  986. while(offset<limit) {
  987. if(data[offset]==nl) {
  988. WriteMain(data, c, offset-c+1);
  989. Thread.Sleep(sp.TransmitDelayPerLine);
  990. System.Windows.Forms.Application.DoEvents();
  991. c = offset+1;
  992. }
  993. offset++;
  994. }
  995. if(c < limit) WriteMain(data, c, limit-c);
  996. }
  997. }
  998. else {
  999. byte nl = (byte)(sp.TransmitNL==NewLine.CR? 13 : 10); 
  1000. for(int i=0; i<length; i++) {
  1001. WriteMain(data, offset+i, 1);
  1002. Thread.Sleep(data[offset+i]==nl? sp.TransmitDelayPerLine : sp.TransmitDelayPerChar);
  1003. System.Windows.Forms.Application.DoEvents();
  1004. }
  1005. }
  1006. }
  1007. private void WriteMain(byte[] buf, int offset, int length) {
  1008. if(length==0) return; //挿偝侽偩偲 fixed(byte* p = buf) 偑幐攕偟偰偟傑偆
  1009. AddSentDataStats(length);
  1010. //_stream.Write(buf, 0, buf.Length);
  1011. int result = 0;
  1012. if(offset!=0) {
  1013. byte[] nb = new byte[length];
  1014. Array.Copy(buf, offset, nb, 0, length);
  1015. buf = nb; //師偺WriteFile偱offset偑侽偱側偄偲偒偼偡偖偵偼僒億乕僩偱偒側偄
  1016. }
  1017. bool success;
  1018. success = Win32.WriteFile(_fileHandle, buf, length, ref result, ref _writeOL);
  1019. if(Win32.GetLastError()!=Win32.ERROR_IO_PENDING)
  1020. throw new IOException("WriteFile failed for " + Win32.GetLastError());
  1021. //偙偺WaitForSingleObject偑昁梫偐偳偆偐偑傛偔傢偐傜側偄
  1022. //err = Win32.WaitForSingleObject(_writeOL.hEvent, 10000);
  1023. success = Win32.GetOverlappedResult(_fileHandle, ref _writeOL, ref result, true);
  1024. Debug.Assert(success);
  1025. }
  1026. public override void SendBreak() {
  1027. Win32.SetCommBreak(_fileHandle);
  1028. System.Threading.Thread.Sleep(500); //500ms懸婡
  1029. Win32.ClearCommBreak(_fileHandle);
  1030. }
  1031. public void ApplySerialParam(SerialTerminalParam param) {
  1032. //param偺撪梕偱DCB傪峏怴偟偰僙僢僩偟側偍偡
  1033. Win32.DCB dcb = new Win32.DCB();
  1034. CommunicationUtil.FillDCB(_fileHandle, ref dcb);
  1035. CommunicationUtil.UpdateDCB(ref dcb, param);
  1036. if(!Win32.SetCommState(_fileHandle, ref dcb))
  1037. throw new ArgumentException(GEnv.Strings.GetString("Message.SerialTerminalConnection.ConfigError"));
  1038. _param = param; //SetCommState偑惉岟偟偨傜峏怴
  1039. }
  1040. }
  1041. /*
  1042. public class ProcessConnection : TerminalConnection {
  1043. private Process _process;
  1044. private byte[] _bufOut;
  1045. private byte[] _bufErr;
  1046. public ProcessConnection(ProcessTerminalParam param, Process proc, int width, int height) : base(param, width, height) {
  1047. _process = proc;
  1048. _bufOut = new byte[0x1000];
  1049. _bufErr = new byte[0x1000];
  1050. }
  1051. internal override void Close() {
  1052. if(_closed) return; //俀搙埲忋僋儘乕僘偟偰傕暃嶌梡側偟 
  1053. base.Close();
  1054. }
  1055. public override ConnectionTag Reproduce() {
  1056. throw new NotSupportedException("僾儘僙僗傊偺愙懕偺暋惢偼偱偒傑偣傫丅");
  1057. }
  1058. internal override void Disconnect() {
  1059. _process.Close();
  1060. }
  1061. public override string ProtocolDescription {
  1062. get {
  1063. return "-";
  1064. }
  1065. }
  1066. public override string[] ConnectionParameter {
  1067. get {
  1068. string[] d = new string[1];
  1069. d[0] = ((ProcessTerminalParam)_param).Process;
  1070. return d;
  1071. }
  1072. }
  1073. public override bool Available {
  1074. get {
  1075. return false; 
  1076. }
  1077. }
  1078. private IDataReceiver _callback;
  1079. internal override void RepeatAsyncRead(IDataReceiver cb) {
  1080. if(_callback!=null) throw new InvalidOperationException("duplicated AsyncRead() is attempted");
  1081. _callback = cb;
  1082. _process.StandardOutput.BaseStream.BeginRead(_bufOut, 0, _bufOut.Length, new AsyncCallback(RepeatCallback), _bufOut);
  1083. _process.StandardError. BaseStream.BeginRead(_bufErr, 0, _bufErr.Length, new AsyncCallback(RepeatCallback), _bufErr);
  1084. }
  1085. private void RepeatCallback(IAsyncResult res) {
  1086. lock(this) {
  1087. try {
  1088. bool stderr = (res.AsyncState==_bufErr);
  1089. Stream strm = stderr? _process.StandardError.BaseStream : _process.StandardOutput.BaseStream;
  1090. int len = strm.EndRead(res);
  1091. //Debug.WriteLine(String.Format("RC {0} {1}", stderr, len));
  1092. byte[] buf = stderr? _bufErr : _bufOut;
  1093. if(len > 0) {
  1094. _callback.DataArrived(buf, 0, len);
  1095. strm.BeginRead(buf, 0, buf.Length, new AsyncCallback(RepeatCallback), buf);
  1096. }
  1097. else if(len < 0) {
  1098. strm.BeginRead(buf, 0, buf.Length, new AsyncCallback(RepeatCallback), buf);
  1099. }
  1100. else
  1101. _callback.DisconnectedFromServer();
  1102. }
  1103. catch(Exception ex) {
  1104. _callback.ErrorOccurred(ex.Message);
  1105. }
  1106. }
  1107. }
  1108. public override void Write(byte[] buf) {
  1109. Write(buf, 0, buf.Length);
  1110. }
  1111. public override void Write(byte[] data, int offset, int length) {
  1112. _process.StandardInput.BaseStream.Write(data, offset, length);
  1113. _process.StandardInput.BaseStream.Flush();
  1114. }
  1115. }
  1116. */
  1117. public class FakeConnection : TerminalConnection {
  1118. public FakeConnection(TerminalParam param) : base(param, 80, 25) {}
  1119. public override void Write(byte[] data) {
  1120. }
  1121. public override void Write(byte[] data, int offset, int length) {
  1122. }
  1123. internal override void RepeatAsyncRead(IDataReceiver r) {
  1124. }
  1125. public override string[] ConnectionParameter {
  1126. get {
  1127. return new string[1] { "fake connection" };
  1128. }
  1129. }
  1130. public override bool Available {
  1131. get {
  1132. return false;
  1133. }
  1134. }
  1135. public override string ProtocolDescription {
  1136. get {
  1137. return "";
  1138. }
  1139. }
  1140. public override ConnectionTag Reproduce() {
  1141. throw new NotSupportedException();
  1142. }
  1143. }
  1144. }