Optionally disable ssl verification for mjpeg (#19277)

This commit is contained in:
Rohan Kapoor 2018-12-18 02:22:47 -08:00 committed by Paulus Schoutsen
parent 2f6ef08959
commit 6c64b315db
4 changed files with 25 additions and 10 deletions

View File

@ -16,7 +16,7 @@ import voluptuous as vol
from homeassistant.const import ( from homeassistant.const import (
CONF_NAME, CONF_USERNAME, CONF_PASSWORD, CONF_AUTHENTICATION, CONF_NAME, CONF_USERNAME, CONF_PASSWORD, CONF_AUTHENTICATION,
HTTP_BASIC_AUTHENTICATION, HTTP_DIGEST_AUTHENTICATION) HTTP_BASIC_AUTHENTICATION, HTTP_DIGEST_AUTHENTICATION, CONF_VERIFY_SSL)
from homeassistant.components.camera import (PLATFORM_SCHEMA, Camera) from homeassistant.components.camera import (PLATFORM_SCHEMA, Camera)
from homeassistant.helpers.aiohttp_client import ( from homeassistant.helpers.aiohttp_client import (
async_get_clientsession, async_aiohttp_proxy_web) async_get_clientsession, async_aiohttp_proxy_web)
@ -29,6 +29,7 @@ CONF_STILL_IMAGE_URL = 'still_image_url'
CONTENT_TYPE_HEADER = 'Content-Type' CONTENT_TYPE_HEADER = 'Content-Type'
DEFAULT_NAME = 'Mjpeg Camera' DEFAULT_NAME = 'Mjpeg Camera'
DEFAULT_VERIFY_SSL = True
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_MJPEG_URL): cv.url, vol.Required(CONF_MJPEG_URL): cv.url,
@ -38,6 +39,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_PASSWORD): cv.string, vol.Optional(CONF_PASSWORD): cv.string,
vol.Optional(CONF_USERNAME): cv.string, vol.Optional(CONF_USERNAME): cv.string,
vol.Optional(CONF_VERIFY_SSL, default=DEFAULT_VERIFY_SSL): cv.boolean,
}) })
@ -95,6 +97,7 @@ class MjpegCamera(Camera):
self._auth = aiohttp.BasicAuth( self._auth = aiohttp.BasicAuth(
self._username, password=self._password self._username, password=self._password
) )
self._verify_ssl = device_info.get(CONF_VERIFY_SSL)
async def async_camera_image(self): async def async_camera_image(self):
"""Return a still image response from the camera.""" """Return a still image response from the camera."""
@ -105,7 +108,10 @@ class MjpegCamera(Camera):
self.camera_image) self.camera_image)
return image return image
websession = async_get_clientsession(self.hass) websession = async_get_clientsession(
self.hass,
verify_ssl=self._verify_ssl
)
try: try:
with async_timeout.timeout(10, loop=self.hass.loop): with async_timeout.timeout(10, loop=self.hass.loop):
response = await websession.get( response = await websession.get(
@ -128,7 +134,12 @@ class MjpegCamera(Camera):
else: else:
auth = HTTPBasicAuth(self._username, self._password) auth = HTTPBasicAuth(self._username, self._password)
req = requests.get( req = requests.get(
self._mjpeg_url, auth=auth, stream=True, timeout=10) self._mjpeg_url,
auth=auth,
stream=True,
timeout=10,
verify=self._verify_ssl
)
else: else:
req = requests.get(self._mjpeg_url, stream=True, timeout=10) req = requests.get(self._mjpeg_url, stream=True, timeout=10)
@ -144,7 +155,10 @@ class MjpegCamera(Camera):
return await super().handle_async_mjpeg_stream(request) return await super().handle_async_mjpeg_stream(request)
# connect to stream # connect to stream
websession = async_get_clientsession(self.hass) websession = async_get_clientsession(
self.hass,
verify_ssl=self._verify_ssl
)
stream_coro = websession.get(self._mjpeg_url, auth=self._auth) stream_coro = websession.get(self._mjpeg_url, auth=self._auth)
return await async_aiohttp_proxy_web(self.hass, request, stream_coro) return await async_aiohttp_proxy_web(self.hass, request, stream_coro)

View File

@ -6,7 +6,7 @@ https://home-assistant.io/components/camera.zoneminder/
""" """
import logging import logging
from homeassistant.const import CONF_NAME from homeassistant.const import CONF_NAME, CONF_VERIFY_SSL
from homeassistant.components.camera.mjpeg import ( from homeassistant.components.camera.mjpeg import (
CONF_MJPEG_URL, CONF_STILL_IMAGE_URL, MjpegCamera) CONF_MJPEG_URL, CONF_STILL_IMAGE_URL, MjpegCamera)
from homeassistant.components.zoneminder import DOMAIN as ZONEMINDER_DOMAIN from homeassistant.components.zoneminder import DOMAIN as ZONEMINDER_DOMAIN
@ -28,19 +28,20 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
cameras = [] cameras = []
for monitor in monitors: for monitor in monitors:
_LOGGER.info("Initializing camera %s", monitor.id) _LOGGER.info("Initializing camera %s", monitor.id)
cameras.append(ZoneMinderCamera(monitor)) cameras.append(ZoneMinderCamera(monitor, zm_client.verify_ssl))
add_entities(cameras) add_entities(cameras)
class ZoneMinderCamera(MjpegCamera): class ZoneMinderCamera(MjpegCamera):
"""Representation of a ZoneMinder Monitor Stream.""" """Representation of a ZoneMinder Monitor Stream."""
def __init__(self, monitor): def __init__(self, monitor, verify_ssl):
"""Initialize as a subclass of MjpegCamera.""" """Initialize as a subclass of MjpegCamera."""
device_info = { device_info = {
CONF_NAME: monitor.name, CONF_NAME: monitor.name,
CONF_MJPEG_URL: monitor.mjpeg_image_url, CONF_MJPEG_URL: monitor.mjpeg_image_url,
CONF_STILL_IMAGE_URL: monitor.still_image_url CONF_STILL_IMAGE_URL: monitor.still_image_url,
CONF_VERIFY_SSL: verify_ssl
} }
super().__init__(device_info) super().__init__(device_info)
self._is_recording = None self._is_recording = None

View File

@ -15,7 +15,7 @@ import homeassistant.helpers.config_validation as cv
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
REQUIREMENTS = ['zm-py==0.2.0'] REQUIREMENTS = ['zm-py==0.3.0']
CONF_PATH_ZMS = 'path_zms' CONF_PATH_ZMS = 'path_zms'

View File

@ -1710,4 +1710,4 @@ zigpy-xbee==0.1.1
zigpy==0.2.0 zigpy==0.2.0
# homeassistant.components.zoneminder # homeassistant.components.zoneminder
zm-py==0.2.0 zm-py==0.3.0