source: lmd/trunk/fuentes/lmd-server.install/usr/share/n4d/python-plugins/LmdBootManager.py @ 392

Last change on this file since 392 was 392, checked in by hectorgh, 4 years ago

adding project files

File size: 11.5 KB
Line 
1import json
2import os
3import crypt
4
5class LmdBootManager:
6       
7               
8        def __init__(self):
9                self.imagepath="/etc/ltsp/bootopts/"
10
11               
12                pass
13        #def __init__
14       
15        def getNetinstall(self):
16                '''
17                Reads file /etc/ltsp/bootopts/netinstall and returns true or false
18                '''
19                # 1. /opt/ltsp/name-chroot
20                # 2. /opt/ltsp/images/name-chroot.img
21                # 3. /var/lib/tftpboot/ltsp/name-chroot
22                # if 1,2 and 3 exist -> show
23                # if 1 but not exist 2 or/and 3 -> show with error
24                #
25               
26                json_data=open(self.imagepath+"netinstall.json")
27                data = json.load(json_data)
28                json_data.close()
29                if(data["netinstall_boot"].lower()=="true"):
30                        netinstall=True;
31                else:
32                        netinstall=False;
33               
34                if(data["netinstall_unattended"].lower()=="true"):
35                        unattended=True;
36                else:
37                        unattended=False;
38                       
39                return {"netinstall":netinstall, "unattended":unattended}
40               
41                return json.dumps(imagelist)
42                       
43                       
44        # END def GetListImages(self)
45
46        def setNetinstall(self, status, unattended):
47                '''
48                sets option for netinstall int bootopt.json (status and unattended install)
49                '''
50                try:
51                        mirror_var="/var/lib/n4d/variables-dir/LLIUREXMIRROR"
52                        if os.path.isfile(mirror_var):
53                                if (status.lower()=="true" or status.lower()=="false"):
54                                        path_to_write = os.path.join(self.imagepath,"netinstall.json")
55                                        f = open(path_to_write,'w')
56                                        data='{"netinstall_boot":"'+str(status)+'", "netinstall_unattended":"'+str(unattended)+'"}'
57                                        f.writelines(data)
58                                        f.close()
59                                       
60                                        # Enable NETINSTALL on menu (the last option, but enabled)
61                                        objects["LlxBootManager"].pushToBootList("netinstall")
62
63                                        # Removing user and password from preseed
64                                        self.setNetinstallUnattended(status, "", "", "")
65                                       
66                                        return {"status":"true", "msg":"all ok"}
67       
68                                else:
69                                        return {"status":"false", "msg":"option not valid"}
70                        else:
71                                        return {"status":"false", "msg":"mirror is not available"}
72                               
73                        #return data;
74                except Exception as e:
75                        return str(e);
76
77                # END def getListTemplate(self, image)
78               
79        def setNetinstallUnattended(self, status, username, password, rootpassword):
80                '''
81                Writing in presseed username and password
82                '''                             
83                try:
84                        filedir="/var/www/preseed"
85                        filepath="/var/www/preseed/unattended.cfg"
86                        filepartman="/var/www/preseed/partman_sda.cfg"
87
88                        if not os.path.exists(filedir):
89                                os.makedirs(filedir)
90                       
91                        preseed=open(filepath,'w')
92                        preseed.write("# LMD Created user account\n")
93                        salt="sw9.tfRI"                 
94                        userpassencrypted=crypt.crypt(str(password),"$1$"+salt+"$")
95                        rootpassencrypted=crypt.crypt(str(rootpassword),"$1$"+salt+"$")
96                       
97                        if(status==True):
98                                # Saving file
99                               
100                                userfullline="d-i passwd/user-fullname string "+str(username)+"\n";
101                                userline="d-i passwd/username string "+str(username)+"\n";
102                                passline="d-i passwd/user-password-crypted password "+str(userpassencrypted)+"\n"
103                               
104                                if (len(rootpassword) > 0):
105                                        rootpassline = "d-i passwd/root-password-crypted password "+str(rootpassencrypted) + "\n"
106                                else:
107                                        rootpassline = "# d-i passwd/root-password-crypted password "+str(rootpassencrypted) + "\n"
108                                       
109                                # Partition preseed
110                                try:
111                                        partman = open(filepartman,'r')
112                                        preseed.writelines(partman.readlines())
113                                        preseed.write("\n")
114                                        partman.close()
115                                except Exception as e:
116                                        return str(e)
117                               
118                        else:
119                                userfullline="#d-i passwd/user-fullname string \n"
120                                userline="#d-i passwd/username string \n"
121                                passline="# d-i passwd/user-password-crypted password \n"
122                                rootpassline = "# d-i passwd/root-password-crypted password \n"
123
124                        preseed.write("# Normal user name\n")
125                        preseed.write(userfullline)
126                        preseed.write(userline)
127                        preseed.write("# Normal user's password, either in clear text\n")
128                        preseed.write("#d-i passwd/user-password password insecure\n")
129                        preseed.write("# Normal user's password encrypted using an MD5 hash.\n")
130                        preseed.write(passline)
131                        preseed.write(rootpassline)
132                       
133                        # Allow weak passwords
134                        preseed.write("d-i user-setup/allow-password-weak boolean true\n")
135                       
136                        preseed.close()
137                       
138                except Exception as e:
139                        return str(e)
140
141                # END def getListTemplate(self, image)
142               
143               
144               
145        '''
146        Methods to work with lts.conf file
147        '''
148       
149        ''' def getTemplate(self, template):
150                '' '
151                Reads the file template from /etc/ltsp/templates
152                Returna a JSON string with the config options
153                '' '
154                try:
155                        config = StringIO.StringIO()
156                        config.write('[meta_inf]\n')
157                        config.write('name="'+template+'"\n')
158                       
159                       
160                        config.write('[default]\n')
161                        config.write(open(str(self.templatepath)+"/"+str(template)).read())
162                        config.seek(0, os.SEEK_SET)
163                        cp = ConfigParser.ConfigParser()
164                        cp.readfp(config)
165                        aux = cp._sections
166                        for x in aux.keys():
167                                for y in aux[x]:
168                                        if aux[x][y][0] == '"' and aux[x][y][-1] == '"':
169                                                aux[x][y] = aux[x][y][1:-1]
170
171                        return json.dumps(aux)
172                        #return json.dumps(cp.items('default'));
173                       
174                       
175                except Exception as e:
176                        return "Exception: "+str(e)
177                # END def getListTemplate(self, template)
178        '''             
179               
180        def getLtsConf(self):
181                # Reads $tftpboot/ltsp/i386/lts.conf and returns it in json format
182               
183                try:
184                        conf_file="/var/lib/tftpboot/ltsp/i386/lts.conf"
185                        cp = ConfigParser.ConfigParser()
186                        cp.optionxform=str
187                        cp.read(conf_file)
188                        aux = cp._sections
189                        print "**",aux,"$$$"
190                        for x in aux.keys():
191                                for y in aux[x]:
192                                        if aux[x][y][0] == '"' and aux[x][y][-1] == '"':
193                                                aux[x][y] = aux[x][y][1:-1]
194
195                        print "@@",aux,"##"
196                        return json.dumps(aux)
197                        pass
198                        #return json.dumps(cp.items('default'));
199                       
200                except Exception as e:
201                        return "Exception: "+str(e)
202                # END def getListTemplate(self, template)
203                               
204                pass
205       
206        def mergeLtsConf(self):
207               
208                # Creates lts.conf file merging lts.default.conf and lts.clients.conf
209                conf_file="/var/lib/tftpboot/ltsp/i386/lts.conf"
210                default_file="/var/lib/tftpboot/ltsp/i386/lts.default.conf"
211                clients_file="/var/lib/tftpboot/ltsp/i386/lts.clients.conf"
212                filenames = [default_file, clients_file]
213               
214                with open(conf_file, 'w') as outfile:
215                    for fname in filenames:
216                        with open(fname) as infile:
217                            for line in infile:
218                                outfile.write(line)             
219                return true;
220       
221        def setLtsClientsConf(self):
222                # Writes json config in $tftpboot/ltsp/i386/lts.conf
223                try:
224                        conf_file="/var/lib/tftpboot/ltsp/i386/lts.clients.conf"
225                        clientpath="/etc/ltsp/bootopts/clients/"
226                        cfgfile = open(conf_file,'w')
227                                               
228                        # Autologin by mac...
229                        try:
230                                for i in os.listdir(clientpath):
231                                        if '.json' in i:
232                                                json_data=open(os.path.join(clientpath,i))
233                                                data = json.load(json_data)
234                                                json_data.close()
235                                                cfgfile.write("["+data["mac"]+"]\n");
236                                                cfgfile.write("LDM_AUTOLOGIN = True\n");
237                                                cfgfile.write("LDM_USERNAME = "+data["user"]+"\n");
238                                                cfgfile.write("LDM_PASSWORD = "+data["pass"]+"\n\n");
239                        except Exception as e:
240                                print "[LMDBootManager] Exception "+str(e)
241                       
242                        cfgfile.close()
243                       
244                        # Finally merge both conf files:
245                        self.mergeLtsConf();
246                except Exception as e:
247                        return "Exception: "+str(e)
248               
249               
250                pass
251
252
253        def setLtsConf(self, config):
254                # Writes json config in $tftpboot/ltsp/i386/lts.conf
255                try:
256                        conf_file="/var/lib/tftpboot/ltsp/i386/lts.default.conf"
257                        cfgfile = open(conf_file,'w')
258                       
259                        #cp = ConfigParser.ConfigParser()
260                        conf=json.loads(config)                                                 
261                       
262                        for (section) in conf:
263                                cfgfile.write("["+section+"]\n");
264                                for i in conf[section]:
265                                        if i.lower()=="__name__":
266                                                pass
267                                        else:
268                                                if (conf[section][i]).find(' ')!=-1: # ound spaces, adding "
269                                                        cfgfile.write("{0}=\"{1}\"\n".format(i, (conf[section][i])))
270                                                else:
271                                                        cfgfile.write("{0}={1}\n".format(i, (conf[section][i])))
272                        cfgfile.close()
273                       
274                        # Finally merge both conf files:
275                        self.mergeLtsConf();
276                       
277                except Exception as e:
278                        return "Exception: "+str(e)
279       
280
281        def getBootOpts(self):
282                '''
283                Reads file /etc/ltsp/bootopts/netinstall and returns true or false
284                '''
285                # 1. /opt/ltsp/name-chroot
286                # 2. /opt/ltsp/images/name-chroot.img
287                # 3. /var/lib/tftpboot/ltsp/name-chroot
288                # if 1,2 and 3 exist -> show
289                # if 1 but not exist 2 or/and 3 -> show with error
290                #
291                json_data=open(self.imagepath+"netinstall.json")
292                data = json.load(json_data)
293                json_data.close()
294                if(data["netinstall_boot"].lower()=="true"):
295                        return True
296                else:
297                        return False
298                       
299                return json.dumps(imagelist)
300                       
301                       
302        # END def GetListImages(self)
303
304        def setBootOpts(self, status):
305                '''
306                sets option for netinstall int bootopt.json
307                '''
308                try:
309                        mirror_var="/var/lib/n4d/variables-dir/LLIUREXMIRROR"
310                        if os.path.isfile(mirror_var):
311                                if (status.lower()=="true" or status.lower()=="false"):
312                                        path_to_write = os.path.join(self.imagepath,"netinstall.json")
313                                        f = open(path_to_write,'w')
314                                        data='{"netinstall_boot":"'+str(status)+'"}'
315                                        f.writelines(data)
316                                        f.close()
317                                        return {"status":"true", "msg":"all ok"}
318       
319                                else:
320                                        return {"status":"false", "msg":"option not valid"}
321                        else:
322                                        return {"status":"false", "msg":"mirror is not available"}
323                               
324                        #return data;
325                except Exception as e:
326                        return str(e);
327
328        def ReadNBDSwapSize(self):
329                nbd_swap_file="/etc/ltsp/nbdswapd.conf"
330                f1 = open(nbd_swap_file,'r')
331                data=f1.readlines()
332                size=data[0][5:];
333                f1.close()
334                return size
335               
336
337        def SaveNBDSwapSize(self, swap_size):
338                '''
339                Saves properly options to set swap size
340                '''
341                try:
342                        # Saving nbdswapd.conf
343                        nbd_swap_file="/etc/ltsp/nbdswapd.conf"
344                        f1 = open(nbd_swap_file,'w')
345                        data='SIZE='+str(swap_size)+'\n';
346                        f1.writelines(data)
347                        f1.close()
348
349                        # Saving swap.conf
350                        swap_conf_file="/etc/nbd-server/conf.d/swap.conf"
351                        f2 = open(swap_conf_file,'w')
352                        f2.writelines("[swap]\n")
353                        f2.writelines("exportname = /tmp/nbd-swap/%s\n")
354                        f2.writelines("prerun = nbdswapd %s\n")
355                        f2.writelines("postrun = rm -f %s\n")
356                        f2.writelines("authfile = /etc/ltsp/nbd-server.allow\n")
357                        f2.close()
358
359                        # Checking /etc/hosts.allow
360                        needs_nbdswap_line=True
361                       
362                        if os.path.isfile("/etc/hosts.allow"):
363                                # if file exists, read it
364                                f3=open("/etc/hosts.allow", 'r')                               
365                                for line in f3.readlines():
366                                        print line
367                                        if (line=="nbdswapd: ALL: keepalive\n"):
368                                                needs_nbdswap_line=False
369                                f3.close();
370                       
371                        # If file does not exists or has no line for nbdwap, write it
372                        if (needs_nbdswap_line):
373                                f4=open("/etc/hosts.allow", 'a')
374                                f4.writelines("nbdswapd: ALL: keepalive\n")
375                                f4.close()
376                       
377                        return {"status":True, "msg":"finished"}
378               
379                except Exception as e:
380                        print "exc,"+str(e)
381                        return {"status":False, "msg":str(e)}
382               
383        def getDefaultBootImage(self):
384                try:
385                        if os.path.exists(self.imagepath+"defaultboot.json"):
386                                json_data=open(self.imagepath+"defaultboot.json")
387                                data = json.load(json_data)
388                                json_data.close()
389                                return data
390                        else:
391                                return {"default_boot":""}
392                except Exception as e:
393                        return {"default_boot":""}
394               
395        def setDefaultBootImage(self, image):
396                try:
397                        path_to_write = os.path.join(self.imagepath,"defaultboot.json")
398                        f = open(path_to_write,'w')
399                        data='{"default_boot":"'+str(image)+'"}'
400                        f.writelines(data)
401                        f.close()
402                        return {"status":"true", "msg":"all ok"}
403                except Exception as e:
404                        return {"status":False, "msg":str(e)}
405
406
407        def getMenuTimeout(self):
408                try:
409                        if os.path.exists(self.imagepath+"menutimeout.json"):
410                                json_data=open(self.imagepath+"menutimeout.json")
411                                data = json.load(json_data)
412                                json_data.close()
413                                return data
414                        else:
415                                return {"timeout":""}
416                except Exception as e:
417                        return {"timeout":""}
418               
419        def setMenuTimeout(self, time):
420                try:
421                        path_to_write = os.path.join(self.imagepath,"menutimeout.json")
422                        f = open(path_to_write,'w')
423                        data='{"timeout":"'+str(time)+'"}'
424                        f.writelines(data)
425                        f.close()
426                        return {"status":"true", "msg":"all ok"}
427                except Exception as e:
428                        return {"status":False, "msg":str(e)}
Note: See TracBrowser for help on using the repository browser.