├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── conn ├── connection.py ├── connhost ├── connplay ├── logs └── .test ├── record.py ├── requirements.txt ├── script ├── scriptreplay └── videos └── .test /.gitignore: -------------------------------------------------------------------------------- 1 | # Byte-compiled / optimized / DLL files 2 | __pycache__/ 3 | *.py[cod] 4 | 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | GNU GENERAL PUBLIC LICENSE 2 | Version 2, June 1991 3 | 4 | Copyright (C) 1989, 1991 Free Software Foundation, Inc., 5 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 6 | Everyone is permitted to copy and distribute verbatim copies 7 | of this license document, but changing it is not allowed. 8 | 9 | Preamble 10 | 11 | The licenses for most software are designed to take away your 12 | freedom to share and change it. By contrast, the GNU General Public 13 | License is intended to guarantee your freedom to share and change free 14 | software--to make sure the software is free for all its users. This 15 | General Public License applies to most of the Free Software 16 | Foundation's software and to any other program whose authors commit to 17 | using it. (Some other Free Software Foundation software is covered by 18 | the GNU Lesser General Public License instead.) You can apply it to 19 | your programs, too. 20 | 21 | When we speak of free software, we are referring to freedom, not 22 | price. Our General Public Licenses are designed to make sure that you 23 | have the freedom to distribute copies of free software (and charge for 24 | this service if you wish), that you receive source code or can get it 25 | if you want it, that you can change the software or use pieces of it 26 | in new free programs; and that you know you can do these things. 27 | 28 | To protect your rights, we need to make restrictions that forbid 29 | anyone to deny you these rights or to ask you to surrender the rights. 30 | These restrictions translate to certain responsibilities for you if you 31 | distribute copies of the software, or if you modify it. 32 | 33 | For example, if you distribute copies of such a program, whether 34 | gratis or for a fee, you must give the recipients all the rights that 35 | you have. You must make sure that they, too, receive or can get the 36 | source code. And you must show them these terms so they know their 37 | rights. 38 | 39 | We protect your rights with two steps: (1) copyright the software, and 40 | (2) offer you this license which gives you legal permission to copy, 41 | distribute and/or modify the software. 42 | 43 | Also, for each author's protection and ours, we want to make certain 44 | that everyone understands that there is no warranty for this free 45 | software. If the software is modified by someone else and passed on, we 46 | want its recipients to know that what they have is not the original, so 47 | that any problems introduced by others will not reflect on the original 48 | authors' reputations. 49 | 50 | Finally, any free program is threatened constantly by software 51 | patents. We wish to avoid the danger that redistributors of a free 52 | program will individually obtain patent licenses, in effect making the 53 | program proprietary. To prevent this, we have made it clear that any 54 | patent must be licensed for everyone's free use or not licensed at all. 55 | 56 | The precise terms and conditions for copying, distribution and 57 | modification follow. 58 | 59 | GNU GENERAL PUBLIC LICENSE 60 | TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION 61 | 62 | 0. This License applies to any program or other work which contains 63 | a notice placed by the copyright holder saying it may be distributed 64 | under the terms of this General Public License. The "Program", below, 65 | refers to any such program or work, and a "work based on the Program" 66 | means either the Program or any derivative work under copyright law: 67 | that is to say, a work containing the Program or a portion of it, 68 | either verbatim or with modifications and/or translated into another 69 | language. (Hereinafter, translation is included without limitation in 70 | the term "modification".) Each licensee is addressed as "you". 71 | 72 | Activities other than copying, distribution and modification are not 73 | covered by this License; they are outside its scope. The act of 74 | running the Program is not restricted, and the output from the Program 75 | is covered only if its contents constitute a work based on the 76 | Program (independent of having been made by running the Program). 77 | Whether that is true depends on what the Program does. 78 | 79 | 1. You may copy and distribute verbatim copies of the Program's 80 | source code as you receive it, in any medium, provided that you 81 | conspicuously and appropriately publish on each copy an appropriate 82 | copyright notice and disclaimer of warranty; keep intact all the 83 | notices that refer to this License and to the absence of any warranty; 84 | and give any other recipients of the Program a copy of this License 85 | along with the Program. 86 | 87 | You may charge a fee for the physical act of transferring a copy, and 88 | you may at your option offer warranty protection in exchange for a fee. 89 | 90 | 2. You may modify your copy or copies of the Program or any portion 91 | of it, thus forming a work based on the Program, and copy and 92 | distribute such modifications or work under the terms of Section 1 93 | above, provided that you also meet all of these conditions: 94 | 95 | a) You must cause the modified files to carry prominent notices 96 | stating that you changed the files and the date of any change. 97 | 98 | b) You must cause any work that you distribute or publish, that in 99 | whole or in part contains or is derived from the Program or any 100 | part thereof, to be licensed as a whole at no charge to all third 101 | parties under the terms of this License. 102 | 103 | c) If the modified program normally reads commands interactively 104 | when run, you must cause it, when started running for such 105 | interactive use in the most ordinary way, to print or display an 106 | announcement including an appropriate copyright notice and a 107 | notice that there is no warranty (or else, saying that you provide 108 | a warranty) and that users may redistribute the program under 109 | these conditions, and telling the user how to view a copy of this 110 | License. (Exception: if the Program itself is interactive but 111 | does not normally print such an announcement, your work based on 112 | the Program is not required to print an announcement.) 113 | 114 | These requirements apply to the modified work as a whole. If 115 | identifiable sections of that work are not derived from the Program, 116 | and can be reasonably considered independent and separate works in 117 | themselves, then this License, and its terms, do not apply to those 118 | sections when you distribute them as separate works. But when you 119 | distribute the same sections as part of a whole which is a work based 120 | on the Program, the distribution of the whole must be on the terms of 121 | this License, whose permissions for other licensees extend to the 122 | entire whole, and thus to each and every part regardless of who wrote it. 123 | 124 | Thus, it is not the intent of this section to claim rights or contest 125 | your rights to work written entirely by you; rather, the intent is to 126 | exercise the right to control the distribution of derivative or 127 | collective works based on the Program. 128 | 129 | In addition, mere aggregation of another work not based on the Program 130 | with the Program (or with a work based on the Program) on a volume of 131 | a storage or distribution medium does not bring the other work under 132 | the scope of this License. 133 | 134 | 3. You may copy and distribute the Program (or a work based on it, 135 | under Section 2) in object code or executable form under the terms of 136 | Sections 1 and 2 above provided that you also do one of the following: 137 | 138 | a) Accompany it with the complete corresponding machine-readable 139 | source code, which must be distributed under the terms of Sections 140 | 1 and 2 above on a medium customarily used for software interchange; or, 141 | 142 | b) Accompany it with a written offer, valid for at least three 143 | years, to give any third party, for a charge no more than your 144 | cost of physically performing source distribution, a complete 145 | machine-readable copy of the corresponding source code, to be 146 | distributed under the terms of Sections 1 and 2 above on a medium 147 | customarily used for software interchange; or, 148 | 149 | c) Accompany it with the information you received as to the offer 150 | to distribute corresponding source code. (This alternative is 151 | allowed only for noncommercial distribution and only if you 152 | received the program in object code or executable form with such 153 | an offer, in accord with Subsection b above.) 154 | 155 | The source code for a work means the preferred form of the work for 156 | making modifications to it. For an executable work, complete source 157 | code means all the source code for all modules it contains, plus any 158 | associated interface definition files, plus the scripts used to 159 | control compilation and installation of the executable. However, as a 160 | special exception, the source code distributed need not include 161 | anything that is normally distributed (in either source or binary 162 | form) with the major components (compiler, kernel, and so on) of the 163 | operating system on which the executable runs, unless that component 164 | itself accompanies the executable. 165 | 166 | If distribution of executable or object code is made by offering 167 | access to copy from a designated place, then offering equivalent 168 | access to copy the source code from the same place counts as 169 | distribution of the source code, even though third parties are not 170 | compelled to copy the source along with the object code. 171 | 172 | 4. You may not copy, modify, sublicense, or distribute the Program 173 | except as expressly provided under this License. Any attempt 174 | otherwise to copy, modify, sublicense or distribute the Program is 175 | void, and will automatically terminate your rights under this License. 176 | However, parties who have received copies, or rights, from you under 177 | this License will not have their licenses terminated so long as such 178 | parties remain in full compliance. 179 | 180 | 5. You are not required to accept this License, since you have not 181 | signed it. However, nothing else grants you permission to modify or 182 | distribute the Program or its derivative works. These actions are 183 | prohibited by law if you do not accept this License. Therefore, by 184 | modifying or distributing the Program (or any work based on the 185 | Program), you indicate your acceptance of this License to do so, and 186 | all its terms and conditions for copying, distributing or modifying 187 | the Program or works based on it. 188 | 189 | 6. Each time you redistribute the Program (or any work based on the 190 | Program), the recipient automatically receives a license from the 191 | original licensor to copy, distribute or modify the Program subject to 192 | these terms and conditions. You may not impose any further 193 | restrictions on the recipients' exercise of the rights granted herein. 194 | You are not responsible for enforcing compliance by third parties to 195 | this License. 196 | 197 | 7. If, as a consequence of a court judgment or allegation of patent 198 | infringement or for any other reason (not limited to patent issues), 199 | conditions are imposed on you (whether by court order, agreement or 200 | otherwise) that contradict the conditions of this License, they do not 201 | excuse you from the conditions of this License. If you cannot 202 | distribute so as to satisfy simultaneously your obligations under this 203 | License and any other pertinent obligations, then as a consequence you 204 | may not distribute the Program at all. For example, if a patent 205 | license would not permit royalty-free redistribution of the Program by 206 | all those who receive copies directly or indirectly through you, then 207 | the only way you could satisfy both it and this License would be to 208 | refrain entirely from distribution of the Program. 209 | 210 | If any portion of this section is held invalid or unenforceable under 211 | any particular circumstance, the balance of the section is intended to 212 | apply and the section as a whole is intended to apply in other 213 | circumstances. 214 | 215 | It is not the purpose of this section to induce you to infringe any 216 | patents or other property right claims or to contest validity of any 217 | such claims; this section has the sole purpose of protecting the 218 | integrity of the free software distribution system, which is 219 | implemented by public license practices. Many people have made 220 | generous contributions to the wide range of software distributed 221 | through that system in reliance on consistent application of that 222 | system; it is up to the author/donor to decide if he or she is willing 223 | to distribute software through any other system and a licensee cannot 224 | impose that choice. 225 | 226 | This section is intended to make thoroughly clear what is believed to 227 | be a consequence of the rest of this License. 228 | 229 | 8. If the distribution and/or use of the Program is restricted in 230 | certain countries either by patents or by copyrighted interfaces, the 231 | original copyright holder who places the Program under this License 232 | may add an explicit geographical distribution limitation excluding 233 | those countries, so that distribution is permitted only in or among 234 | countries not thus excluded. In such case, this License incorporates 235 | the limitation as if written in the body of this License. 236 | 237 | 9. The Free Software Foundation may publish revised and/or new versions 238 | of the General Public License from time to time. Such new versions will 239 | be similar in spirit to the present version, but may differ in detail to 240 | address new problems or concerns. 241 | 242 | Each version is given a distinguishing version number. If the Program 243 | specifies a version number of this License which applies to it and "any 244 | later version", you have the option of following the terms and conditions 245 | either of that version or of any later version published by the Free 246 | Software Foundation. If the Program does not specify a version number of 247 | this License, you may choose any version ever published by the Free Software 248 | Foundation. 249 | 250 | 10. If you wish to incorporate parts of the Program into other free 251 | programs whose distribution conditions are different, write to the author 252 | to ask for permission. For software which is copyrighted by the Free 253 | Software Foundation, write to the Free Software Foundation; we sometimes 254 | make exceptions for this. Our decision will be guided by the two goals 255 | of preserving the free status of all derivatives of our free software and 256 | of promoting the sharing and reuse of software generally. 257 | 258 | NO WARRANTY 259 | 260 | 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY 261 | FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN 262 | OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES 263 | PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED 264 | OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 265 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS 266 | TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE 267 | PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING, 268 | REPAIR OR CORRECTION. 269 | 270 | 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING 271 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR 272 | REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, 273 | INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING 274 | OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED 275 | TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY 276 | YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER 277 | PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE 278 | POSSIBILITY OF SUCH DAMAGES. 279 | 280 | END OF TERMS AND CONDITIONS 281 | 282 | How to Apply These Terms to Your New Programs 283 | 284 | If you develop a new program, and you want it to be of the greatest 285 | possible use to the public, the best way to achieve this is to make it 286 | free software which everyone can redistribute and change under these terms. 287 | 288 | To do so, attach the following notices to the program. It is safest 289 | to attach them to the start of each source file to most effectively 290 | convey the exclusion of warranty; and each file should have at least 291 | the "copyright" line and a pointer to where the full notice is found. 292 | 293 | {description} 294 | Copyright (C) {year} {fullname} 295 | 296 | This program is free software; you can redistribute it and/or modify 297 | it under the terms of the GNU General Public License as published by 298 | the Free Software Foundation; either version 2 of the License, or 299 | (at your option) any later version. 300 | 301 | This program is distributed in the hope that it will be useful, 302 | but WITHOUT ANY WARRANTY; without even the implied warranty of 303 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 304 | GNU General Public License for more details. 305 | 306 | You should have received a copy of the GNU General Public License along 307 | with this program; if not, write to the Free Software Foundation, Inc., 308 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 309 | 310 | Also add information on how to contact you by electronic and paper mail. 311 | 312 | If the program is interactive, make it output a short notice like this 313 | when it starts in an interactive mode: 314 | 315 | Gnomovision version 69, Copyright (C) year name of author 316 | Gnomovision comes with ABSOLUTELY NO WARRANTY; for details type `show w'. 317 | This is free software, and you are welcome to redistribute it 318 | under certain conditions; type `show c' for details. 319 | 320 | The hypothetical commands `show w' and `show c' should show the appropriate 321 | parts of the General Public License. Of course, the commands you use may 322 | be called something other than `show w' and `show c'; they could even be 323 | mouse-clicks or menu items--whatever suits your program. 324 | 325 | You should also get your employer (if you work as a programmer) or your 326 | school, if any, to sign a "copyright disclaimer" for the program, if 327 | necessary. Here is a sample; alter the names: 328 | 329 | Yoyodyne, Inc., hereby disclaims all copyright interest in the program 330 | `Gnomovision' (which makes passes at compilers) written by James Hacker. 331 | 332 | {signature of Ty Coon}, 1 April 1989 333 | Ty Coon, President of Vice 334 | 335 | This General Public License does not permit incorporating your program into 336 | proprietary programs. If your program is a subroutine library, you may 337 | consider it more useful to permit linking proprietary applications with the 338 | library. If this is what you want to do, use the GNU Lesser General 339 | Public License instead of this License. 340 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | conn 2 | ==== 3 | 4 | 审计在运维中占有很重要的一部分，审计追溯历史问题.conn是connection的前缀. 5 | 6 | conn堡垒机:conn修改了paramiko的源码,支持详细日志审计和录像功能,conn服务器上每一步操作都能记录. 7 | conn命令: 登录远端服务器，并记录所有操作和录像 8 | connplay: 播放历史录像，审核服务器记录 9 | connhost: 正向和反向解析主机名或ip地址(可批量) 10 | 11 | 12 | 安装部署 13 | ``` 14 | 堡垒机安装 15 | git clone https://github.com/shanhuhai5739/conn.git 16 | cd conn 17 | pip install -r requirements.txt -i http://pypi.douban.com/simple/ 18 | 客户机安装 19 | yum install xterm -y 20 | 21 | 如果想让conn支持tab显示主机，将下面命令编辑到/etc/profile 22 | complete -W "$(echo `cat ~/.ssh/known_hosts | cut -f 1 -d ' ' | sed -e s/,.*//g | uniq | grep -v "\["`;)" conn 23 | source /etc/profile 24 | ``` 25 | 26 | 登录主机 27 | ``` 28 | conn 用户@主机名|IP:端口 29 | 支持rsa dsa passwd方式登录，优先rsa登录 30 | 31 | conn coral@118.244.168.45:52101 32 | ``` 33 | 34 | 日志审计 35 | ``` 36 | 20140827-118.244.168.45-coral-detail.log 37 | 日志格式: 时间-主机-用户 38 | 39 | 两种日志格式: 40 | logs/*.log 操作命令记录 41 | logs/*detail.log 操作命令结果详细记录 42 | 用户在登录状态下可以实时看到输出结果,tail -f xxx-xxx-xxx.log 43 | ``` 44 | 45 | 录像审计 46 | ``` 47 | 20140827173556-118.244.168.45-coral 48 | 录像格式: 时间-主机-用户 49 | 50 | videos/*.his 操作结果录像记录 51 | 用户推出后生成.his和.time文件，使用hostplay进行播放 52 | connplay ls 显示所有录像记录文件 53 | connplay play 20140827173556-118.244.168.45-coral 播放该记录文件 54 | ``` 55 | 56 | 主机(域名)解析 57 | ``` 58 | connhost 10.237.39.27 59 | 10.237.39.27 | hh-c2c-dev01.bj 60 | connhost hh-c2c-dev01.bj 61 | hh-c2c-dev01.bj | 10.237.39.27 62 | 63 | 支持文件批量解析 64 | connhost -f ipfile 65 | 10.100.2.43 | hh-c2c-dev11.bj 66 | 10.100.2.49 | hh-c2c-dev12.bj 67 | ``` 68 | 69 | 联系我(珊瑚) 70 | ``` 71 | QQ : 919953500 72 | QQ群: 222496448 73 | ``` 74 | -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | 3 | import os, sys 4 | 5 | conn_path = os.path.dirname(os.path.abspath(__file__)) 6 | os.chdir(conn_path) 7 | -------------------------------------------------------------------------------- /conn: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | import os, sys 3 | import getpass 4 | # Import conn Lib 5 | import __init__ 6 | from record import record_videos 7 | 8 | # set linux tab 9 | #if os.path.isfile(os.path.expanduser('~/.ssh/known_hosts')): 10 | # cmd = '''complete -W "$(echo `cat ~/.ssh/known_hosts | cut -f 1 -d ' ' | sed -e s/,.*//g | uniq | grep -v "\["`;)" %s''' % sys.argv[0] 11 | # os.system(cmd) 12 | 13 | 14 | def usage(): 15 | print 16 | print '''\033[;32m connection remote hostname operator and record all infomation. 17 | 18 | Usage: %s hostname or ip 19 | %s user@hostname:port 20 | 21 | %s by 919953500@qq.com 22 | \033[0m''' % (sys.argv[0],sys.argv[0],sys.argv[0]) 23 | sys.exit() 24 | 25 | @record_videos 26 | def run(username, hostname, port): 27 | print hostname, 'closed.' 28 | 29 | # currently username 30 | username = getpass.getuser() 31 | if len(sys.argv) > 1: 32 | hostname = sys.argv[1] 33 | if hostname.find('@') >= 0: 34 | username, hostname = hostname.split('@') 35 | else: 36 | usage() 37 | sys.exit(1) 38 | port = 22 39 | if hostname.find(':') >= 0: 40 | hostname, portstr = hostname.split(':') 41 | port = int(portstr) 42 | 43 | run(username, hostname, port) 44 | 45 | 46 | 47 | -------------------------------------------------------------------------------- /connection.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | 3 | import base64 4 | from binascii import hexlify 5 | import getpass 6 | import os 7 | import select 8 | import socket 9 | import sys 10 | import time 11 | import traceback 12 | import paramiko 13 | from paramiko.py3compat import input 14 | from paramiko.py3compat import u 15 | import getpass 16 | # Import conn Lib 17 | from record import record_videos 18 | 19 | 20 | # windows does not have termios... 21 | try: 22 | import termios 23 | import tty 24 | has_termios = True 25 | except ImportError: 26 | has_termios = False 27 | 28 | def agent_auth(transport, username): 29 | """ 30 | Attempt to authenticate to the given transport using any of the private 31 | keys available from an SSH agent. 32 | """ 33 | 34 | agent = paramiko.Agent() 35 | agent_keys = agent.get_keys() 36 | if len(agent_keys) == 0: 37 | return 38 | 39 | for key in agent_keys: 40 | print('Trying ssh-agent key %s' % hexlify(key.get_fingerprint())) 41 | try: 42 | transport.auth_publickey(username, key) 43 | print('... success!') 44 | return 45 | except paramiko.SSHException: 46 | print('... nope.') 47 | 48 | 49 | def manual_auth(username, hostname): 50 | try: 51 | default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa') 52 | path = default_path 53 | try: 54 | key = paramiko.RSAKey.from_private_key_file(path) 55 | except paramiko.PasswordRequiredException: 56 | password = getpass.getpass('RSA key password: ') 57 | key = paramiko.RSAKey.from_private_key_file(path, password) 58 | t.auth_publickey(username, key) 59 | except IOError: 60 | default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_dsa') 61 | path = default_path 62 | try: 63 | key = paramiko.DSSKey.from_private_key_file(path) 64 | except paramiko.PasswordRequiredException: 65 | password = getpass.getpass('DSS key password: ') 66 | key = paramiko.DSSKey.from_private_key_file(path, password) 67 | t.auth_publickey(username, key) 68 | #except paramiko.AuthenticationException as e: 69 | except: 70 | pw = getpass.getpass('Password for %s@%s: ' % (username, hostname)) 71 | try: 72 | t.auth_password(username, pw) 73 | except (paramiko.AuthenticationException, KeyboardInterrupt) as error: 74 | print 'Permission denied, Authentication Error' 75 | 76 | 77 | username = getpass.getuser() 78 | if len(sys.argv) > 1: 79 | hostname = sys.argv[1] 80 | if hostname.find('@') >= 0: 81 | username, hostname = hostname.split('@') 82 | else: 83 | hostname = input('Hostname: ') 84 | if len(hostname) == 0: 85 | print('*** Hostname required.') 86 | sys.exit(1) 87 | port = 22 88 | if hostname.find(':') >= 0: 89 | hostname, portstr = hostname.split(':') 90 | port = int(portstr) 91 | 92 | # now connect 93 | try: 94 | sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 95 | sock.connect((hostname, port)) 96 | except Exception as e: 97 | print('*** Connect failed: ' + str(e)) 98 | sys.exit(1) 99 | 100 | 101 | def posix_shell(chan): 102 | import select 103 | record = [] 104 | 105 | oldtty = termios.tcgetattr(sys.stdin) 106 | try: 107 | tty.setraw(sys.stdin.fileno()) 108 | tty.setcbreak(sys.stdin.fileno()) 109 | chan.settimeout(0.0) 110 | 111 | 112 | # needs to be sent to give vim correct size FIX 113 | chan.send('eval $(resize)\n') 114 | 115 | 116 | day =time.strftime('%Y%m%d') 117 | reduce_log = open('logs/%s-%s-%s.log' % (day, hostname, username), 'a') 118 | detail_log = open('logs/%s-%s-%s-detail.log' % (day, hostname, username), 'a') 119 | while True: 120 | date =time.strftime('%Y-%m-%d %H:%M:%S') 121 | r, w, e = select.select([chan, sys.stdin], [], []) 122 | if chan in r: 123 | try: 124 | x = u(chan.recv(1024)) 125 | if len(x) == 0: 126 | sys.stdout.write('\r\n*** EOF\r\n') 127 | break 128 | sys.stdout.write(x) 129 | sys.stdout.flush() 130 | except socket.timeout: 131 | pass 132 | 133 | detail_log.write('%s' % x.encode('utf8')) 134 | detail_log.flush() 135 | 136 | if sys.stdin in r: 137 | #x = sys.stdin.read(1) 138 | # history command 139 | x = os.read(sys.stdin.fileno(), 1) 140 | if len(x) == 0: 141 | break 142 | 143 | # conn record ... 144 | record.append(x) 145 | if x == '\b': 146 | try: 147 | record.pop() 148 | record.pop() 149 | except IndexError: 150 | pass 151 | #print record,'-' 152 | if x == '\r': 153 | cmd = ''.join(record).split('\r')[-2] 154 | if len(cmd) != 0: 155 | reduce_log.write('%s\t%s\t%s\t%s\n' % (date, 'hh-b2c-miorder-backend-web01.bj', 'root', cmd)) 156 | reduce_log.flush() 157 | #print record,'-' 158 | record = [] 159 | chan.send(x) 160 | 161 | reduce_log.close() 162 | detail_log.close() 163 | 164 | finally: 165 | termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty) 166 | 167 | 168 | try: 169 | t = paramiko.Transport(sock) 170 | try: 171 | t.start_client() 172 | except paramiko.SSHException: 173 | print('*** SSH negotiation failed.') 174 | sys.exit(1) 175 | 176 | try: 177 | keys = paramiko.util.load_host_keys(os.path.expanduser('~/.ssh/known_hosts')) 178 | except IOError: 179 | try: 180 | keys = paramiko.util.load_host_keys(os.path.expanduser('~/ssh/known_hosts')) 181 | except IOError: 182 | print('*** Unable to open host keys file') 183 | keys = {} 184 | 185 | # check server's host key -- this is important. 186 | key = t.get_remote_server_key() 187 | if hostname not in keys: 188 | print('*** WARNING: Unknown host key!') 189 | elif key.get_name() not in keys[hostname]: 190 | print('*** WARNING: Unknown host key!') 191 | elif keys[hostname][key.get_name()] != key: 192 | print('*** WARNING: Host key has changed!!!') 193 | sys.exit(1) 194 | else: 195 | print('*** Host key OK.') 196 | 197 | # get username 198 | if username == '': 199 | default_username = getpass.getuser() 200 | username = input('Username [%s]: ' % default_username) 201 | if len(username) == 0: 202 | username = default_username 203 | 204 | agent_auth(t, username) 205 | if not t.is_authenticated(): 206 | manual_auth(username, hostname) 207 | if not t.is_authenticated(): 208 | print('*** Authentication failed. :(') 209 | t.close() 210 | sys.exit(1) 211 | 212 | chan = t.open_session() 213 | chan.get_pty() 214 | chan.invoke_shell() 215 | print('*** Here we go!\n') 216 | 217 | #interactive.interactive_shell(chan) 218 | posix_shell(chan) 219 | chan.close() 220 | t.close() 221 | 222 | except Exception as e: 223 | print('*** Caught exception: ' + str(e.__class__) + ': ' + str(e)) 224 | traceback.print_exc() 225 | try: 226 | t.close() 227 | except: 228 | pass 229 | sys.exit(1) 230 | 231 | 232 | -------------------------------------------------------------------------------- /connhost: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding=utf8 3 | 4 | import socket 5 | import sys 6 | 7 | port = int(22) 8 | 9 | def Usage(): 10 | print 11 | print '''\033[;32m resolve hostname or ip 12 | 13 | Usage: %s hostname or ip 14 | %s -f file 15 | 16 | %s by 919953500@qq.com 17 | \033[0m''' % (sys.argv[0],sys.argv[0],sys.argv[0]) 18 | sys.exit() 19 | 20 | 21 | try: 22 | if len(sys.argv[0]) == 1 or sys.argv[1] == "-h" or sys.argv[1] == "--help": 23 | Usage() 24 | except IndexError: 25 | Usage() 26 | 27 | 28 | # 将IP地址解析成主机名 29 | def ip_to_host(ip): 30 | try: 31 | result = socket.gethostbyaddr(ip) 32 | print result[0] + ' | ' + ip 33 | except (socket.gaierror, socket.herror): 34 | print '\033[;33m IP : %s Not found\033[0m' % ip 35 | 36 | # 将主机名解析成IP地址 37 | def host_to_ip(host): 38 | try: 39 | #result = socket.getaddrinfo(host,port) 40 | result = socket.getaddrinfo(host,None) 41 | print result[0][-1][0] + ' | ' + host 42 | except (socket.gaierror, socket.herror): 43 | print '\033[;33m HOST : %s Not found\033[0m' % host 44 | 45 | # 验证IP是否合法 46 | def valid_ip(ip): 47 | try: 48 | socket.inet_aton(ip) 49 | return 'True' 50 | except: 51 | return 'False' 52 | 53 | 54 | # -f 指定文件批量操作 55 | argv = sys.argv[1] 56 | if argv == '-f': 57 | try: 58 | file = sys.argv[2] 59 | f = open(file,'r') 60 | for line in f.readlines(): 61 | line = line.strip().strip('\n') 62 | if valid_ip(line) == 'True': 63 | ip_to_host(line) 64 | elif valid_ip(line) == 'False': 65 | host_to_ip(line) 66 | f.close() 67 | except: 68 | print '\033[;31m Please specify a file\033[0m' 69 | elif valid_ip(argv) == 'True': 70 | ip_to_host(argv) 71 | elif valid_ip(argv) == 'False': 72 | host_to_ip(argv) 73 | else: 74 | Usage() 75 | -------------------------------------------------------------------------------- /connplay: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | import os, sys 3 | import __init__ 4 | 5 | def usage(): 6 | print 7 | print '''\033[;32m replay video of record infomation. 8 | 9 | Usage: %s ls display all videos list. 10 | %s play video_name play video file. 11 | 12 | %s by 919953500@qq.com 13 | \033[0m''' % (sys.argv[0],sys.argv[0],sys.argv[0]) 14 | sys.exit() 15 | 16 | try: 17 | videolist = [] 18 | for root, dirs, files in os.walk('videos'): 19 | for f in files: 20 | if not f[:1] == '.': 21 | videolist.append(f.replace('.time', '').replace('.his', '')) 22 | videolist = list(set(videolist)) 23 | 24 | if sys.argv[1] == 'ls': 25 | for i in videolist: print i 26 | elif sys.argv[1] == 'play': 27 | try: 28 | if sys.argv[2] in videolist: 29 | print sys.argv[2] 30 | cmd = './scriptreplay videos/%s.time videos/%s.his' % (sys.argv[2], sys.argv[2]) 31 | os.system(cmd) 32 | else: 33 | print 'choice play file name' 34 | except: 35 | print 'choice play file name' 36 | else: 37 | raise error 38 | except: 39 | usage() 40 | #pass 41 | 42 | 43 | -------------------------------------------------------------------------------- /logs/.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/svenlu/conn/e6f727db4c7db555fbd890086cb8f317c7859ac7/logs/.test -------------------------------------------------------------------------------- /record.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | import os, sys, time 3 | 4 | class Record(object): 5 | def __init__(self, hostname=None, user='root', password=None, port=22): 6 | self.hostname = hostname 7 | self.password = password 8 | self.user = user 9 | self.port = port 10 | 11 | def logs(self, *args): 12 | current_time = time.strftime('%Y%m%d-%H:%M:%S') 13 | logs_dir = '.logs' 14 | logs_file = '%s/%s-%s-%s' % (logs_dir, current_time, 'web01', 'root') 15 | if not os.path.isdir(logs_dir): os.mkdir(logs_dir) 16 | with open(logs_file) as f: 17 | f.write(11111) 18 | f.flush() 19 | #if os.path.isfile('%s/%s' % logs_dir, logs_file): 20 | 21 | 22 | def videos(self): 23 | current_time = time.strftime('%Y%m%d%H%M%S') 24 | file_name = '%s-%s-%s' % (current_time, self.hostname, self.user) 25 | cmd = './script -t 2> %s.time -a %s.his -c "python connection.py %s@%s:%d"' % (file_name, file_name, self.user, self.hostname, self.port) 26 | print cmd 27 | #os.system(cmd) 28 | 29 | def record_videos(func): 30 | def info(username='root', hostname=None, port=22): 31 | #print hostname, username, port, '-----------' 32 | current_time = time.strftime('%Y%m%d%H%M%S') 33 | file_name = '%s-%s-%s' % (current_time, hostname, username) 34 | cmd = './script -t 2> videos/%s.time -a videos/%s.his -c "python connection.py %s@%s:%d"' % (file_name, file_name, username, hostname, port) 35 | #result = func(cmd) 36 | #print cmd 37 | os.system(cmd) 38 | return info 39 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | paramiko==1.14.0 2 | ecdsa==0.11 3 | -------------------------------------------------------------------------------- /script: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/svenlu/conn/e6f727db4c7db555fbd890086cb8f317c7859ac7/script -------------------------------------------------------------------------------- /scriptreplay: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/svenlu/conn/e6f727db4c7db555fbd890086cb8f317c7859ac7/scriptreplay -------------------------------------------------------------------------------- /videos/.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/svenlu/conn/e6f727db4c7db555fbd890086cb8f317c7859ac7/videos/.test --------------------------------------------------------------------------------