/usr/lib/python2.7/dist-packages/pylxd/deprecated/connection.py is in python-pylxd 2.2.6-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 | # Copyright (c) 2015 Canonical Ltd
# Copyright (c) 2015 Mirantis inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from collections import namedtuple
import copy
import json
import os
import six
import socket
import ssl
import threading
from six.moves import http_client
from six.moves import queue
try:
from ws4py import client as websocket
except ImportError:
websocket = None
from pylxd.deprecated import exceptions
from pylxd.deprecated import utils
if hasattr(ssl, 'SSLContext'):
# For Python >= 2.7.9 and Python 3.x
if hasattr(ssl, 'PROTOCOL_TLSv1_2'):
DEFAULT_TLS_VERSION = ssl.PROTOCOL_TLSv1_2
else:
DEFAULT_TLS_VERSION = ssl.PROTOCOL_TLSv1
else:
# For Python 2.6 and <= 2.7.8
from OpenSSL import SSL
DEFAULT_TLS_VERSION = SSL.TLSv1_2_METHOD
class UnixHTTPConnection(http_client.HTTPConnection):
def __init__(self, path, host='localhost', port=None, strict=None,
timeout=None):
if six.PY3:
http_client.HTTPConnection.__init__(self, host, port=port,
timeout=timeout)
else:
http_client.HTTPConnection.__init__(self, host, port=port,
strict=strict,
timeout=timeout)
self.path = path
def connect(self):
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.connect(self.path)
self.sock = sock
class HTTPSConnection(http_client.HTTPConnection):
default_port = 8443
def __init__(self, *args, **kwargs):
http_client.HTTPConnection.__init__(self, *args, **kwargs)
def connect(self):
sock = socket.create_connection((self.host, self.port),
self.timeout, self.source_address)
if self._tunnel_host:
self.sock = sock
self._tunnel()
(cert_file, key_file) = self._get_ssl_certs()
self.sock = ssl.wrap_socket(sock, certfile=cert_file,
keyfile=key_file,
ssl_version=DEFAULT_TLS_VERSION)
@staticmethod
def _get_ssl_certs():
return (os.path.join(os.environ['HOME'], '.config/lxc/client.crt'),
os.path.join(os.environ['HOME'], '.config/lxc/client.key'))
_LXDResponse = namedtuple('LXDResponse', ['status', 'body', 'json'])
if websocket is not None:
class WebSocketClient(websocket.WebSocketBaseClient):
def __init__(self, url, protocols=None, extensions=None,
ssl_options=None, headers=None):
"""WebSocket client that executes into a eventlet green thread."""
websocket.WebSocketBaseClient.__init__(self, url, protocols,
extensions,
ssl_options=ssl_options,
headers=headers)
self._th = threading.Thread(
target=self.run, name='WebSocketClient')
self._th.daemon = True
self.messages = queue.Queue()
def handshake_ok(self):
"""Starts the client's thread."""
self._th.start()
def received_message(self, message):
"""Override the base class to store the incoming message."""
self.messages.put(copy.deepcopy(message))
def closed(self, code, reason=None):
# When the connection is closed, put a StopIteration
# on the message queue to signal there's nothing left
# to wait for
self.messages.put(StopIteration)
def receive(self):
# If the websocket was terminated and there are no messages
# left in the queue, return None immediately otherwise the client
# will block forever
if self.terminated and self.messages.empty():
return None
message = self.messages.get()
if message is StopIteration:
return None
return message
class LXDConnection(object):
def __init__(self, host=None, port=8443):
if host:
self.host = host
self.port = port
self.unix_socket = None
else:
if 'LXD_DIR' in os.environ:
self.unix_socket = os.path.join(os.environ['LXD_DIR'],
'unix.socket')
else:
self.unix_socket = '/var/lib/lxd/unix.socket'
self.host, self.port = None, None
self.connection = None
def _request(self, *args, **kwargs):
if self.connection is None:
self.connection = self.get_connection()
self.connection.request(*args, **kwargs)
response = self.connection.getresponse()
status = response.status
raw_body = response.read()
try:
if six.PY3:
body = json.loads(raw_body.decode())
else:
body = json.loads(raw_body)
except ValueError:
body = None
return _LXDResponse(status, raw_body, body)
def get_connection(self):
if self.host:
return HTTPSConnection(self.host, self.port)
return UnixHTTPConnection(self.unix_socket)
def get_object(self, *args, **kwargs):
response = self._request(*args, **kwargs)
if not response.json:
raise exceptions.PyLXDException('Null Data')
elif response.status == 200 or (
response.status == 202 and
response.json.get('status_code') == 100):
return response.status, response.json
else:
utils.get_lxd_error(response.status, response.json)
def get_status(self, *args, **kwargs):
response = self._request(*args, **kwargs)
if not response.json:
raise exceptions.PyLXDException('Null Data')
elif response.json.get('error'):
utils.get_lxd_error(response.status, response.json)
elif response.status == 200 or (
response.status == 202 and
response.json.get('status_code') == 100):
return True
return False
def get_raw(self, *args, **kwargs):
response = self._request(*args, **kwargs)
if not response.body:
raise exceptions.PyLXDException('Null Body')
elif response.status == 200:
return response.body
else:
raise exceptions.PyLXDException('Failed to get raw response')
def get_ws(self, path):
if websocket is None:
raise ValueError(
'This feature requires the optional ws4py library.')
if self.unix_socket:
connection_string = 'ws+unix://%s' % self.unix_socket
else:
connection_string = (
'wss://%(host)s:%(port)s' % {'host': self.host,
'port': self.port}
)
ws = WebSocketClient(connection_string)
ws.resource = path
ws.connect()
return ws
|