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
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
"""
Description:
This file is used to make connections
Include ssh & exchange public-key to each other so that
it can run without password
lanqinglong@huawei.com
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
#
"""
import os
import pexpect
import re
from foundation import Foundation
import functest.utils.functest_logger as ft_logger
class Connection(Foundation):
logger = ft_logger.Logger("connection").getLogger()
def __init__(self):
Foundation.__init__(self)
self.loginfo = Foundation()
def AddKnownHost(self, handle, ipaddr, username, password):
"""
Add an user to known host,so that onos can login in with onos $ipaddr.
parameters:
ipaddr: ip address
username: login user name
password: login password
"""
self.logger.info("Now Adding an user to known hosts " + ipaddr)
login = handle
login.sendline("ssh -l %s -p 8101 %s" % (username, ipaddr))
index = 0
while index != 2:
index = login.expect(['assword:', 'yes/no', pexpect.EOF,
pexpect.TIMEOUT])
if index == 0:
login.sendline(password)
login.sendline("logout")
index = login.expect(["closed", pexpect.EOF])
if index == 0:
self.loginfo.log("Add SSH Known Host Success!")
break
else:
self.loginfo.log("Add SSH Known Host Failed! "
"Please Check!")
break
login.prompt()
if index == 1:
login.sendline('yes')
def GetEnvValue(self, handle, envname):
"""
os.getenv only returns current user value
GetEnvValue returns a environment value of
current handle
eg: GetEnvValue(handle,'HOME')
"""
envhandle = handle
envhandle.sendline('echo $' + envname)
envhandle.prompt()
reg = envname + '\r\n(.*)\r'
envaluereg = re.compile(reg)
envalue = envaluereg.search(envhandle.before)
if envalue:
return envalue.groups()[0]
else:
return None
def Gensshkey(self, handle):
"""
Generate ssh keys, used for some server have no sshkey.
"""
self.logger.info("Now Generating SSH keys...")
# Here file name may be id_rsa or id_ecdsa or others
# So here will have a judgement
keysub = handle
filepath = self.GetEnvValue(keysub, 'HOME') + '/.ssh'
filelist = os.listdir(filepath)
for item in filelist:
if 'id' in item:
self.loginfo.log("SSH keys are exsit in ssh directory.")
return True
keysub.sendline("ssh-keygen -t rsa")
Result = 0
while Result != 2:
Result = keysub.expect(["Overwrite", "Enter", pexpect.EOF,
'PEXPECT]#', pexpect.TIMEOUT])
if Result == 0:
keysub.sendline("y")
if Result == 1 or Result == 2:
keysub.sendline("\n")
if Result == 3:
self.loginfo.log("Generate SSH key success.")
keysub.prompt()
break
if Result == 4:
self.loginfo.log("Generate SSH key failed.")
keysub.prompt()
break
def GetRootAuth(self, password):
"""
Get root user
parameters:
password: root login password
"""
self.logger.info("Now changing to user root")
login = pexpect.spawn("su - root")
index = 0
while index != 2:
index = login.expect(['assword:', "failure",
pexpect.EOF, pexpect.TIMEOUT])
if index == 0:
login.sendline(password)
if index == 1:
self.loginfo.log("Change user to root failed.")
login.interact()
def ReleaseRootAuth(self):
"""
Exit root user.
"""
self.logger.info("Now Release user root")
login = pexpect.spawn("exit")
index = login.expect(['logout', pexpect.EOF, pexpect.TIMEOUT])
if index == 0:
self.loginfo.log("Release root user success.")
if index == 1:
self.loginfo.log("Release root user failed.")
login.interact()
def AddEnvIntoBashrc(self, envalue):
"""
Add Env var into /etc/profile.
parameters:
envalue: environment value to add
"""
self.logger.info("Now Adding bash environment")
fileopen = open("/etc/profile", 'r')
findContext = 1
while findContext:
findContext = fileopen.readline()
result = findContext.find(envalue)
if result != -1:
break
fileopen.close
if result == -1:
envAdd = open("/etc/profile", 'a+')
envAdd.writelines("\n" + envalue)
envAdd.close()
self.loginfo.log("Add env to bashrc success!")
def OnosRootPathChange(self, onospath):
"""
Change ONOS root path in file:bash_profile
onospath: path of onos root
"""
self.logger.info("Now Changing ONOS Root Path")
filepath = onospath + 'onos/tools/dev/bash_profile'
line = open(filepath, 'r').readlines()
lenall = len(line) - 1
for i in range(lenall):
if "export ONOS_ROOT" in line[i]:
line[i] = 'export ONOS_ROOT=' + onospath + 'onos\n'
NewFile = open(filepath, 'w')
NewFile.writelines(line)
NewFile.close
self.logger.info("Done!")
def OnosConnectionSet(self):
"""
Intergrate for ONOS connection setup
"""
if self.masterusername == 'root':
filepath = '/root/'
else:
filepath = '/home/' + self.masterusername + '/'
filepath = os.path.join(filepath, "onos/tools/dev/bash_profile")
self.AddEnvIntoBashrc("source " + filepath + "\n")
self.AddEnvIntoBashrc("export OCT=" + self.OCT)
self.AddEnvIntoBashrc("export OC1=" + self.OC1)
self.AddEnvIntoBashrc("export OC2=" + self.OC2)
self.AddEnvIntoBashrc("export OC3=" + self.OC3)
self.AddEnvIntoBashrc("export OCN=" + self.OCN)
self.AddEnvIntoBashrc("export OCN2=" + self.OCN2)
self.AddEnvIntoBashrc("export localhost=" + self.localhost)
|
