Python-OpenFlow-RyuController.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<div class=\"alert alert-info\">\n",
    "\n",
    "**Note:**    WARNING: This notebook uses a new, early release Chameleon python API. The notebook and API will likely be updated without notice. To request a new python feature, please create a help desk ticket. \n",
    "    \n",
    "\n",
    "</div>\n",
    "\n",
    "# Create a Ryu OpenFlow Controller"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import json\n",
    "import os\n",
    "import chi\n",
    "\n",
    "from chi.server import *\n",
    "from chi.lease import *\n",
    "from chi.network import *\n",
    "\n",
    "#Config with your project and site\n",
    "chi.set('project_name', 'CH-822154') # Replace with your project name\n",
    "#chi.set('project_name', 'FABRIC@Chameleon') # Replace with your project name\n",
    "chi.set('region_name', 'CHI@UC')     # Optional, defaults to 'CHI@UC'\n",
    "chi.set('auth_url','https://chi.uc.chameleoncloud.org:5000/v3')\n",
    "\n",
    "#Insert keypair name\n",
    "key_name = 'my_chameleon_key'  # Change to your keypair\n",
    "\n",
    "# Tip: Name resources with your username for easier identification\n",
    "username = os.getenv(\"USER\")\n",
    "server_name = username+'RyuController'\n",
    "lease_name = username+'RyuControllerLease'\n",
    "\n",
    "#Server attributes\n",
    "image_name='CC-CentOS7'\n",
    "flavor_name='baremetal'\n",
    "node_type=\"compute_haswell\"\n",
    "server_count=1\n",
    "network_name='sharednet1'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#import datetime\n",
    "#from datetime import timedelta\n",
    "#import time\n",
    "from datetime import datetime, timedelta\n",
    "from dateutil import tz\n",
    "from chi.util import get_public_network\n",
    "\n",
    "BLAZAR_TIME_FORMAT = \"%Y-%m-%d %H:%M\"\n",
    "\n",
    "# Set start/end date for lease\n",
    "# Start one minute into future to avoid Blazar thinking lease is in past\n",
    "# due to rounding to closest minute.\n",
    "start_date = (datetime.now(tz=tz.tzutc()) + timedelta(minutes=1)).strftime(BLAZAR_TIME_FORMAT)\n",
    "end_date   = (datetime.now(tz=tz.tzutc()) + timedelta(days=1)).strftime(BLAZAR_TIME_FORMAT)\n",
    "\n",
    "# Build list of reservations (in this case there is only one reservation)\n",
    "reservation_list = []\n",
    "add_node_reservation(reservation_list, count=server_count, node_type=node_type)\n",
    "add_fip_reservation(reservation_list, count=server_count)\n",
    "\n",
    "# Create the lease\n",
    "chi.blazar().lease.create(name=lease_name, \n",
    "                            start=start_date,\n",
    "                            end=end_date,\n",
    "                            reservations=reservation_list, events=[])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#Get the lease by name\n",
    "lease = get_lease_by_name(lease_name)\n",
    "    \n",
    "#Print the lease info\n",
    "print(json.dumps(lease, indent=2))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#Get the lease by name\n",
    "lease = get_lease_by_name(lease_name)\n",
    "\n",
    "compute_reservation_id = list(filter(lambda reservation: reservation['resource_type'] == 'physical:host', lease['reservations']))[0]['id']\n",
    "floatingip_reservation_id = list(filter(lambda reservation: reservation['resource_type'] == 'virtual:floatingip', lease['reservations']))[0]['id']\n",
    "\n",
    "print(\"compute_reservation_id: \" + compute_reservation_id)\n",
    "print(\"floatingip_reservation_id: \" + floatingip_reservation_id)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#create the server\n",
    "server = create_server(server_name, \n",
    "                       reservation_id=compute_reservation_id, \n",
    "                       key_name=key_name, \n",
    "                       network_name=network_name, \n",
    "                       image_name=image_name, \n",
    "                       flavor_name=flavor_name)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "floating_ip = associate_floating_ip(server_name)\n",
    "\n",
    "print('Floating IP: ' + str(floating_ip))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "ryu_app='simple_switch_13_custom_chameleon.py'\n",
    "ryu_port='6653'\n",
    "mirror_port=''\n",
    "\n",
    "script= '#!/bin/bash'   '\\n' \\\n",
    "    '{'   '\\n' \\\n",
    "    'yum install -y yum-utils device-mapper-persistent-data lvm2 vim'   '\\n' \\\n",
    "    'yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo'   '\\n' \\\n",
    "    'yum install -y docker-ce'   '\\n' \\\n",
    "    'systemctl start docker'   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    r'RECIPE_REPO=\\\"https://github.com/RENCI-NRIG/exogeni-recipes.git\\\"'   '\\n' \\\n",
    "    'RECIPE_DIR=\\\"/opt/exogeni-recipes\\\"'   '\\n' \\\n",
    "    'RECIPE_APP=\\\"openflow-controller/docker\\\"'   '\\n' \\\n",
    "    'DOCKER_IMAGE=\\\"centos-ryu\\\"'   '\\n' \\\n",
    "    'DOCKER_CONTAINER_NAME=\\\"ryu-controller\\\"'   '\\n' \\\n",
    "    'OFP_TCP_LISTEN_PORT=\\\"'+ryu_port+'\\\"'   '\\n' \\\n",
    "    'RYU_APP=\\\"/opt/ryu_app/'+ryu_app+'\\\"'   '\\n' \\\n",
    "    'MIRROR_PORT='+mirror_port+   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    'git clone  --no-checkout \\${RECIPE_REPO} \\${RECIPE_DIR}'   '\\n' \\\n",
    "    'cd \\${RECIPE_DIR} && git config core.sparsecheckout true'   '\\n' \\\n",
    "    'echo \\\"\\${RECIPE_APP}/*\\\" >> .git/info/sparse-checkout'   '\\n' \\\n",
    "    'git read-tree -m -u HEAD'   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    'cd \\${RECIPE_DIR}/\\${RECIPE_APP}'   '\\n' \\\n",
    "    'docker volume create var_run_ryu'   '\\n' \\\n",
    "    'docker volume create var_log_ryu'   '\\n' \\\n",
    "    'docker volume create opt_ryu'   '\\n' \\\n",
    "    'docker volume create opt_ryu_chameleon'   '\\n' \\\n",
    "    \"sed -r -i \\'s/^(RYU_APP=.*)/#\\1/g\\' ryu_start.sh\"   '\\n' \\\n",
    "    \"sed -r -i \\'s/^(OFP_TCP_LISTEN_PORT=.*)/#\\1/g\\' ryu_start.sh\"   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    'docker build -t \\${DOCKER_IMAGE} .'   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    'docker run --rm -dit -p \\${OFP_TCP_LISTEN_PORT}:\\${OFP_TCP_LISTEN_PORT} -p 8080:8080 -v opt_ryu_chameleon:/opt/ryu_chameleon -v opt_ryu:/opt/ryu -v var_log_ryu:/var/log/ryu -v var_run_ryu:/var/run/ryu -e RYU_APP=\\${RYU_APP} -e OFP_TCP_LISTEN_PORT=\\${OFP_TCP_LISTEN_PORT}  --name=\\${DOCKER_CONTAINER_NAME} \\${DOCKER_IMAGE}'   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    'echo done'   '\\n' \\\n",
    "    '   \\n' \\\n",
    "    '} > /tmp/boot.log 2>&1'   '\\n' "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import paramiko \n",
    "#from scp import SCPClient, SCPException\n",
    "\n",
    "key = paramiko.RSAKey.from_private_key_file(\"/home/pruth/work/my_chameleon_key\")\n",
    "client = paramiko.SSHClient()\n",
    "client.load_system_host_keys()\n",
    "client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())\n",
    "\n",
    "client.set_missing_host_key_policy(paramiko.AutoAddPolicy())\n",
    "client.connect(floating_ip,username='cc',pkey = key)\n",
    "\n",
    "stdin, stdout, stderr = client.exec_command('echo \\\"' + script + '\\\" > script.sh; chmod +x script.sh; sudo ./script.sh')\n",
    "print (stdout.read())\n",
    "print (stderr.read())\n",
    "\n",
    "client.close()\n",
    "print ('done!')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "delete_server_by_name(server_name)\n",
    "delete_lease_by_name(lease_name)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}