#+
# Copyright 2010 iXsystems, Inc.
# All rights reserved
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted providing that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
#####################################################################
import os
import smtplib
import sqlite3
import subprocess
import traceback
import syslog
from email.mime.text import MIMEText
from email.Utils import formatdate
from datetime import datetime, timedelta
from freenasUI.system.models import Email
from account.models import bsdUsers
VERSION_FILE = '/etc/version'
_VERSION = None
def get_sw_version(strip_build_num=False):
"""Return the full version string, e.g. FreeNAS-8.1-r7794-amd64."""
global _VERSION
if _VERSION is None:
with open(VERSION_FILE) as fd:
_VERSION = fd.read().strip()
if strip_build_num:
return _VERSION.split(' ')[0]
return _VERSION
def get_sw_login_version():
"""Return a shortened version string, e.g. 8.0.1-RC1, 8.1, etc. """
return '-'.join(get_sw_version(strip_build_num=True).split('-')[1:-2])
def get_sw_name():
"""Return the software name, e.g. FreeNAS"""
return get_sw_version().split('-')[0]
def get_freenas_var_by_file(f, var):
assert f and var
pipe = os.popen('. "%s"; echo "${%s}"' % (f, var, ))
try:
val = pipe.readlines()[-1].rstrip()
finally:
pipe.close()
return val
def get_freenas_var(var, default = None):
val = get_freenas_var_by_file("/etc/rc.freenas", var)
if not val:
val = default
return val
def send_mail(subject=None, text=None, interval=timedelta(),
channel=get_sw_name().lower(), to=None, extra_headers=None):
if interval > timedelta():
channelfile = '/tmp/.msg.%s' % (channel)
last_update = datetime.now() - interval
try:
last_update = datetime.fromtimestamp(os.stat(channelfile).st_mtime)
except OSError:
pass
timediff = datetime.now() - last_update
if (timediff >= interval) or (timediff < timedelta()):
open(channelfile, 'w').close()
else:
return
error = False
errmsg = ''
em = Email.objects.all().order_by('-id')[0]
if not to:
to = [ bsdUsers.objects.get(bsdusr_username='root').bsdusr_email ]
msg = MIMEText(text, _charset='utf-8')
if subject:
msg['Subject'] = subject
msg['From'] = em.em_fromemail
msg['To'] = ', '.join(to)
msg['Date'] = formatdate()
if not extra_headers:
extra_headers = {}
for key, val in extra_headers.items():
if key in msg:
msg.replace_header(key, val)
else:
msg[key] = val
msg = msg.as_string()
try:
if not em.em_outgoingserver or not em.em_port:
# See NOTE below.
raise ValueError('you must provide an outgoing mailserver and mail '
'server port when sending mail')
if em.em_security == 'ssl':
server = smtplib.SMTP_SSL(em.em_outgoingserver, em.em_port,
timeout=10)
else:
server = smtplib.SMTP(em.em_outgoingserver, em.em_port,
timeout=10)
if em.em_security == 'tls':
server.starttls()
if em.em_smtp:
server.login(em.em_user.encode('utf-8'), em.em_pass.encode('utf-8'))
# NOTE: Don't do this.
#
# If smtplib.SMTP* tells you to run connect() first, it's because the
# mailserver it tried connecting to via the outgoing server argument was
# unreachable and it tried to connect to 'localhost' and barfed. This is
# because FreeNAS doesn't run a full MTA.
#else:
# server.connect()
server.sendmail(em.em_fromemail, to, msg)
server.quit()
except ValueError, ve:
# Don't spam syslog with these messages. They should only end up in the
# test-email pane.
errmsg = str(ve)
error = True
except Exception, e:
syslog.openlog(channel, syslog.LOG_CONS | syslog.LOG_PID,
facility=syslog.LOG_MAIL)
try:
for line in traceback.format_exc().splitlines():
syslog.syslog(syslog.LOG_ERR, line)
finally:
syslog.closelog()
errmsg = str(e)
error = True
return error, errmsg
def get_fstype(path):
assert path
if not os.access(path, os.F_OK):
return None
lines = subprocess.check_output(['/bin/df', '-T', path]).splitlines()
out = (lines[len(lines) - 1]).split()
return (out[1].upper())
def get_mounted_filesystems():
mounted = []
lines = subprocess.check_output(['/sbin/mount']).splitlines()
for line in lines:
parts = line.split()
mountinfo = {}
mountinfo['fs_spec'] = parts[0]
mountinfo['fs_file'] = parts[2]
end = min(parts[3].find(')'), parts[3].find(','))
mountinfo['fs_vfstype'] = parts[3][1:end]
mounted.append(mountinfo)
return mounted
def is_mounted(**kwargs):
ret = False
mounted = get_mounted_filesystems()
for mountpt in mounted:
if 'device' in kwargs:
if mountpt['fs_spec'] == kwargs['device']:
ret = True
break
elif 'path' in kwargs:
if mountpt['fs_file'] == kwargs['path']:
ret = True
break
return ret
def mount(dev, path, mntopts=None):
if mntopts:
opts = ['-o', mntopts]
else:
opts = []
try:
subprocess.check_call(['/sbin/mount', ] + opts + [dev, path, ])
except:
return False
else:
return True
def umount(path):
try:
subprocess.check_call(['/sbin/umount', path, ])
except:
return False
else:
return True
def service_enabled(name):
db = get_freenas_var("FREENAS_DATABASE", "/data/freenas-v1.db")
h = sqlite3.connect(db)
c = h.cursor()
enabled = False
sql = "select srv_enable from services_services " \
"where srv_service = '%s' order by -id limit 1" % name
c.execute(sql)
row = c.fetchone()
if row and row[0] != 0:
enabled = True
c.close()
h.close()
return enabled
