#!/usr/bin/env python3 
 | 
  
 | 
# Sends an error report (if the report-error class was enabled) to a 
 | 
# remote server. 
 | 
# 
 | 
# Copyright (C) 2013 Intel Corporation 
 | 
# Author: Andreea Proca <andreea.b.proca@intel.com> 
 | 
# Author: Michael Wood <michael.g.wood@intel.com> 
 | 
# 
 | 
# SPDX-License-Identifier: GPL-2.0-only 
 | 
# 
 | 
  
 | 
import urllib.request, urllib.error 
 | 
import sys 
 | 
import json 
 | 
import os 
 | 
import subprocess 
 | 
import argparse 
 | 
import logging 
 | 
  
 | 
scripts_lib_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'lib') 
 | 
sys.path.insert(0, scripts_lib_path) 
 | 
import argparse_oe 
 | 
  
 | 
version = "0.3" 
 | 
  
 | 
log = logging.getLogger("send-error-report") 
 | 
logging.basicConfig(format='%(levelname)s: %(message)s') 
 | 
  
 | 
def getPayloadLimit(url): 
 | 
    req = urllib.request.Request(url, None) 
 | 
    try: 
 | 
        response = urllib.request.urlopen(req) 
 | 
    except urllib.error.URLError as e: 
 | 
        # Use this opportunity to bail out if we can't even contact the server 
 | 
        log.error("Could not contact server: " + url) 
 | 
        log.error(e.reason) 
 | 
        sys.exit(1) 
 | 
    try: 
 | 
        ret = json.loads(response.read()) 
 | 
        max_log_size = ret.get('max_log_size', 0) 
 | 
        return int(max_log_size) 
 | 
    except: 
 | 
        pass 
 | 
  
 | 
    return 0 
 | 
  
 | 
def ask_for_contactdetails(): 
 | 
    print("Please enter your name and your email (optionally), they'll be saved in the file you send.") 
 | 
    username = input("Name (required): ") 
 | 
    email = input("E-mail (not required): ") 
 | 
    return username, email 
 | 
  
 | 
def edit_content(json_file_path): 
 | 
    edit = input("Review information before sending? (y/n): ") 
 | 
    if 'y' in edit or 'Y' in edit: 
 | 
        editor = os.environ.get('EDITOR', None) 
 | 
        if editor: 
 | 
            subprocess.check_call([editor, json_file_path]) 
 | 
        else: 
 | 
            log.error("Please set your EDITOR value") 
 | 
            sys.exit(1) 
 | 
        return True 
 | 
    return False 
 | 
  
 | 
def prepare_data(args): 
 | 
    # attempt to get the max_log_size from the server's settings 
 | 
    max_log_size = getPayloadLimit(args.protocol+args.server+"/ClientPost/JSON") 
 | 
  
 | 
    if not os.path.isfile(args.error_file): 
 | 
        log.error("No data file found.") 
 | 
        sys.exit(1) 
 | 
  
 | 
    home = os.path.expanduser("~") 
 | 
    userfile = os.path.join(home, ".oe-send-error") 
 | 
  
 | 
    try: 
 | 
        with open(userfile, 'r') as userfile_fp: 
 | 
            if len(args.name) == 0: 
 | 
                args.name = userfile_fp.readline() 
 | 
            else: 
 | 
                #use empty readline to increment the fp 
 | 
                userfile_fp.readline() 
 | 
  
 | 
            if len(args.email) == 0: 
 | 
                args.email = userfile_fp.readline() 
 | 
    except: 
 | 
        pass 
 | 
  
 | 
    if args.assume_yes == True and len(args.name) == 0: 
 | 
        log.error("Name needs to be provided either via "+userfile+" or as an argument (-n).") 
 | 
        sys.exit(1) 
 | 
  
 | 
    while len(args.name) <= 0 or len(args.name) > 50: 
 | 
        print("\nName needs to be given and must not more than 50 characters.") 
 | 
        args.name, args.email = ask_for_contactdetails() 
 | 
  
 | 
    with open(userfile, 'w') as userfile_fp: 
 | 
        userfile_fp.write(args.name.strip() + "\n") 
 | 
        userfile_fp.write(args.email.strip() + "\n") 
 | 
  
 | 
    with open(args.error_file, 'r') as json_fp: 
 | 
        data = json_fp.read() 
 | 
  
 | 
        jsondata = json.loads(data) 
 | 
        jsondata['username'] = args.name.strip() 
 | 
        jsondata['email'] = args.email.strip() 
 | 
        jsondata['link_back'] = args.link_back.strip() 
 | 
        # If we got a max_log_size then use this to truncate to get the last 
 | 
        # max_log_size bytes from the end 
 | 
        if max_log_size != 0: 
 | 
            for fail in jsondata['failures']: 
 | 
                if len(fail['log']) > max_log_size: 
 | 
                    print("Truncating log to allow for upload") 
 | 
                    fail['log'] = fail['log'][-max_log_size:] 
 | 
  
 | 
        data = json.dumps(jsondata, indent=4, sort_keys=True) 
 | 
  
 | 
    # Write back the result which will contain all fields filled in and 
 | 
    # any post processing done on the log data 
 | 
    with open(args.error_file, "w") as json_fp: 
 | 
        if data: 
 | 
            json_fp.write(data) 
 | 
  
 | 
  
 | 
    if args.assume_yes == False and edit_content(args.error_file): 
 | 
        #We'll need to re-read the content if we edited it 
 | 
        with open(args.error_file, 'r') as json_fp: 
 | 
            data = json_fp.read() 
 | 
  
 | 
    return data.encode('utf-8') 
 | 
  
 | 
  
 | 
