src/sendprofile

   1 #!/usr/bin/python3
   2 ###############################################################################
   3 #                                                                             #
   4 # Fireinfo                                                                    #
   5 # Copyright (C) 2010, 2011 IPFire Team (www.ipfire.org)                       #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) any later version.                                         #
  11 #                                                                             #
  12 # This program is distributed in the hope that it will be useful,             #
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
  15 # GNU General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 import json
  23 import logging
  24 import logging.handlers
  25 import os
  26 import sys
  27 import urllib.request, urllib.parse, urllib.error
  28
  29 import fireinfo
  30
  31 ENABLED_FILE = "/var/ipfire/main/send_profile"
  32 PROXY_SETTINGS = "/var/ipfire/proxy/advanced/settings"
  33
  34 log_level = logging.INFO
  35 if "-d" in sys.argv:
  36         log_level = logging.DEBUG
  37
  38 # Setup logging
  39 log = logging.getLogger()
  40 log.setLevel(log_level)
  41 log.addHandler(logging.handlers.SysLogHandler("/dev/log"))
  42 log.addHandler(logging.StreamHandler(sys.stderr))
  43 for handler in log.handlers:
  44         handler.setLevel(log_level)
  45
  46 PROFILE_URL = "https://fireinfo.ipfire.org/send/%(public_id)s"
  47
  48 def get_upstream_proxy():
  49         if not os.path.exists(PROXY_SETTINGS):
  50                 return {"host" : ""}
  51
  52         proxy_settings = {}
  53         with open(PROXY_SETTINGS) as f:
  54                 for line in f.readlines():
  55                         k, v = line.split("=", 1)
  56                         proxy_settings[k] = v.strip()
  57
  58         return {
  59                 "host" : proxy_settings.get("UPSTREAM_PROXY", ""),
  60                 "user" : proxy_settings.get("UPSTREAM_USER", ""),
  61                 "pass" : proxy_settings.get("UPSTREAM_PASSWORD", ""),
  62         }
  63
  64 def send_profile(profile):
  65         logging.debug("Sending profile:")
  66         for line in json.dumps(profile, sort_keys=True, indent=4).splitlines():
  67                 logging.debug(line)
  68
  69         # Compile payload
  70         data = urllib.parse.urlencode({"profile" : json.dumps(profile)})
  71
  72         request = urllib.request.Request(PROFILE_URL % profile, data=data.encode())
  73         request.add_header("User-Agent", "fireinfo/%s" % fireinfo.__version__)
  74
  75         # Set upstream proxy if we have one.
  76         proxy = get_upstream_proxy()
  77
  78         if proxy["host"]:
  79                 # handling upstream proxies with authentication is more tricky...
  80                 if proxy["user"] and proxy["pass"]:
  81                         prx_auth_string = "http://%s:%s@%s/" % (proxy["user"], proxy["pass"], proxy["host"])
  82
  83                         proxy_handler = urllib.request.ProxyHandler({'http': prx_auth_string, 'https': prx_auth_string})
  84                         auth = urllib.request.HTTPBasicAuthHandler()
  85                         opener = urllib.request.build_opener(proxy_handler, auth, urllib.request.HTTPHandler)
  86                         urllib.request.install_opener(opener)
  87                 else:
  88                         request.set_proxy(proxy["host"], "http")
  89                         request.set_proxy(proxy["host"], "https")
  90
  91         try:
  92                 urllib.request.urlopen(request, timeout=60)
  93         except (urllib.error.HTTPError, urllib.error.URLError) as e:
  94                 reason = "Unknown reason"
  95
  96                 if isinstance(e, urllib.error.HTTPError):
  97                         reason = "%s" % e
  98                 elif isinstance(e, urllib.error.URLError):
  99                         reason = e.reason
 100
 101                 logging.error("Profile was not sent propertly: %s" % reason)
 102                 return
 103
 104         logging.debug("Profile was sent successfully.")
 105
 106 def main():
 107         # Collect system information
 108         system = fireinfo.System()
 109         profile = system.profile()
 110
 111         # If --dump is passed -> only dump the output.
 112         if "--dump" in sys.argv:
 113                 # Remove the private id when dumping the profile because
 114                 # it contains no information and may confuse people.
 115                 del profile["private_id"]
 116
 117                 print(json.dumps(profile, sort_keys=True, indent=4))
 118                 return 0
 119
 120         if "--secret-id" in sys.argv:
 121                 print(system.secret_id)
 122                 return 0
 123
 124         if "--hardware-string" in sys.argv:
 125                 print(system._unique_id)
 126                 return 0
 127
 128         if "--public-id" in sys.argv:
 129                 print(system.public_id)
 130                 return 0
 131
 132         if not os.path.exists(ENABLED_FILE):
 133                 del profile["profile"]
 134
 135         try:
 136                 send_profile(profile)
 137         except urllib.error.URLError:
 138                 return 1
 139
 140         return 0
 141
 142 sys.exit(main())