475 lines
16 KiB
Python
Executable File
475 lines
16 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Author: Stéphane Graber <stgraber@ubuntu.com>
|
|
# Written by Stéphane Graber <stgraber@stgraber.org>
|
|
# Daniel Bartlett <dan@f-box.org>
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
from __future__ import print_function
|
|
|
|
import sys
|
|
from configparser import ConfigParser, NoOptionError
|
|
from urllib.parse import urlencode
|
|
from urllib import request
|
|
|
|
# Set the default pastebin
|
|
defaultPB = "dpaste.com"
|
|
|
|
# Now try to override it with a distributor pastebin
|
|
try:
|
|
import distro
|
|
release = distro.id()
|
|
if release in ('debian', 'raspbian'):
|
|
defaultPB = "paste.debian.net"
|
|
elif release in ('fedora', 'centos', 'rhel', 'rocky'):
|
|
defaultPB = "paste.centos.org"
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
import getopt
|
|
import gettext
|
|
import os
|
|
import re
|
|
import socket
|
|
import xml.dom.minidom
|
|
|
|
try:
|
|
import json
|
|
except ImportError:
|
|
json = None
|
|
|
|
_ = gettext.gettext
|
|
gettext.textdomain("pastebinit")
|
|
|
|
# Timeout after 5s
|
|
socket.setdefaulttimeout(15)
|
|
|
|
# Version number to show in the usage
|
|
version = "1.6.2"
|
|
configfiles = []
|
|
|
|
for conffile in list(reversed(os.environ.get(
|
|
'XDG_CONFIG_DIRS', '').split(':'))) + ['/etc', '/usr/local/etc',
|
|
os.environ.get('XDG_CONFIG_HOME',
|
|
os.path.expanduser('~/.config')),
|
|
os.path.expanduser('~/.pastebinit.xml')]:
|
|
|
|
conffile = conffile.rstrip('/')
|
|
if not conffile.endswith('pastebinit.xml'):
|
|
conffile = os.path.join(conffile, 'pastebinit.xml')
|
|
if conffile not in configfiles:
|
|
configfiles.append(conffile)
|
|
|
|
class PasteRequest(request.Request):
|
|
version = "Pastebinit v%s" % version
|
|
|
|
def __init__(self, *args, **opts):
|
|
super(PasteRequest, self).__init__(*args, **opts)
|
|
if 'User-Agent' not in self.headers:
|
|
self.add_header('User-Agent', self.version)
|
|
|
|
def preloadPastebins():
|
|
# Check several places for config files:
|
|
# - global config in /etc/pastebin.d
|
|
# - for source checkout, config in the checkout
|
|
# - user's overrides in ~/.pastebin.d
|
|
# Files found later override files found earlier.
|
|
pastebind = {}
|
|
confdirs = []
|
|
for confdir in ['/usr/share', '/usr/local/share'] \
|
|
+ list(reversed(os.environ.get('XDG_DATA_DIRS', '').split(':'))) \
|
|
+ list(reversed(os.environ.get('XDG_CONFIG_DIRS', '').split(':'))) \
|
|
+ ['/etc', '/usr/local/etc',
|
|
os.environ.get('XDG_CONFIG_HOME',
|
|
os.path.expanduser('~/.config')),
|
|
os.path.expanduser('~/.pastebin.d')]:
|
|
|
|
confdir = confdir.rstrip('/')
|
|
if not confdir.endswith('pastebin.d'):
|
|
confdir = os.path.join(confdir, 'pastebin.d')
|
|
if confdir not in confdirs:
|
|
confdirs.append(confdir)
|
|
|
|
for confdir in confdirs:
|
|
try:
|
|
confdirlist = os.listdir(confdir)
|
|
except OSError:
|
|
continue
|
|
|
|
for fileitem in confdirlist:
|
|
if fileitem.startswith('.') or not fileitem.endswith('.conf'):
|
|
continue
|
|
|
|
filename = os.path.join(confdir, fileitem)
|
|
instance = ConfigParser()
|
|
try:
|
|
instance.read(filename)
|
|
except UnicodeError:
|
|
continue
|
|
|
|
if not instance.has_section('pastebin'):
|
|
print(_('%s: no section [pastebin]') % filename,
|
|
file=sys.stderr)
|
|
continue
|
|
|
|
if not instance.has_option('pastebin', 'basename'):
|
|
print(_("%s: no 'basename' in [pastebin]") % filename,
|
|
file=sys.stderr)
|
|
continue
|
|
|
|
pastebind[instance.get('pastebin', 'basename')] = instance
|
|
return pastebind
|
|
|
|
# Return the configuration depending of the pastebin used
|
|
def getConfig(website):
|
|
for config in list(pastebind.values()):
|
|
basename = config.get('pastebin', 'basename')
|
|
try:
|
|
https = config.get('pastebin', 'https')
|
|
except:
|
|
https = False
|
|
|
|
if basename == website:
|
|
if https:
|
|
website = "https://%s/" % basename
|
|
else:
|
|
website = "http://%s/" % basename
|
|
return website, config['pastebin']
|
|
|
|
print(_("Unknown website, please post a bugreport to request "
|
|
"this pastebin to be added (%s)") % website,
|
|
file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# Return the parameters depending of the pastebin used
|
|
def getParameters(website, content):
|
|
config = pastebind[website]
|
|
params = {}
|
|
for param in config.options('format'):
|
|
paramname = config.get('format', param)
|
|
if param == 'content':
|
|
params[paramname] = content
|
|
else:
|
|
params[paramname] = config.get('defaults', param,
|
|
vars=spec_opts, fallback=inline_opts.get(param))
|
|
return params
|
|
|
|
# XML Handling methods
|
|
def getText(nodelist):
|
|
rc = ""
|
|
for node in nodelist:
|
|
if node.nodeType == node.TEXT_NODE:
|
|
rc = rc + node.data
|
|
return rc
|
|
|
|
def getNodes(nodes, title):
|
|
return nodes.getElementsByTagName(title)
|
|
|
|
def getFirstNode(nodes, title):
|
|
return getNodes(nodes, title)[0]
|
|
|
|
def getFirstNodeText(nodes, title):
|
|
return getText(getFirstNode(nodes, title).childNodes)
|
|
|
|
# Display usage instructions
|
|
def Usage(fd=sys.stdout):
|
|
print("pastebinit v" + version, file=fd)
|
|
print(_("\nReads on stdin for input or takes a list of files "
|
|
"as parameters"), file=fd)
|
|
print(_("\nUsage:\tpastebinit [OPTION...] [FILE...]"), file=fd)
|
|
print(_("\nGeneral arguments:\n"), file=fd)
|
|
print(_("\t-b <pastebin> (default is '%s')") % website, file=fd)
|
|
print(_("\t-i <input file>"), file=fd)
|
|
print(_("\t-l List all supported pastebins"), file=fd)
|
|
print(_("\t-E Print the content to stdout too"), file=fd)
|
|
print(_("\t-h Print this help screen"), file=fd)
|
|
print(_("\t-v Print the version number"), file=fd)
|
|
print(_("\t-V Print verbose output to stderr"), file=fd)
|
|
print(_("\nOptional arguments (not supported by all pastebins):\n"),
|
|
file=fd)
|
|
print(_("\t-a <author> (default is '%s')") %
|
|
inline_opts['user'], file=fd)
|
|
print(_("\t-t <title of paste> (default is none)"), file=fd)
|
|
print(_("\t-f <format of paste> (default is '%s')") %
|
|
inline_opts['format'], file=fd)
|
|
print(_("\t-j <jabberid> (default is '%s')") %
|
|
inline_opts['jabberid'], file=fd)
|
|
print(_("\t-m <permatag> (default is none)"),
|
|
file=fd)
|
|
print(_("\t-P <private level> (default is '%s')") %
|
|
inline_opts['private'], file=fd)
|
|
print(_("\t-u <username> -p <password>"), file=fd)
|
|
|
|
# Set defaults
|
|
website = defaultPB
|
|
inline_opts = {
|
|
'user' : os.environ.get('USER', os.environ.get('LOGNAME')),
|
|
'jabberid' : "",
|
|
'title' : "",
|
|
'permatag' : "",
|
|
'format' : "text",
|
|
'private' : 1,
|
|
'username' : "",
|
|
'password' : ""
|
|
}
|
|
spec_opts = {}
|
|
filenames = []
|
|
verbose = False
|
|
echo = False
|
|
|
|
# Example configuration file string
|
|
configexample = """\
|
|
<pastebinit>
|
|
<pastebin>paste.debian.net</pastebin>
|
|
<author>A pastebinit user</author>
|
|
<jabberid>nobody@nowhere.org</jabberid>
|
|
<format>text</format>
|
|
<private>1</private>
|
|
</pastebinit>
|
|
"""
|
|
|
|
for configfile in configfiles:
|
|
# Open configuration file if it exists
|
|
try:
|
|
f = open(configfile)
|
|
configtext = f.read()
|
|
f.close()
|
|
gotconfigxml = True
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|
|
except:
|
|
gotconfigxml = False
|
|
|
|
# Parse configuration file
|
|
if gotconfigxml:
|
|
try:
|
|
configxml = xml.dom.minidom.parseString(configtext)
|
|
for variable, key in (('pastebin', 'website'),
|
|
('author', 'user'),
|
|
('format', 'format'),
|
|
('private', 'private'),
|
|
('jabberid', 'jabberid')):
|
|
try:
|
|
value = getFirstNodeText(configxml, variable)
|
|
if variable == 'pastebin':
|
|
website = value
|
|
else:
|
|
spec_opts[key] = value
|
|
except:
|
|
pass
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|
|
except:
|
|
print(_("Error parsing configuration file!"), file=sys.stderr)
|
|
print(_("Please ensure that your configuration file looks "
|
|
"similar to the following:"), file=sys.stderr)
|
|
print(configexample, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# Get options
|
|
try:
|
|
optlist, arglist = getopt.getopt(sys.argv[1:],
|
|
'EVP:hlva:b:f:i:t:j:m:u:p:')
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|
|
except getopt.GetoptError as e:
|
|
print((_("Invalid arguments: %s!") % e) + "\n", file=sys.stderr)
|
|
Usage(sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# Get the configuration of all pastebins
|
|
pastebind = preloadPastebins()
|
|
|
|
# Iterate through options
|
|
for opt in optlist:
|
|
if opt[0] == "-h":
|
|
Usage()
|
|
sys.exit(0)
|
|
elif opt[0] == "-v":
|
|
print("pastebinit v" + version)
|
|
sys.exit(0)
|
|
elif opt[0] == "-l":
|
|
print(_("Supported pastebins:"))
|
|
for pastebin in sorted(list(pastebind.keys())):
|
|
print("- %s" % pastebin)
|
|
sys.exit(0)
|
|
elif opt[0] == "-b":
|
|
website = opt[1]
|
|
elif opt[0] == "-i":
|
|
filenames.append(opt[1])
|
|
elif opt[0] == "-a":
|
|
spec_opts['user'] = opt[1]
|
|
elif opt[0] == "-t":
|
|
spec_opts['title'] = opt[1]
|
|
elif opt[0] == "-f":
|
|
spec_opts['format'] = opt[1]
|
|
elif opt[0] == "-j":
|
|
spec_opts['jabberid'] = opt[1]
|
|
elif opt[0] == "-m":
|
|
spec_opts['permatag'] = opt[1]
|
|
elif opt[0] == "-P":
|
|
spec_opts['private'] = opt[1]
|
|
elif opt[0] == "-u":
|
|
spec_opts['username'] = opt[1]
|
|
elif opt[0] == "-p":
|
|
spec_opts['password'] = opt[1]
|
|
elif opt[0] == "-E":
|
|
echo = True
|
|
elif opt[0] == "-V":
|
|
verbose = True
|
|
|
|
# Get the configuration
|
|
if '://' in website:
|
|
website = website.split('://')[1]
|
|
website = website.strip('/')
|
|
website, config = getConfig(website)
|
|
|
|
filenames += arglist
|
|
|
|
if not filenames:
|
|
filenames.append("-")
|
|
|
|
contents = []
|
|
for filename in filenames:
|
|
# If - is specified as a filename read from stdin
|
|
# otherwise load the specified files.
|
|
content = ""
|
|
if filename == "-":
|
|
content = sys.stdin.read().rstrip()
|
|
else:
|
|
try:
|
|
with open(filename, "r") as fd:
|
|
content = fd.read().rstrip()
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|
|
except:
|
|
print(_("Unable to read from: %s") % filename, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if not content:
|
|
print(_("You are trying to send an empty document, exiting."),
|
|
file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
contents.append(content)
|
|
if echo:
|
|
print(content)
|
|
|
|
if "post_page" in config:
|
|
# Use post page, without leading slash, website has a trailing one.
|
|
post_url = website + config['post_page'].lstrip('/')
|
|
else:
|
|
post_url = website
|
|
|
|
if "paste_regexp" in config:
|
|
reLink = config['paste_regexp']
|
|
else:
|
|
reLink = False
|
|
|
|
# Get target_page for replacement, only used with reLink.
|
|
if "target_page" in config:
|
|
relink_target_url = website + config['target_page'].lstrip('/')
|
|
if not reLink:
|
|
print("Warning: using target_page without paste_regexp.",
|
|
file=sys.stderr)
|
|
elif reLink:
|
|
relink_target_url = website
|
|
|
|
if "post_format" in config:
|
|
post_format = config['post_format']
|
|
else:
|
|
post_format = 'standard'
|
|
|
|
if "user_length" in config:
|
|
user_length = int(config['user_length'])
|
|
user = pastebind[config['basename']].get('defaults', 'user',
|
|
vars=spec_opts, fallback=inline_opts.get('user'))
|
|
spec_opts['user'] = "%s~" % user[:user_length-1]
|
|
|
|
for content in contents:
|
|
# Get the parameters
|
|
params = getParameters(config['basename'], content)
|
|
|
|
if "sizelimit" in config:
|
|
if len(content) > int(config['sizelimit']):
|
|
print(_("The content you are trying to send exceeds "
|
|
"the pastebin's size limit."), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
req = PasteRequest(post_url)
|
|
|
|
if post_format == 'json':
|
|
if json:
|
|
params = bytes(json.dumps(params), encoding='US-ASCII')
|
|
req.add_header('Content-Type', 'text/json')
|
|
else:
|
|
print(_("Could not find any JSON library."), file=sys.stderr)
|
|
sys.exit(1)
|
|
else:
|
|
# Convert to a format usable with the HTML POST
|
|
params = bytes(urlencode(params), encoding='US-ASCII')
|
|
|
|
# Send the informations and be redirected to the final page
|
|
if verbose:
|
|
print("POSTing to: %s\nParams: %s" % (
|
|
post_url, str(params)), file=sys.stderr)
|
|
try:
|
|
page = request.urlopen(req, params)
|
|
except Exception as e:
|
|
print(_("Failed to contact the server: %s") % e, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
try:
|
|
# Check if we have to apply a regexp
|
|
paste_url = ""
|
|
if reLink:
|
|
result = page.read().decode('utf-8').strip()
|
|
if reLink == '(.*)':
|
|
paste_url = result
|
|
else:
|
|
# Print the result of the regexp
|
|
result = re.split(reLink, result)[1]
|
|
if '%s' not in relink_target_url:
|
|
paste_url = relink_target_url + result
|
|
else:
|
|
paste_url = relink_target_url % result
|
|
else:
|
|
# Get the final page
|
|
paste_url = page.url
|
|
|
|
# Print the paste URL
|
|
if echo:
|
|
print("-" * len(paste_url))
|
|
print(paste_url)
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|
|
except:
|
|
print(_("Unable to read or parse the result page, it could be a "
|
|
"server timeout or a change server side, "
|
|
"try with another pastebin."), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
except KeyboardInterrupt:
|
|
print(_("KeyboardInterrupt caught."), file=sys.stderr)
|
|
sys.exit(1)
|