def send_data(data, args): 
 | 
    headers={'Content-type': 'application/json', 'User-Agent': "send-error-report/"+version} 
 | 
  
 | 
    if args.json: 
 | 
        url = args.protocol+args.server+"/ClientPost/JSON/" 
 | 
    else: 
 | 
        url = args.protocol+args.server+"/ClientPost/" 
 | 
  
 | 
    req = urllib.request.Request(url, data=data, headers=headers) 
 | 
    try: 
 | 
        response = urllib.request.urlopen(req) 
 | 
    except urllib.error.HTTPError as e: 
 | 
        logging.error(str(e)) 
 | 
        sys.exit(1) 
 | 
  
 | 
    print(response.read().decode('utf-8')) 
 | 
  
 | 
  
 | 
if __name__ == '__main__': 
 | 
    arg_parse = argparse_oe.ArgumentParser(description="This scripts will send an error report to your specified error-report-web server.") 
 | 
  
 | 
    arg_parse.add_argument("error_file", 
 | 
                           help="Generated error report file location", 
 | 
                           type=str) 
 | 
  
 | 
    arg_parse.add_argument("-y", 
 | 
                           "--assume-yes", 
 | 
                           help="Assume yes to all queries and do not prompt", 
 | 
                           action="store_true") 
 | 
  
 | 
    arg_parse.add_argument("-s", 
 | 
                           "--server", 
 | 
                           help="Server to send error report to", 
 | 
                           type=str, 
 | 
                           default="errors.yoctoproject.org") 
 | 
  
 | 
    arg_parse.add_argument("-e", 
 | 
                           "--email", 
 | 
                           help="Email address to be used for contact", 
 | 
                           type=str, 
 | 
                           default="") 
 | 
  
 | 
    arg_parse.add_argument("-n", 
 | 
                           "--name", 
 | 
                           help="Submitter name used to identify your error report", 
 | 
                           type=str, 
 | 
                           default="") 
 | 
  
 | 
    arg_parse.add_argument("-l", 
 | 
                           "--link-back", 
 | 
                           help="A url to link back to this build from the error report server", 
 | 
                           type=str, 
 | 
                           default="") 
 | 
  
 | 
    arg_parse.add_argument("-j", 
 | 
                           "--json", 
 | 
                           help="Return the result in json format, silences all other output", 
 | 
                           action="store_true") 
 | 
  
 | 
    arg_parse.add_argument("--no-ssl", 
 | 
                           help="Use http instead of https protocol", 
 | 
                           dest="protocol", 
 | 
                           action="store_const", const="http://", default="https://") 
 | 
  
 | 
  
 | 
  
 | 
    args = arg_parse.parse_args() 
 | 
  
 | 
    if (args.json == False): 
 | 
        print("Preparing to send errors to: "+args.server) 
 | 
  
 | 
    data = prepare_data(args) 
 | 
    send_data(data, args) 
 | 
  
 | 
    sys.exit(0) 
 |