forked from iDPL/placement
/
SCPMover.py
129 lines (112 loc) · 3.46 KB
/
SCPMover.py
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
import os
import time
import socket
import sys
import signal
import hashlib
import pwd
import tempfile
import TimedExec
from IDPLException import *
from DataMover import *
class SCPMover(DataMover):
def __init__(self, workDir=None):
super(SCPMover,self).__init__()
self.SCPexe = '/usr/bin/scp'
self.SSHDexe = '/usr/sbin/sshd'
self.KEYGENexe = '/usr/bin/ssh-keygen'
self.setExe(self.SSHDexe)
self.setOutputHandler(self.sshout)
self.setErrHandler(self.ssherr)
self.rawData = None
self.transferredKB=0
self.serverDir = workDir
self.user = None
self.hostkey = None
self.userkey = None
self.userkeypub = None
self.inputFile = None # source file on client
self.outputFile = None # dest file on server
def sshout(self,pid,str):
""" stdout handler when running sshd under TimedExec """
message = "%s(%d): %s" % (socket.getfqdn(),pid,str)
sys.stdout.write(message)
host = socket.getfqdn()
def ssherr(self,pid,str):
""" stderr handler when running sshd under TimedExec """
sys.stderr.write("%d#: %s" %(pid,str))
if str.find("bind failed") != -1:
raise PortInUseException("sshd", self.port)
def setInputfile(self,fname):
self.inputFile = fname
def setOutputfile(self,fname):
self.outputFile = fname
def setAuthorizedKey(self,key):
"""Drop the authorized key into a secure temporary file"""
if self.userkeypub is not None:
os.unlink(self.userkeypub)
fh,self.userkeypub = tempfile.mkstemp(dir=os.getcwd())
os.write(fh,key)
os.close(fh)
def getUser(self):
return self.user
def setUser(self,nuser):
self.user = nuser
def getUserPubKeyFile(self):
return self.userkeypub
def clientSetup(self):
# Set up a user key
self.userkey,self.userkeypub = self.genkey("dsa")
def client(self,server,port=5001):
self.setExe(self.SCPexe)
if self.user is None:
self.user = pwd.getpwuid(os.geteuid()).pw_name
args = ["-o","StrictHostKeyChecking=no"]
args.extend(["-i",self.userkey])
args.extend(["-P","%d" % int(port)])
args.extend([self.inputFile, "%s@%s:%s" %
(self.user, server,self.outputFile)])
self.setArgs(args)
print "client: " , args
self.run()
def server(self):
args = ["-o","AuthorizedKeysFile=%s" % self.userkeypub]
args.extend (["-o","StrictModes=no"])
args.extend (["-o","UsePam=no"])
args.extend (["-o","PermitRootLogin=no"])
args.extend (["-o","PasswordAuthentication=no"])
args.extend (["-o","PidFile=/dev/null"])
args.extend(["-h",self.hostkey,"-D", "-e"])
args.extend(["-f","/dev/null"])
# put the server in debug mode to accept only one incoming connection
args.extend(["-d"])
self.setArgs(args)
self.setPortRange(5001,5010)
print "server: " , args
self.run()
def serverSetup(self):
"""This is to setup the local ssh server"""
if self.serverDir is None:
self.serverDir = os.getcwd()
if self.user is None:
self.user = pwd.getpwuid(os.geteuid()).pw_name
# Set up the host key
self.hostkey,self.hostkeypub = self.genkey("rsa")
def genkey(self,bname,type="rsa"):
"""generates host/user key in a temporary file"""
cwd = os.getcwd()
fh,key= tempfile.mkstemp(dir=cwd,text=True)
os.close(fh)
keypub = "%s.pub" % key
if os.path.isfile(key):
os.unlink(key)
if os.path.isfile(keypub):
os.unlink(keypub)
keygencmd = [self.KEYGENexe,"-q","-f",key,"-t", type, "-N", ""]
opcode,out,err = TimedExec.runTimedCmd(5,keygencmd)
if opcode != 0:
print out
print err
raise SSHServerException("ssh-keygen",err)
return (key,keypub)
# vim: ts=4:sw=4: