# HG changeset patch
# User igor
# Date 1184335053 -10800
# Node ID ab37f0ad70e58cda760d31638f623853d8efcc30

Initial revision

diff -r 000000000000 -r ab37f0ad70e5 build-xenomips-image
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/build-xenomips-image	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,123 @@
+#!/bin/sh
+
+DYNAMIPS_BUILD=yes
+XENOMIPS_FS_BUILD=yes
+CONFIG_FS_BUILD=no
+IOS_FS_BUILD=yes
+
+#Xenomips FS
+XENOMIPS_FS_SIZE=2048k                  # size in 1k blocks
+XENOMIPS_FS_NAME=xenomips1.img
+
+# Config FS
+CONFIG_FS_SIZE=100k
+CONFIG_FS_NAME=xenomips-config1.img
+
+# IOS FS
+IOS_FS_SIZE=2048k
+IOS_FS_NAME=xenomips-ios1.img
+
+
+XENOMIPS_DIR="files/"
+
+
+DEBIAN_MIRROR=http://debian.org.ua/debian
+DISTRO=sid
+PACKAGES="openssh-server libc6-xen libelf1 libpcap0.8 screen nfs-common quagga"
+DYNAMIPS_SOURCES=http://www.ipflow.utc.fr/dynamips/dynamips-0.2.7.tar.gz
+DYNAMIPS_BUILD_DIR=dynamips/
+KERNEL_VERSION=`uname -r`
+SSH_OPEN_KEY=files/id_dsa.pub
+
+IOS_DIR="ios/"
+
+WORK_DIR=${XENOMIPS_FS_NAME%%.img}
+
+build_and_install_dynamips()
+{
+    OLD_PWD="`pwd`"
+    DESTDIR=`pwd`/${WORK_DIR}
+
+    if echo $DYNAMIPS_BUILD | grep -qi ^y
+    then
+        rm -rf ${DYNAMIPS_BUILD_DIR}/
+        mkdir -p ${DYNAMIPS_BUILD_DIR}/
+        cd ${DYNAMIPS_BUILD_DIR}/
+        wget $DYNAMIPS_SOURCES
+        tar xfz dynamips*tar*
+        cd dynamips*/
+        perl -p -i -e s@^PCAP_LIB=.*@PCAP_LIB=-lpcap@ Makefile
+        make
+    else
+        cd ${DYNAMIPS_BUILD_DIR}/dynamips*/
+    fi
+    make install DESTDIR=${DESTDIR}
+    cd ${OLD_PWD}
+}
+
+
+dd if=/dev/zero of=${XENOMIPS_FS_NAME} count=${XENOMIPS_FS_SIZE} bs=1024
+mkfs.ext3 -F ${XENOMIPS_FS_NAME}
+mkdir -p ${WORK_DIR}
+mount -o loop ${XENOMIPS_FS_NAME} ${WORK_DIR}
+debootstrap  ${DISTRO} ${WORK_DIR} ${DEBIAN_MIRROR} 
+cat <<SCRIPT | chroot ${WORK_DIR} sh -s
+unset LANG
+apt-get -y --force-yes install ${PACKAGES}
+/etc/init.d/ssh stop
+cat <<FSTAB > /etc/fstab
+/dev/hda1       /                       ext3    ro              0       0
+/dev/hda2       /xenomips/ios           ext3    ro              0       0
+/dev/hda3       /xenomips/config        ext3    rw              0       0
+FSTAB
+mkdir -p /data/Cisco_IOS /xenomips
+
+cat <<'INTERFACES' > /etc/network/interfaces
+auto lo eth0
+
+iface lo inet loopback 
+        up for i in \`ifconfig -a | grep 'eth.*Link encap' | awk '{print \$1}'\`; do ifconfig \$i up ; done || true
+
+iface eth0 inet dhcp
+INTERFACES
+
+echo /usr/local/bin/xenomips.sh > /etc/rc.local
+cd /etc/rcS.d; ln -s ../init.d/xenomips-init S35xenomips-init
+rm /etc/rc2.d/S*quagga
+
+perl -i -p -e s@C_PATH=/etc/quagga@C_PATH=/xenomips/etc/quagga@ /etc/init.d/quagga
+
+mkdir -p /xenomips/{ios,config,dynamips,etc/quagga} /var/lib/xenomips /root/.ssh
+
+screen -ls | grep -q Attached && screen -x || screen -ls | grep -q Detached && screen -r
+SCRIPT
+
+cp -R /lib/modules/${KERNEL_VERSION} ${WORK_DIR}/lib/modules
+
+cp ${XENOMIPS_DIR}/xenomips-init ${WORK_DIR}/etc/init.d/
+cp ${XENOMIPS_DIR}/xenomips.sh ${WORK_DIR}/usr/local/bin
+sudo chmod +x  ${WORK_DIR}/usr/local/bin/xenomips.sh ${WORK_DIR}/etc/init.d/xenomips-init
+
+cat ${SSH_OPEN_KEY} > ${WORK_DIR}/root/.ssh/authorized_keys
+chmod 600  ${WORK_DIR}/root/.ssh/authorized_keys
+
+build_and_install_dynamips
+
+umount ${WORK_DIR}
+
+if  echo $CONFIG_FS_BUILD | grep -qi ^y
+then
+    dd if=/dev/zero of=${CONFIG_FS_NAME} bs=1k count=${CONFIG_FS_SIZE}
+    mkfs.ext3 -F ${CONFIG_FS_NAME}
+fi
+
+
+if echo $IOS_FS_BUILD | grep -qi ^y
+then
+    dd if=/dev/zero of=${IOS_FS_NAME} bs=1k count=${CONFIG_FS_SIZE}
+    mkfs.ext3 -F ${IOS_FS_NAME}
+    mount -o loop ${IOS_FS_NAME} ${WORK_DIR}
+    cp ${IOS_DIR}/* ${WORK_DIR}
+    umount ${WORK_DIR}
+fi
+
diff -r 000000000000 -r ab37f0ad70e5 files/id_dsa
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/files/id_dsa	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,12 @@
+-----BEGIN DSA PRIVATE KEY-----
+MIIBvAIBAAKBgQC1BnqCB3rBr6sGdTbd7sgab+mfLxULlyS7G/lKkQcc6B3WnpdT
+j/puo/p3DVsXYQJYgPJlpxtZ462xpPuzfVc3zYlwdu6iC1kqhHq8i4EkKv3+5OAq
+A/5m+haOH4JOcLu8751JsWXn3QqnLpwxuPyJrA9g9MQzVkvwAlUxjplVPwIVAKLf
+ClG5F1P7VSbZz9r7emQx1MurAoGAX65WRoG4lG/4RC/RbbqHV8Cgjw7iJeX6alLK
+VL7sTpPUZj0GsAjqvO9VEK5o7No/OA40uFuCFJg7v5ZkeEzT4RP9RpoVsX7OUm9k
+Y/frldeDVmmFqQeWQ3s7JjjQzFuemojpOWfIcUvWeXPoaksY+c8D8Mjr6pKQPzhQ
+JzgzqPACgYEApwMEoYQac3kj1XUvCreyXZYhBAS3JbS+K6U7zg+jWd2Bt5mKkTbf
+ALkdMAm9i7lheRg4LBgxsezDJkF9qrTybOJIFwYJ3v1H+UOGjQo9XVCcHbuYuQAB
+mzMg25o3nK0AqAl9jZepx0FLiqkY/zOAMOV+u9mBFFent0XKUqfLKCoCFQCG/KmB
+2CCWLMk9OzbSHlpN5QpFqw==
+-----END DSA PRIVATE KEY-----
diff -r 000000000000 -r ab37f0ad70e5 files/id_dsa.pub
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/files/id_dsa.pub	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,1 @@
+ssh-dss AAAAB3NzaC1kc3MAAACBALUGeoIHesGvqwZ1Nt3uyBpv6Z8vFQuXJLsb+UqRBxzoHdael1OP+m6j+ncNWxdhAliA8mWnG1njrbGk+7N9VzfNiXB27qILWSqEeryLgSQq/f7k4CoD/mb6Fo4fgk5wu7zvnUmxZefdCqcunDG4/ImsD2D0xDNWS/ACVTGOmVU/AAAAFQCi3wpRuRdT+1Um2c/a+3pkMdTLqwAAAIBfrlZGgbiUb/hEL9FtuodXwKCPDuIl5fpqUspUvuxOk9RmPQawCOq871UQrmjs2j84DjS4W4IUmDu/lmR4TNPhE/1GmhWxfs5Sb2Rj9+uV14NWaYWpB5ZDezsmONDMW56aiOk5Z8hxS9Z5c+hqSxj5zwPwyOvqkpA/OFAnODOo8AAAAIEApwMEoYQac3kj1XUvCreyXZYhBAS3JbS+K6U7zg+jWd2Bt5mKkTbfALkdMAm9i7lheRg4LBgxsezDJkF9qrTybOJIFwYJ3v1H+UOGjQo9XVCcHbuYuQABmzMg25o3nK0AqAl9jZepx0FLiqkY/zOAMOV+u9mBFFent0XKUqfLKCo= igor@cholpon
diff -r 000000000000 -r ab37f0ad70e5 files/xenomips-init
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/files/xenomips-init	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,47 @@
+#!/bin/sh
+
+bind_mounts () {
+  # set defaults
+  test -z "$tmpfs_dir" && tmpfs_dir=/var/lib/xenomips
+  test -z "$rw_dirs" \
+  && rw_dirs="/var/cache/man /var/lock /var/run /var/log /var/spool /var/tmp /tmp /var/lib/urandom /var/lib/dhcp3 /tmp"
+  test -z "$copy_dirs" && copy_dirs=""
+  test -z "$temp_copy_dirs" && temp_copy_dirs="/var/cache/debconf"
+  test -z "$bindfiles" && bindfiles=""
+  mount -t tmpfs -o mode=0755 tmpfs $tmpfs_dir
+  # preserve directory structure
+  for d in $rw_dirs ; do
+    if [ -d "$d" ]; then
+      cd $tmpfs_dir
+      tar --no-recursion -cpf - $(find $d -type d 2> /dev/null) 2> /dev/null | tar xpf -
+      mount --bind $tmpfs_dir/$d $d
+    else
+      echo "WARNING: $d does not exist"
+    fi
+  done  
+  # copy contents into tmpfs
+  for d in $copy_dirs $temp_copy_dirs; do
+    if [ -d "$d" ]; then
+      cd $tmpfs_dir
+      tar -cpf - $d 2> /dev/null | tar xpf -
+      mount --bind $tmpfs_dir/$d $d
+    else
+      echo "WARNING: $d does not exist"
+    fi
+  done
+  # mount one file on top of another
+  for f in $bindfiles ; do
+    if [ -e "$f" ]; then
+      mkdir -p "$(dirname $tmpfs_dir/$f)"
+      cp $f $tmpfs_dir/$f
+      mount --bind $tmpfs_dir/$f $f
+    else
+      echo "WARNING: $f does not exist"
+    fi
+  done
+  touch /var/log/dmesg
+}
+
+
+bind_mounts
+
diff -r 000000000000 -r ab37f0ad70e5 files/xenomips.sh
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/files/xenomips.sh	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,9 @@
+#!/bin/sh
+
+DYNAMIPS_WORKDIR="/xenomips/config"
+DYNAMIPS_ARGS=`cat /proc/cmdline | sed 's/.*xenomips="//; s/".*//'`
+
+
+cd "$DYNAMIPS_WORKDIR"
+screen -e '^``' -d -m dynamips $DYNAMIPS_ARGS
+
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/bridge.png
Binary file shapes/all/bridge.png has changed
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/quagga.png
Binary file shapes/all/quagga.png has changed
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/router.png
Binary file shapes/all/router.png has changed
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/switch.png
Binary file shapes/all/switch.png has changed
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/xenomips.png
Binary file shapes/all/xenomips.png has changed
diff -r 000000000000 -r ab37f0ad70e5 shapes/all/zebralogo.jpg
Binary file shapes/all/zebralogo.jpg has changed
diff -r 000000000000 -r ab37f0ad70e5 xenomips_vars.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/xenomips_vars.py	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,4 @@
+
+N=1
+
+
diff -r 000000000000 -r ab37f0ad70e5 xentaur.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/xentaur.py	Fri Jul 13 16:57:33 2007 +0300
@@ -0,0 +1,211 @@
+#!/usr/bin/python
+
+
+import sys
+import xenomips_vars
+xenomips_vars.N='1'
+sys.path.append('/etc/xen')
+from xenomipsN import vbridges_table, hidden_bridges, domains, broken_links, temporary_links, domain_types
+from IPython.Shell import IPShellEmbed
+
+def create_bridges_script():
+    unbound_bridges=bridges
+    create_unbound_bridges="\n".join(map(lambda x: "brctl show | awk '{print $1}' | grep -q "+x+" || brctl addbr "+x, unbound_bridges))
+    create_unbound_bridges+="\n"+"\n".join(map(lambda x: "ip link set "+x+" up", unbound_bridges))
+
+    print """#!/bin/sh
+# create unbound bridges
+%(create_unbound_bridges)s
+""" % {'create_unbound_bridges' : create_unbound_bridges}
+
+
+def create_domains_script():
+    for N in range(len(domains)):
+        print "xm create xenomipsN N="+str(N)+" && sleep 1 && xm sched-credit -d $(xm list | grep "+domains[N]+" | awk '{print $2}') -c 10 && sleep 1"
+
+def destroy_domains_script():
+    for N in range(len(domains)):
+        print "xm shutdown "+domains[N]
+
+def create_screens_script():
+
+    N=0
+    screens=[]
+    for domain in domains:
+        ip="192.168.80."+str(200+N)
+        screens.append("screen -t "+domain+"   "+str(N)+" sh -c 'while true; do ssh root@"+ip+" ; done'")
+        N+=1
+    screenlist="\n".join(screens)
+
+    print """
+cat <<SCREENRC > screenrc
+screen -t console 0 bash
+%(screenlist)s
+SCREENRC
+""" % { 'screenlist' : screenlist }
+
+def graph_node(node):
+    i=0
+    domain_type={}
+    for domain in domains:
+        domain_type[domain]=domain_types[i]
+        i+=1
+    return node+" [label=\"  "+node+"\",shapefile=\"shapes/all/"+domain_type[node]+".png\",fontcolor=navy,fontsize=14]"
+
+def graph():
+    
+    nodelist=""
+    bridgelist=""
+    linklist=""
+    physicallist=""
+    networklist=""
+
+    nodelist=";\n    ".join(map(graph_node,nodes))
+    if nodelist: nodelist += ";"
+
+    bridgelist=";\n    ".join(bridges-set(hidden_bridges))
+    if bridgelist: bridgelist += ";"
+
+    links=[]
+    for host, bridges_raw in vbridges_table.iteritems():
+        i=0
+        for this_bridge in bridges_raw:
+            if this_bridge in hidden_bridges:
+                continue 
+            if [ host, this_bridge ] in broken_links:
+                links.append(host+" -- "+this_bridge+" [taillabel=\"fa"+str(i)+"/0\",style=dashed]")
+            else:
+                links.append(host+" -- "+this_bridge+" [taillabel=\"fa"+str(i)+"/0\"]")
+            i+=1
+
+    for link in temporary_links:
+        links.append(link[0]+" -- "+link[1]+" [color=blue,len=10,w=5,weight=5]")
+
+    linklist=";\n    ".join(links)
+
+    graph_dot = {
+            'nodelist'      : nodelist,
+            'bridgelist'    : bridgelist,
+            'linklist'      : linklist,
+            'physicallist'  : physicallist,
+            'networklist'   : networklist,
+          }
+
+    print """
+    cat <<'GRAPH' > xenomips.dot
+graph G {
+    edge [len=1.25];
+    splines=true;
+// nodes
+
+    node [shape=plaintext,color=white,shapefile="shapes/cisco.bmp/router.png"];
+    %(nodelist)s
+
+// bridges
+
+    node [shape=none,shapefile="shapes/cisco.bmp/switch.png"];
+    %(bridgelist)s
+
+// physical
+
+    node [shape=rectangle,color=blue];
+    %(physicallist)s
+
+// networks (not bridges, not physical)
+    node [shape=rectangle,color=green];
+    %(networklist)s
+
+// links (between nodes and bridges)
+    %(linklist)s
+
+};
+GRAPH
+    neato -Tpng -o xenomips.png xenomips.dot 
+""" % graph_dot
+
+def start_all():
+        create_bridges_script()
+        create_screens_script()
+        create_domains_script()
+        graph()
+        print """
+        cat <<NOTE_FOR_USER
+# To view virtual network map, run:
+    gqview xenomips.png
+# To attach to VM consoles, run:
+    screen -c screenrc
+NOTE_FOR_USER
+"""
+
+def shell():
+    ipshell = IPShellEmbed()
+    ipshell()
+
+def stop_all():
+        destroy_domains_script(domains)
+
+def show_usage():
+    print """Usage:
+    xentaur {start|stop|start-bridges|start-domains|stop-domains|screen|graph}
+"""
+
+#-----------------------------------------------------------------------
+
+def run(program, *args):
+    pid = os.fork()
+    if not pid:
+        os.execvp(program, (program,) +  args)
+    return os.wait()[0]
+
+def run_command(line)
+    cmds=line.split()
+    run(cmds[0],cmds[1:])
+
+def add_domain(name,type):
+    domains.append(name)
+    domain_types.append(type)
+
+def brake_link(domain,bridge):
+    broken_links.append([domain,bridge])
+
+def type_to(screen,string)
+    run_command("screen -X readreg p /tmp/xentaurbuf")
+    run_command("screen -X select 1 ")
+    run_command("screen -X paste p ")
+    run_command("sleep 1")
+    run_command("screen -X select 0")
+
+#-----------------------------------------------------------------------
+
+bridges=[]
+for domain in vbridges_table.keys():
+    bridges += set(vbridges_table[domain])
+bridges=set(bridges)
+
+nodes=domains
+
+if len(sys.argv) > 1:
+    if sys.argv[1] == 'start':
+        start_all()
+    if sys.argv[1] == 'stop':
+        stop_all()
+    if sys.argv[1] == 'start-bridges':
+        create_bridges_script(bridges)
+    if sys.argv[1] == 'start-domains':
+        create_domains_script(domains)
+    if sys.argv[1] == 'stop-domains':
+        destroy_domains_script(domains)
+    elif sys.argv[1] == 'screen':
+        create_screens_script(domains)
+    elif sys.argv[1] == 'graph':
+        graph()
+    elif sys.argv[1] == 'shell':
+        shell()
+
+else:
+    show_usage()
+    sys.exit(1)
+
+sys.exit(0)
+
+