[ViewVC] Diff of: proj/trunk/Spectre/Spectre.cpp

Comparing trunk/Spectre/Spectre.cpp (file contents):
Revision 148 by douglas, 2003-06-08T16:55:49-07:00 vs.
Revision 176 by douglas, 2003-07-05T17:36:35-07:00

-+
+// Spectre Samba Mounter
-+
+//
-+
+// Douglas Thrift
-+
+//
-+
+// Spectre.cpp
-+
-+
+#include "Spectre.h"
-+
+#include "Maker.h"
-+
+#include "Mounter.h"
-+
-+
+string program;
-+
+string programName = "Spectre Samba Mounter";
-+
+string programVersion = "1.0beta";
-+
+bool debug = false;
-+
-+
+Config config;
-+
-+
+int spectre(vector<string>& args);
-+
+void automake(set<string>& makes);
-+
+void automake(set<string>& makes, ipstream& pin);
-+
+void automount(set<string>& mounts);
-+
+void configure();
-+
-+
+int main(int argc, char* argv[])
-+
+{
-+
+        vector<string> args(argc);
-+
-+
+        for (int index = 0; index < argc; index++)
-+
+        {
-+
+                args[index] = argv[index];
-+
+        }
-+
-+
+        return spectre(args);
-+
+}
-+
-+
+int spectre(vector<string>& args)
-+
+{
-+
+        program = args[0];
-+
-+
+#include "configure.h"
-+
-+
+        bool make = false;
-+
+        bool mount = false;
-+
+        bool automake = false;
-+
+        bool automount = false;
-+
-+
+        set<string> makes;
-+
+        set<string> mounts;
-+
-+
+        for (int index = 1; index < args.size(); index++)
-+
+        {
-+
+                if (args[index] == "-help")
-+
+                {
-+
+                        usage();
-+
+                        return 0;
-+
+                }
-+
+                else if (args[index] == "-version")
-+
+                {
-+
+                        version();
-+
+                        return 0;
-+
+                }
-+
+                else if (args[index] == "-make")
-+
+                {
-+
+                        if (!make) make = true;
-+
-+
+                        if (++index < args.size())
-+
+                        {
-+
+                                makes.insert(args[index]);
-+
+                        }
-+
+                        else
-+
+                        {
-+
+                                cerr << program << ": Bad arguments\n";
-+
+                                usage();
-+
+                                return 1;
-+
+                        }
-+
+                }
-+
+                else if (args[index] == "-mount")
-+
+                {
-+
+                        if (!mount) mount = true;
-+
-+
+                        if (++index < args.size())
-+
+                        {
-+
+                                mounts.insert(args[index]);
-+
+                        }
-+
+                        else
-+
+                        {
-+
+                                cerr << program << ": Bad arguments\n";
-+
+                                usage();
-+
+                                return 1;
-+
+                        }
-+
+                }
-+
+                else if (args[index] == "-automake")
-+
+                {
-+
+                        if (!automake) automake = true;
-+
+                }
-+
+                else if (args[index] == "-automount")
-+
+                {
-+
+                        if (!automount) automount = true;
-+
+                }
-+
+                else if (args[index] == "-D")
-+
+                {
-+
+                        if (!debug)
-+
+                        {
-+
+                                debug = true;
-+
+                                cerr.setf(ios_base::boolalpha);
-+
+                        }
-+
+                }
-+
+        }
-+
-+
+        if (!make && !mount && !automake && !automount)
-+
+        {
-+
+                usage();
-+
+                return 0;
-+
+        }
-+
-+
+        if (debug)
-+
+        {
-+
+                cerr << "make = " << make << "\n"
-+
+                        << (make ? "makes = {\n" : "");
-+
-+
+                for (set<string>::iterator itor = makes.begin(); itor != makes.end();
-+
+                        itor++)
-+
+                {
-+
+                        cerr << "   " << *itor << "\n";
-+
+                }
-+
-+
+                cerr << (make ? "}\n" : "")
-+
+                        << "mount = " << mount << "\n"
-+
+                        << (mount ? "mounts = {\n" : "");
-+
-+
+                for (set<string>::iterator itor = mounts.begin(); itor != mounts.end();
-+
+                        itor++)
-+
+                {
-+
+                        cerr << "   " << *itor << "\n";
-+
+                }
-+
-+
+                cerr << (mount ? "}\n" : "")
-+
+                        << "automake = " << automake << "\n"
-+
+                        << "automount = " << automount << "\n"
-+
+                        << "config.install = " << config.install << "\n";
-+
+        }
-+
-+
+        configure();
-+
-+
+        if (debug)
-+
+        {
-+
+                cerr << "config.findsmb = " << config.findsmb << "\n"
-+
+                        << "config.smbclient = " << config.smbclient << "\n"
-+
+                        << "config.mount = " << config.mount << "\n"
-+
+                        << "config.root = " << config.root << "\n"
-+
+                        << (!config.hosts.empty() ? "config.hosts = {\n" : "");
-+
-+
+                for (multimap<string, string>::iterator itor = config.hosts.begin();
-+
+                        itor != config.hosts.end(); itor++)
-+
+                {
-+
+                        cerr << "   " << itor->first << " = " << itor->second << "\n";
-+
+                }
-+
-+
+                cerr << (!config.hosts.empty() ? "}\n" : "");
-+
+        }
-+
-+
+        if (automake) ::automake(makes);
-+
-+
+        if (make || automake)
-+
+        {
-+
+                for (set<string>::iterator itor = makes.begin(); itor != makes.end();
-+
+                        itor++)
-+
+                {
-+
+                        Maker maker(*itor);
-+
-+
+                        maker.make();
-+
+                }
-+
+        }
-+
-+
+        if (automount) ::automount(mounts);
-+
-+
+        if (mount || automount)
-+
+        {
-+
+                for (set<string>::iterator itor = mounts.begin(); itor != mounts.end();
-+
+                        itor++)
-+
+                {
-+
+                        Mounter mounter(*itor);
-+
-+
+                        mounter.mount();
-+
+                }
-+
+        }
-+
-+
+        return 0;
-+
+}
-+
-+
+string platform()
-+
+{
-+
+        utsname* computer = new utsname;
-+
+        uname(computer);
-+
-+
+        string os = computer->sysname;
-+
+        string version = computer->release;
-+
+        string architecture = computer->machine;
-+
-+
+        delete computer;
-+
-+
+        string platform = "(" + os + " " + version + " " + architecture + ")";
-+
-+
+        return platform;
-+
+}
-+
-+
+void usage()
-+
+{
-+
+        string tab(8 + program.length(), ' ');
-+
-+
+        cout << "Usage: " << program << " [-make host ...] [-mount host ...]\n"
-+
+                << tab << "[-automake] [-automount]\n"
-+
+                << tab << "[-D] [-version] [-help]\n"
-+
+                << "Options:\n"
-+
+                << "  -make host        Make the mount tree for host\n"
-+
+                << "  -mount host       Mount the shares on host to its tree\n"
-+
+                << "  -automake         Automagically make the mount tree\n"
-+
+                << "  -automount        Automagically mount shares to the tree\n"
-+
+                << "  -D                Display debug information\n"
-+
+                << "  -version          Display version information and exit\n"
-+
+                << "  -help             Display this message and exit\n";
-+
+}
-+
-+
+void version()
-+
+{
-+
+        cout << programName << " " << programVersion << " "<< platform() << "\n\n"
-+
+                << "   Copyright (C) 2003, Douglas Thrift. All Rights Reserved.\n\n"
-+
+                << "   This product includes software developed by Douglas Thrift\n"
-+
+                << "   (http://computers.douglasthrift.net/).\n";
-+
+}
-+
-+
+void automake(set<string>& makes)
-+
+{
-+
+        vector<string> args;
-+
-+
+        args.push_back("spectre");
-+
-+
+        ipstream findsmb(config.findsmb, args);
-+
-+
+        if (debug) cerr << "findsmb = {\n";
-+
-+
+        for (unsigned index = 0; index < 5; index++)
-+
+        {
-+
+                string line;
-+
-+
+                getline(findsmb, line);
-+
-+
+                if (debug) cerr << line << "\n";
-+
+        }
-+
-+
+        automake(makes, findsmb);
-+
-+
+        if (debug) cerr << "}\n";
-+
-+
+        findsmb.close();
-+
-+
+        if (debug)
-+
+        {
-+
+                cerr << "makes = {\n";
-+
-+
+                for (set<string>::iterator itor = makes.begin(); itor != makes.end();
-+
+                        itor++)
-+
+                {
-+
+                        cerr << "   " << *itor << "\n";
-+
+                }
-+
-+
+                cerr << "}\n";
-+
+        }
-+
+}
-+
-+
+void automake(set<string>& makes, ipstream& pin)
-+
+{
-+
+        while (pin.good())
-+
+        {
-+
+                if (!isdigit(pin.peek())) break;
-+
-+
+                char ip[17];
-+
+                char host[16];
-+
+                string info;
-+
-+
+                pin.get(ip, 17);
-+
+                pin.ignore();
-+
+                pin.get(host, 16);
-+
+                getline(pin, info);
-+
-+
+                if (debug) cerr << ip << "\t" << host << info << "\n";
-+
-+
+                strip(host);
-+
-+
+                makes.insert(tolower(host));
-+
+        }
-+
-+
+        string line;
-+
-+
+        getline(pin, line);
-+
-+
+        if (debug) cerr << line << "\n";
-+
-+
+        if (line != "")
-+
+        {
-+
+                cerr << program << ": Unknown error\n";
-+
+        }
-+
+}
-+
-+
+void automount(set<string>& mounts)
-+
+{
-+
+}
-+
-+
+void configure()
-+
+{
-+
+        string conf = config.install + "/conf/spectre.conf";
-+
-+
+        ifstream fin(conf.c_str());
-+
-+
+        if (!fin.is_open())
-+
+        {
-+
+                cerr << program << ": Could not open " << conf << "\n";
-+
+                exit(1);
-+
+        }
-+
-+
+        do
-+
+        {
-+
+                string line;
-+
-+
+                getline(fin, line);
-+
-+
+                if (line.find('#') == 0 || line == "")
-+
+                {
-+
+                        // ignore
-+
+                }
-+
+                else if (line.find("findsmb=") == 0)
-+
+                {
-+
+                        config.findsmb = line.substr(8);
-+
+                }
-+
+                else if (line.find("smbclient=") == 0)
-+
+                {
-+
+                        config.smbclient = line.substr(10);
-+
+                }
-+
+                else if (line.find("mount=") == 0)
-+
+                {
-+
+                        config.mount = line.substr(6);
-+
+                }
-+
+                else if (line.find("root=") == 0)
-+
+                {
-+
+                        config.root = line.substr(5);
-+
+                }
-+
+                else if (line.find("host=") == 0)
-+
+                {
-+
+                        string host = line.substr(5);
-+
-+
+                        getline(fin, line);
-+
-+
+                        if (line != "{")
-+
+                        {
-+
+                                cerr << program << ": Munged configuration for " << host
-+
+                                   << "\n";
-+
+                                exit(1);
-+
+                        }
-+
-+
+                        do
-+
+                        {
-+
+                                getline(fin, line);
-+
-+
+                                if (line.find("\t#") == 0)
-+
+                                {
-+
+                                        // ignore
-+
+                                }
-+
+                                else if (line.find('\t') == 0)
-+
+                                {
-+
+                                        config.hosts.insert(pair<string, string>(host,
-+
+                                                line.substr(1)));
-+
+                                }
-+
+                                else if (line == "}")
-+
+                                {
-+
+                                        break;
-+
+                                }
-+
+                        }
-+
+                        while (fin.good());
-+
+                }
-+
+        }
-+
+        while (fin.good());
-+
-+
+        fin.close();
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing trunk/Spectre/Spectre.cpp (file contents): Revision 148 by douglas, 2003-06-08T16:55:49-07:00 vs. Revision 176 by douglas, 2003-07-05T17:36:35-07:00

Diff Legend

Comparing trunk/Spectre/Spectre.cpp (file contents):
Revision 148 by douglas, 2003-06-08T16:55:49-07:00 vs.
Revision 176 by douglas, 2003-07-05T17:36:35-07:00