No OneTemporary
Actions

Size

30 KB

Referenced Files

None

Subscribers

None

View Options

	diff --git a/heartbeat/gcp-vpc-move-route.in b/heartbeat/gcp-vpc-move-route.in
	index 179eba15a..9fe985832 100644
	--- a/heartbeat/gcp-vpc-move-route.in
	+++ b/heartbeat/gcp-vpc-move-route.in
	@@ -1,451 +1,462 @@
	#!@PYTHON@ -tt
	# - - coding: utf- 8 - -
	#
	#
	# OCF resource agent to move an IP address within a VPC in GCP
	#
	# License: GNU General Public License (GPL)
	# Copyright (c) 2018 Hervé Werner (MFG Labs)
	# Copyright 2018 Google Inc.
	# Based on code from Markus Guertler (aws-vpc-move-ip)
	# All Rights Reserved.
	#
	# This program is free software; you can redistribute it and/or modify
	# it under the terms of version 2 of the GNU General Public License as
	# published by the Free Software Foundation.
	#
	# This program is distributed in the hope that it would be useful, but
	# WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
	#
	# Further, this software is distributed without any warranty that it is
	# free of the rightful claim of any third person regarding infringement
	# or the like. Any license provided herein, whether implied or
	# otherwise, applies only to this software file. Patent licenses, if
	# any, provided herein do not apply to combinations of this program with
	# other software, or any other product whatsoever.
	#
	# You should have received a copy of the GNU General Public License
	# along with this program; if not, write the Free Software Foundation,
	# Inc., 59 Temple Place - Suite 330, Boston MA 02111-1307, USA.
	#


	#######################################################################

	import atexit
	import logging
	import os
	import sys
	import time

	OCF_FUNCTIONS_DIR = os.environ.get("OCF_FUNCTIONS_DIR", "%s/lib/heartbeat" % os.environ.get("OCF_ROOT"))
	sys.path.append(OCF_FUNCTIONS_DIR)

	from ocf import *

	try:
	import googleapiclient.discovery
	import pyroute2
	except ImportError:
	pass

	if sys.version_info >= (3, 0):
	# Python 3 imports.
	import urllib.parse as urlparse
	import urllib.request as urlrequest
	else:
	# Python 2 imports.
	import urllib as urlparse
	import urllib2 as urlrequest


	GCP_API_URL_PREFIX = 'https://www.googleapis.com/compute/v1'
	METADATA_SERVER = 'http://metadata.google.internal/computeMetadata/v1/'
	METADATA_HEADERS = {'Metadata-Flavor': 'Google'}
	METADATA = \
	'''<?xml version="1.0"?>
	<!DOCTYPE resource-agent SYSTEM "ra-api-1.dtd">
	<resource-agent name="gcp-vpc-move-route">
	<version>1.0</version>
	<longdesc lang="en">
	Resource Agent that can move a floating IP addresse within a GCP VPC by changing an
	entry in the routing table. This agent also configures the floating IP locally
	on the instance OS.
	Requirements :
	- IP forwarding must be enabled on all instances in order to be able to
	terminate the route
	- The floating IP address must be chosen so that it is outside all existing
	subnets in the VPC network
	- IAM permissions
	(see https://cloud.google.com/compute/docs/access/iam-permissions) :
	1) compute.routes.delete, compute.routes.get and compute.routes.update on the
	route
	2) compute.networks.updatePolicy on the network (to add a new route)
	3) compute.networks.get on the network (to check the VPC network existence)
	4) compute.routes.list on the project (to check conflicting routes)
	</longdesc>
	<shortdesc lang="en">Move IP within a GCP VPC</shortdesc>

	<parameters>

	<parameter name="ip" unique="1" required="1">
	<longdesc lang="en">
	Floating IP address. Note that this IP must be chosen outside of all existing
	subnet ranges
	</longdesc>
	<shortdesc lang="en">Floating IP</shortdesc>
	<content type="string" />
	</parameter>

	<parameter name="vpc_network" required="1">
	<longdesc lang="en">
	Name of the VPC network
	</longdesc>
	<shortdesc lang="en">VPC network</shortdesc>
	<content type="string" default="default" />
	</parameter>

	<parameter name="project">
	<longdesc lang="en">
	Project ID of the instance. It can be useful to set this attribute if
	the instance is in a shared service project. Otherwise, the agent should
	be able to determine the project ID automatically.
	</longdesc>
	<shortdesc lang="en">Project ID</shortdesc>
	<content type="string" default="default" />
	</parameter>

	<parameter name="interface">
	<longdesc lang="en">
	Name of the network interface
	</longdesc>
	<shortdesc lang="en">Network interface name</shortdesc>
	<content type="string" default="eth0" />
	</parameter>

	<parameter name="route_name" unique="1">
	<longdesc lang="en">
	Route name
	</longdesc>
	<shortdesc lang="en">Route name</shortdesc>
	<content type="string" default="ra-%s" />
	</parameter>

	<parameter name="stackdriver_logging" unique="0" required="0">
	<longdesc lang="en">If enabled (set to true), IP failover logs will be posted to stackdriver logging</longdesc>
	<shortdesc lang="en">Stackdriver-logging support</shortdesc>
	<content type="boolean" default="" />
	</parameter>
	</parameters>

	<actions>
	<action name="start" timeout="180s" />
	<action name="stop" timeout="180s" />
	<action name="monitor" depth="0" timeout="30s" interval="60s" />
	<action name="validate-all" timeout="5s" />
	<action name="meta-data" timeout="5s" />
	</actions>
	</resource-agent>
	''' % os.path.basename(sys.argv[0])


	class Context(object):
	__slots__ = 'conn', 'iface_idx', 'instance', 'instance_url', 'interface', \
	'ip', 'iproute', 'project', 'route_name', 'vpc_network', \
	'vpc_network_url', 'zone'


	def wait_for_operation(ctx, response):
	"""Blocks until operation completes.
	Code from GitHub's GoogleCloudPlatform/python-docs-samples

	Args:
	response: dict, a request's response
	"""
	def _OperationGetter(response):
	operation = response[u'name']
	if response.get(u'zone'):
	return ctx.conn.zoneOperations().get(
	project=ctx.project, zone=ctx.zone, operation=operation)
	else:
	return ctx.conn.globalOperations().get(
	project=ctx.project, operation=operation)

	while True:
	result = _OperationGetter(response).execute()

	if result['status'] == 'DONE':
	if 'error' in result:
	raise Exception(result['error'])
	return result

	time.sleep(1)


	def get_metadata(metadata_key, params=None, timeout=None):
	"""Performs a GET request with the metadata headers.

	Args:
	metadata_key: string, the metadata to perform a GET request on.
	params: dictionary, the query parameters in the GET request.
	timeout: int, timeout in seconds for metadata requests.

	Returns:
	HTTP response from the GET request.

	Raises:
	urlerror.HTTPError: raises when the GET request fails.
	"""
	timeout = timeout or 60
	metadata_url = os.path.join(METADATA_SERVER, metadata_key)
	params = urlparse.urlencode(params or {})
	url = '%s?%s' % (metadata_url, params)
	request = urlrequest.Request(url, headers=METADATA_HEADERS)
	request_opener = urlrequest.build_opener(urlrequest.ProxyHandler({}))
	return request_opener.open(request, timeout=timeout * 1.1).read().decode("utf-8")


	def validate(ctx):
	if os.geteuid() != 0:
	logger.error('You must run this agent as root')
	sys.exit(OCF_ERR_PERM)

	try:
	ctx.conn = googleapiclient.discovery.build('compute', 'v1', cache_discovery=False)
	except Exception as e:
	logger.error('Couldn\'t connect with google api: ' + str(e))
	sys.exit(OCF_ERR_CONFIGURED)

	ctx.ip = os.environ.get('OCF_RESKEY_ip')
	if not ctx.ip:
	logger.error('Missing ip parameter')
	sys.exit(OCF_ERR_CONFIGURED)

	try:
	ctx.instance = get_metadata('instance/name')
	ctx.zone = get_metadata('instance/zone').split('/')[-1]
	ctx.project = os.environ.get(
	'OCF_RESKEY_project', get_metadata('project/project-id'))
	except Exception as e:
	logger.error(
	'Instance information not found. Is this a GCE instance ?: %s', str(e))
	sys.exit(OCF_ERR_CONFIGURED)

	ctx.instance_url = '%s/projects/%s/zones/%s/instances/%s' % (
	GCP_API_URL_PREFIX, ctx.project, ctx.zone, ctx.instance)
	ctx.vpc_network = os.environ.get('OCF_RESKEY_vpc_network', 'default')
	ctx.vpc_network_url = '%s/projects/%s/global/networks/%s' % (
	GCP_API_URL_PREFIX, ctx.project, ctx.vpc_network)
	ctx.interface = os.environ.get('OCF_RESKEY_interface', 'eth0')
	ctx.route_name = os.environ.get(
	'OCF_RESKEY_route_name', 'ra-%s' % os.path.basename(sys.argv[0]))
	ctx.iproute = pyroute2.IPRoute()
	atexit.register(ctx.iproute.close)
	idxs = ctx.iproute.link_lookup(ifname=ctx.interface)
	if not idxs:
	logger.error('Network interface not found')
	sys.exit(OCF_ERR_CONFIGURED)
	ctx.iface_idx = idxs[0]


	def check_conflicting_routes(ctx):
	fl = '(destRange = "%s*") AND (network = "%s") AND (name != "%s")' % (
	ctx.ip, ctx.vpc_network_url, ctx.route_name)
	- request = ctx.conn.routes().list(project=ctx.project, filter=fl)
	- response = request.execute()
	+ try:
	+ request = ctx.conn.routes().list(project=ctx.project, filter=fl)
	+ response = request.execute()
	+ except googleapiclient.errors.HttpError as e:
	+ if e.resp.status == 404:
	+ logger.error('VPC network not found')
	+ if 'stop' in sys.argv[1]:
	+ sys.exit(OCF_SUCCESS)
	+ else:
	+ sys.exit(OCF_ERR_CONFIGURED)
	+ else:
	+ raise
	+
	route_list = response.get('items', None)
	if route_list:
	logger.error(
	'Conflicting unnmanaged routes for destination %s/32 in VPC %s found : %s',
	ctx.ip, ctx.vpc_network, str(route_list))
	sys.exit(OCF_ERR_CONFIGURED)


	def route_release(ctx):
	request = ctx.conn.routes().delete(project=ctx.project, route=ctx.route_name)
	wait_for_operation(ctx, request.execute())


	def ip_monitor(ctx):
	logger.info('IP monitor: checking local network configuration')

	def address_filter(addr):
	for attr in addr['attrs']:
	if attr[0] == 'IFA_LOCAL':
	if attr[1] == ctx.ip:
	return True
	else:
	return False

	route = ctx.iproute.get_addr(
	index=ctx.iface_idx, match=address_filter)
	if not route:
	logger.warning(
	'The floating IP %s is not locally configured on this instance (%s)',
	ctx.ip, ctx.instance)
	return OCF_NOT_RUNNING

	logger.debug(
	'The floating IP %s is correctly configured on this instance (%s)',
	ctx.ip, ctx.instance)
	return OCF_SUCCESS


	def ip_release(ctx):
	ctx.iproute.addr('del', index=ctx.iface_idx, address=ctx.ip, mask=32)


	def ip_and_route_start(ctx):
	logger.info('Bringing up the floating IP %s', ctx.ip)

	# Add a new entry in the routing table
	# If the route entry exists and is pointing to another instance, take it over

	# Ensure that there is no route that we are not aware of that is also handling our IP
	check_conflicting_routes(ctx)

	# There is no replace API, We need to first delete the existing route if any
	try:
	request = ctx.conn.routes().get(project=ctx.project, route=ctx.route_name)
	request.execute()
	# TODO: check specific exception for 404
	except googleapiclient.errors.HttpError as e:
	if e.resp.status != 404:
	raise
	else:
	route_release(ctx)

	route_body = {
	'name': ctx.route_name,
	'network': ctx.vpc_network_url,
	'destRange': '%s/32' % ctx.ip,
	'nextHopInstance': ctx.instance_url,
	}
	try:
	request = ctx.conn.routes().insert(project=ctx.project, body=route_body)
	wait_for_operation(ctx, request.execute())
	except googleapiclient.errors.HttpError:
	try:
	request = ctx.conn.networks().get(
	project=ctx.project, network=ctx.vpc_network)
	request.execute()
	except googleapiclient.errors.HttpError as e:
	if e.resp.status == 404:
	logger.error('VPC network not found')
	sys.exit(OCF_ERR_CONFIGURED)
	else:
	raise
	else:
	raise

	# Configure the IP address locally
	# We need to release the IP first
	if ip_monitor(ctx) == OCF_SUCCESS:
	ip_release(ctx)

	ctx.iproute.addr('add', index=ctx.iface_idx, address=ctx.ip, mask=32)
	ctx.iproute.link('set', index=ctx.iface_idx, state='up')
	logger.info('Successfully brought up the floating IP %s', ctx.ip)


	def route_monitor(ctx):
	logger.info('GCP route monitor: checking route table')

	# Ensure that there is no route that we are not aware of that is also handling our IP
	- check_conflicting_routes
	+ check_conflicting_routes(ctx)

	try:
	request = ctx.conn.routes().get(project=ctx.project, route=ctx.route_name)
	response = request.execute()
	except googleapiclient.errors.HttpError as e:
	- if 'Insufficient Permission' in e.content:
	- return OCF_ERR_PERM
	- elif e.resp.status == 404:
	+ if e.resp.status == 404:
	return OCF_NOT_RUNNING
	+ elif 'Insufficient Permission' in e.content:
	+ return OCF_ERR_PERM
	else:
	raise

	routed_to_instance = response.get('nextHopInstance', '<unknown>')
	instance_url = '%s/projects/%s/zones/%s/instances/%s' % (
	GCP_API_URL_PREFIX, ctx.project, ctx.zone, ctx.instance)
	if routed_to_instance != instance_url:
	logger.warning(
	'The floating IP %s is not routed to this instance (%s) but to instance %s',
	ctx.ip, ctx.instance, routed_to_instance.split('/')[-1])
	return OCF_NOT_RUNNING

	logger.debug(
	'The floating IP %s is correctly routed to this instance (%s)',
	ctx.ip, ctx.instance)
	return OCF_SUCCESS


	def ip_and_route_stop(ctx):
	logger.info('Bringing down the floating IP %s', ctx.ip)

	# Delete the route entry
	# If the route entry exists and is pointing to another instance, don't touch it
	if route_monitor(ctx) == OCF_NOT_RUNNING:
	logger.info(
	'The floating IP %s is already not routed to this instance (%s)',
	ctx.ip, ctx.instance)
	else:
	route_release(ctx)

	if ip_monitor(ctx) == OCF_NOT_RUNNING:
	logger.info('The floating IP %s is already down', ctx.ip)
	else:
	ip_release(ctx)


	def configure_logs(ctx):
	# Prepare logging
	global logger
	logging.getLogger('googleapiclient').setLevel(logging.WARN)
	logging_env = os.environ.get('OCF_RESKEY_stackdriver_logging')
	if logging_env:
	logging_env = logging_env.lower()
	if any(x in logging_env for x in ['yes', 'true', 'enabled']):
	try:
	import google.cloud.logging.handlers
	client = google.cloud.logging.Client()
	handler = google.cloud.logging.handlers.CloudLoggingHandler(
	client, name=ctx.instance)
	handler.setLevel(logging.INFO)
	formatter = logging.Formatter('gcp:route "%(message)s"')
	handler.setFormatter(formatter)
	log.addHandler(handler)
	logger = logging.LoggerAdapter(log, {'OCF_RESOURCE_INSTANCE': OCF_RESOURCE_INSTANCE})
	except ImportError:
	logger.error('Couldn\'t import google.cloud.logging, '
	'disabling Stackdriver-logging support')


	def main():
	if 'meta-data' in sys.argv[1]:
	print(METADATA)
	return

	ctx = Context()

	validate(ctx)
	if 'validate-all' in sys.argv[1]:
	return

	configure_logs(ctx)
	if 'start' in sys.argv[1]:
	ip_and_route_start(ctx)
	elif 'stop' in sys.argv[1]:
	ip_and_route_stop(ctx)
	elif 'status' in sys.argv[1] or 'monitor' in sys.argv[1]:
	sys.exit(ip_monitor(ctx))
	else:
	usage = 'usage: %s {start\|stop\|monitor\|status\|meta-data\|validate-all}' % \
	os.path.basename(sys.argv[0])
	logger.error(usage)
	sys.exit(OCF_ERR_UNIMPLEMENTED)


	if __name__ == "__main__":
	main()
	diff --git a/heartbeat/gcp-vpc-move-vip.in b/heartbeat/gcp-vpc-move-vip.in
	index e792f71d5..bbbd87b7a 100755
	--- a/heartbeat/gcp-vpc-move-vip.in
	+++ b/heartbeat/gcp-vpc-move-vip.in
	@@ -1,420 +1,430 @@
	#!@PYTHON@ -tt
	# ---------------------------------------------------------------------
	# Copyright 2016 Google 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.
	# ---------------------------------------------------------------------
	# Description: Google Cloud Platform - Floating IP Address (Alias)
	# ---------------------------------------------------------------------

	import json
	import logging
	import os
	import sys
	import time

	OCF_FUNCTIONS_DIR = os.environ.get("OCF_FUNCTIONS_DIR", "%s/lib/heartbeat"
	% os.environ.get("OCF_ROOT"))
	sys.path.append(OCF_FUNCTIONS_DIR)

	from ocf import *

	try:
	import googleapiclient.discovery
	except ImportError:
	pass

	if sys.version_info >= (3, 0):
	# Python 3 imports.
	import urllib.parse as urlparse
	import urllib.request as urlrequest
	else:
	# Python 2 imports.
	import urllib as urlparse
	import urllib2 as urlrequest


	# Constants for alias add/remove modes
	ADD = 0
	REMOVE = 1

	CONN = None
	THIS_VM = None
	ALIAS = None
	METADATA_SERVER = 'http://metadata.google.internal/computeMetadata/v1/'
	METADATA_HEADERS = {'Metadata-Flavor': 'Google'}
	METADATA = \
	'''<?xml version="1.0"?>
	<!DOCTYPE resource-agent SYSTEM "ra-api-1.dtd">
	<resource-agent name="gcp-vpc-move-vip">
	<version>1.0</version>
	<longdesc lang="en">Floating IP Address or Range on Google Cloud Platform - Using Alias IP address functionality to attach a secondary IP range to a running instance</longdesc>
	<shortdesc lang="en">Floating IP Address or Range on Google Cloud Platform</shortdesc>
	<parameters>
	<parameter name="alias_ip" unique="1" required="1">
	<longdesc lang="en">IP range to be added including CIDR netmask (e.g., 192.168.0.1/32)</longdesc>
	<shortdesc lang="en">IP range to be added including CIDR netmask (e.g., 192.168.0.1/32)</shortdesc>
	<content type="string" default="" />
	</parameter>
	<parameter name="alias_range_name" unique="0" required="0">
	<longdesc lang="en">Subnet name for the Alias IP</longdesc>
	<shortdesc lang="en">Subnet name for the Alias IP</shortdesc>
	<content type="string" default="" />
	</parameter>
	<parameter name="hostlist" unique="0" required="0">
	<longdesc lang="en">List of hosts in the cluster, separated by spaces</longdesc>
	<shortdesc lang="en">Host list</shortdesc>
	<content type="string" default="" />
	</parameter>
	<parameter name="project" unique="0" required="0">
	<longdesc lang="en">
	Project ID of the instance. It can be useful to set this
	attribute if the instance is in a shared service project.
	Otherwise, the agent should be able to determine the project ID
	automatically.
	</longdesc>
	<shortdesc lang="en">Project ID</shortdesc>
	<content type="string" default="default" />
	</parameter>
	<parameter name="stackdriver_logging" unique="0" required="0">
	<longdesc lang="en">If enabled (set to true), IP failover logs will be posted to stackdriver logging</longdesc>
	<shortdesc lang="en">Stackdriver-logging support</shortdesc>
	<content type="boolean" default="" />
	</parameter>
	</parameters>
	<actions>
	<action name="start" timeout="300s" />
	<action name="stop" timeout="15s" />
	<action name="monitor" timeout="15s" interval="60s" depth="0" />
	<action name="meta-data" timeout="15s" />
	<action name="validate-all" timeout="15s" />
	</actions>
	</resource-agent>'''


	def get_metadata(metadata_key, params=None, timeout=None):
	"""Performs a GET request with the metadata headers.

	Args:
	metadata_key: string, the metadata to perform a GET request on.
	params: dictionary, the query parameters in the GET request.
	timeout: int, timeout in seconds for metadata requests.

	Returns:
	HTTP response from the GET request.

	Raises:
	urlerror.HTTPError: raises when the GET request fails.
	"""
	timeout = timeout or 60
	metadata_url = os.path.join(METADATA_SERVER, metadata_key)
	params = urlparse.urlencode(params or {})
	url = '%s?%s' % (metadata_url, params)
	request = urlrequest.Request(url, headers=METADATA_HEADERS)
	request_opener = urlrequest.build_opener(urlrequest.ProxyHandler({}))
	return request_opener.open(
	request, timeout=timeout * 1.1).read().decode("utf-8")


	def get_instance(project, zone, instance):
	request = CONN.instances().get(
	project=project, zone=zone, instance=instance)
	return request.execute()


	def get_network_ifaces(project, zone, instance):
	return get_instance(project, zone, instance)['networkInterfaces']


	def wait_for_operation(project, zone, operation):
	while True:
	result = CONN.zoneOperations().get(
	project=project,
	zone=zone,
	operation=operation['name']).execute()

	if result['status'] == 'DONE':
	if 'error' in result:
	raise Exception(result['error'])
	return
	time.sleep(1)


	def set_aliases(project, zone, instance, aliases, fingerprint):
	"""Sets the alias IP ranges for an instance.

	Args:
	project: string, the project in which the instance resides.
	zone: string, the zone in which the instance resides.
	instance: string, the name of the instance.
	aliases: list, the list of dictionaries containing alias IP ranges
	to be added to or removed from the instance.
	fingerprint: string, the fingerprint of the network interface.
	"""
	body = {
	'aliasIpRanges': aliases,
	'fingerprint': fingerprint
	}

	request = CONN.instances().updateNetworkInterface(
	instance=instance, networkInterface='nic0', project=project, zone=zone,
	body=body)
	operation = request.execute()
	wait_for_operation(project, zone, operation)


	def add_rm_alias(mode, project, zone, instance, alias, alias_range_name=None):
	"""Adds or removes an alias IP range for a GCE instance.

	Args:
	mode: int, a constant (ADD (0) or REMOVE (1)) indicating the
	operation type.
	project: string, the project in which the instance resides.
	zone: string, the zone in which the instance resides.
	instance: string, the name of the instance.
	alias: string, the alias IP range to be added to or removed from
	the instance.
	alias_range_name: string, the subnet name for the alias IP range.

	Returns:
	True if the existing list of alias IP ranges was modified, or False
	otherwise.
	"""
	ifaces = get_network_ifaces(project, zone, instance)
	fingerprint = ifaces[0]['fingerprint']

	try:
	old_aliases = ifaces[0]['aliasIpRanges']
	except KeyError:
	old_aliases = []

	new_aliases = [a for a in old_aliases if a['ipCidrRange'] != alias]

	if alias:
	if mode == ADD:
	obj = {'ipCidrRange': alias}
	if alias_range_name:
	obj['subnetworkRangeName'] = alias_range_name
	new_aliases.append(obj)
	elif mode == REMOVE:
	pass # already removed during new_aliases build
	else:
	raise ValueError('Invalid value for mode: {}'.format(mode))

	if (sorted(new_aliases, key=lambda item: item.get('ipCidrRange'))
	!= sorted(old_aliases, key=lambda item: item.get('ipCidrRange'))):
	set_aliases(project, zone, instance, new_aliases, fingerprint)
	return True
	else:
	return False


	def add_alias(project, zone, instance, alias, alias_range_name=None):
	return add_rm_alias(ADD, project, zone, instance, alias, alias_range_name)


	def remove_alias(project, zone, instance, alias):
	return add_rm_alias(REMOVE, project, zone, instance, alias)


	def get_aliases(project, zone, instance):
	ifaces = get_network_ifaces(project, zone, instance)
	try:
	aliases = ifaces[0]['aliasIpRanges']
	return [a['ipCidrRange'] for a in aliases]
	except KeyError:
	return []


	def get_localhost_aliases():
	net_iface = get_metadata('instance/network-interfaces', {'recursive': True})
	net_iface = json.loads(net_iface)
	try:
	return net_iface[0]['ipAliases']
	except (KeyError, IndexError):
	return []


	def get_zone(project, instance):
	fl = 'name="%s"' % instance
	request = CONN.instances().aggregatedList(project=project, filter=fl)
	while request is not None:
	response = request.execute()
	zones = response.get('items', {})
	for zone in zones.values():
	for inst in zone.get('instances', []):
	if inst['name'] == instance:
	return inst['zone'].split("/")[-1]
	request = CONN.instances().aggregatedList_next(
	previous_request=request, previous_response=response)
	raise Exception("Unable to find instance %s" % (instance))


	def get_instances_list(project, exclude):
	hostlist = []
	request = CONN.instances().aggregatedList(project=project)
	while request is not None:
	- response = request.execute()
	- zones = response.get('items', {})
	+ try:
	+ response = request.execute()
	+ zones = response.get('items', {})
	+ except googleapiclient.errors.HttpError as e:
	+ if e.resp.status == 404:
	+ logger.debug('get_instances_list(): no instances found')
	+ return ''
	+
	for zone in zones.values():
	for inst in zone.get('instances', []):
	if inst['name'] != exclude:
	hostlist.append(inst['name'])
	request = CONN.instances().aggregatedList_next(
	previous_request=request, previous_response=response)
	return hostlist


	def gcp_alias_start(alias):
	my_aliases = get_localhost_aliases()
	my_zone = get_metadata('instance/zone').split('/')[-1]
	project = os.environ.get(
	'OCF_RESKEY_project', get_metadata('project/project-id'))

	if alias in my_aliases:
	# TODO: Do we need to check alias_range_name?
	logger.info(
	'%s already has %s attached. No action required' % (THIS_VM, alias))
	sys.exit(OCF_SUCCESS)

	# If the alias is currently attached to another host, detach it.
	hostlist = os.environ.get('OCF_RESKEY_hostlist', '')
	if hostlist:
	hostlist = hostlist.replace(THIS_VM, '').split()
	else:
	hostlist = get_instances_list(project, THIS_VM)
	for host in hostlist:
	host_zone = get_zone(project, host)
	host_aliases = get_aliases(project, host_zone, host)
	if alias in host_aliases:
	logger.info(
	'%s is attached to %s - Removing %s from %s' %
	(alias, host, alias, host))
	remove_alias(project, host_zone, host, alias)
	break

	# Add alias IP range to localhost
	- add_alias(
	- project, my_zone, THIS_VM, alias,
	- os.environ.get('OCF_RESKEY_alias_range_name'))
	+ try:
	+ add_alias(
	+ project, my_zone, THIS_VM, alias,
	+ os.environ.get('OCF_RESKEY_alias_range_name'))
	+ except googleapiclient.errors.HttpError as e:
	+ if e.resp.status == 404:
	+ sys.exit(OCF_ERR_CONFIGURED)

	# Verify that the IP range has been added
	my_aliases = get_localhost_aliases()
	if alias in my_aliases:
	logger.info('Finished adding %s to %s' % (alias, THIS_VM))
	else:
	if my_aliases:
	logger.error(
	'Failed to add alias IP range %s. %s has alias IP ranges attached but'
	+ ' they don\'t include %s' % (alias, THIS_VM, alias))
	else:
	logger.error(
	'Failed to add IP range %s. %s has no alias IP ranges attached'
	% (alias, THIS_VM))
	sys.exit(OCF_ERR_GENERIC)


	def gcp_alias_stop(alias):
	my_aliases = get_localhost_aliases()
	my_zone = get_metadata('instance/zone').split('/')[-1]
	project = os.environ.get(
	'OCF_RESKEY_project', get_metadata('project/project-id'))

	if alias in my_aliases:
	logger.info('Removing %s from %s' % (alias, THIS_VM))
	remove_alias(project, my_zone, THIS_VM, alias)
	else:
	logger.info(
	'%s is not attached to %s. No action required'
	% (alias, THIS_VM))


	def gcp_alias_status(alias):
	my_aliases = get_localhost_aliases()
	if alias in my_aliases:
	logger.info('%s has the correct IP range attached' % THIS_VM)
	else:
	sys.exit(OCF_NOT_RUNNING)


	def validate():
	global ALIAS
	global CONN
	global THIS_VM

	# Populate global vars
	try:
	CONN = googleapiclient.discovery.build('compute', 'v1',
	cache_discovery=False)
	except Exception as e:
	logger.error('Couldn\'t connect with google api: ' + str(e))
	sys.exit(OCF_ERR_CONFIGURED)

	try:
	THIS_VM = get_metadata('instance/name')
	except Exception as e:
	logger.error('Couldn\'t get instance name, is this running inside GCE?: '
	+ str(e))
	sys.exit(OCF_ERR_CONFIGURED)

	ALIAS = os.environ.get('OCF_RESKEY_alias_ip')
	if not ALIAS:
	logger.error('Missing alias_ip parameter')
	sys.exit(OCF_ERR_CONFIGURED)


	def configure_logs():
	# Prepare logging
	global logger
	logging.getLogger('googleapiclient').setLevel(logging.WARN)
	logging_env = os.environ.get('OCF_RESKEY_stackdriver_logging')
	if logging_env:
	logging_env = logging_env.lower()
	if any(x in logging_env for x in ['yes', 'true', 'enabled']):
	try:
	import google.cloud.logging.handlers
	client = google.cloud.logging.Client()
	handler = google.cloud.logging.handlers.CloudLoggingHandler(
	client, name=THIS_VM)
	handler.setLevel(logging.INFO)
	formatter = logging.Formatter('gcp:alias "%(message)s"')
	handler.setFormatter(formatter)
	log.addHandler(handler)
	logger = logging.LoggerAdapter(log, {'OCF_RESOURCE_INSTANCE':
	OCF_RESOURCE_INSTANCE})
	except ImportError:
	logger.error('Couldn\'t import google.cloud.logging, '
	'disabling Stackdriver-logging support')


	def main():
	if 'meta-data' in sys.argv[1]:
	print(METADATA)
	return

	validate()
	if 'validate-all' in sys.argv[1]:
	return

	configure_logs()
	if 'start' in sys.argv[1]:
	gcp_alias_start(ALIAS)
	elif 'stop' in sys.argv[1]:
	gcp_alias_stop(ALIAS)
	elif 'status' in sys.argv[1] or 'monitor' in sys.argv[1]:
	gcp_alias_status(ALIAS)
	else:
	logger.error('no such function %s' % str(sys.argv[1]))


	if __name__ == "__main__":
	main()

File Metadata

Mime Type: text/x-diff
Expires: Wed, Feb 26, 4:22 PM (2 h, 39 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1465552
Default Alt Text: (30 KB)

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions