Implement "bos install"

Signed-off-by: David Howells <dhowells@redhat.com>

diff --git a/suite/commands/bos/install.py b/suite/commands/bos/install.py
new file mode 100644 (file)
index 0000000..e9ada32
--- /dev/null
+++ b/suite/commands/bos/install.py
@@ -0,0 +1,103 @@
+#
+# AFS Server management toolkit: Install new server instances
+# -*- coding: utf-8 -*-
+#
+
+__copyright__ = """
+Copyright (C) 2014 Red Hat, Inc. All Rights Reserved.
+Written by David Howells (dhowells@redhat.com)
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public Licence version 2 as
+published by the Free Software Foundation.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public Licence for more details.
+
+You should have received a copy of the GNU General Public Licence
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+"""
+
+from afs.exception import AFSException
+from afs.argparse import *
+from afs.lib.output import *
+import kafs
+import os
+
+help = "Revert to the former version of a process's binary file"
+
+command_arguments = [
+    [ "server",         get_bosserver,          "rs",         "<machine name>" ],
+    [ "file",           get_file_names,         "rm",         "<files to install>+" ],
+    [ "dir",            get_path_name,          "os",         "<destination dir>" ],
+    [ "cell",           get_cell,               "os",         "<cell name>" ],
+    [ "noauth",         get_auth,               "fn" ],
+    [ "localauth",      get_auth,               "fn" ],
+    [ "verbose",        get_verbose,            "fn" ],
+    [ "encrypt",        get_dummy,              "fn" ],
+]
+
+cant_combine_arguments = [
+    ( "cell",           "localauth" ),
+    ( "noauth",         "localauth" ),
+]
+
+argument_size_limits = {
+    "file"              : kafs.BOZO_BSSIZE,
+    "dir"               : kafs.BOZO_BSSIZE,
+}
+
+description = r"""
+Revert to the former version of a process's binary file
+"""
+
+class split_handler:
+    def __init__(self, name, fd, size):
+        self.__name = fd
+        self.__fd = fd
+        self.__size = size
+        self.__pos = 0
+
+    def client_transmit(self, split_call):
+        while self.__pos < self.__size:
+            verbose("--- XMIT ", self.__pos, "/", self.__size, "\n")
+            size = self.__size - self.__pos
+            if size > 4096:
+                size = 4096
+            self.__pos += size
+
+            buf = self.__fd.read(size)
+            if len(buf) < size:
+                raise IOError("Short read on file " + self.__name)
+            split_call.send(buf, self.__pos < self.__size)
+        return True
+
+    def client_receive(self, split_call):
+        return True
+
+def main(params):
+    cell = params["cell"]
+    bos_conn = cell.open_bos_server(params["server"], params)
+
+    if "dir" in params:
+        d = params["dir"]
+    else:
+        d = "/usr/afs/bin"
+
+    for f in params["file"]:
+        stat = os.stat(f)
+        fd = open(f, "rb")
+        split = split_handler(f, fd, stat.st_size)
+
+        i = f.rfind('/')
+        if i > -1:
+            f = f[i + 1:]
+        remote_file = d + "/" + f
+
+        verbose("Installing file ", remote_file, "\n")
+        ret = kafs.BOZO_Install(bos_conn, remote_file, stat.st_size, 0,
+                                int(stat.st_mtime), split)
+        output(program_name, ": installed file ", f, "\n");