summaryrefslogtreecommitdiffstats
path: root/mcop/mcoputils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'mcop/mcoputils.cpp')
-rw-r--r--mcop/mcoputils.cpp590
1 files changed, 590 insertions, 0 deletions
diff --git a/mcop/mcoputils.cpp b/mcop/mcoputils.cpp
new file mode 100644
index 0000000..52eb78f
--- /dev/null
+++ b/mcop/mcoputils.cpp
@@ -0,0 +1,590 @@
+ /*
+
+ Copyright (C) 2000 Stefan Westerfeld
+ [email protected]
+
+ This library is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Library General Public
+ License as published by the Free Software Foundation; either
+ version 2 of the License, or (at your option) any later version.
+
+ This library 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
+ Library General Public License for more details.
+
+ You should have received a copy of the GNU Library General Public License
+ along with this library; see the file COPYING.LIB. If not, write to
+ the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ Boston, MA 02110-1301, USA.
+
+ */
+
+#include "mcoputils.h"
+#include "mcopconfig.h"
+#include "debug.h"
+#include <pwd.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <netdb.h>
+#include <errno.h>
+#include <config.h>
+#include <ctype.h>
+#include <cstdlib>
+#include <cstring>
+#include <stdio.h>
+#include <map>
+#include <limits.h>
+
+using namespace std;
+using namespace Arts;
+
+// 0: Dir ok
+// 1: Dir not ok
+static int check_tmp_dir(const char *tmp_dir)
+{
+ int result;
+ struct stat stat_buf;
+ result = lstat(tmp_dir, &stat_buf);
+
+if ((result == -1) && (errno == ENOENT))
+ {
+ result = mkdir(tmp_dir, 0700);
+ if (result == -1)
+ {
+ arts_warning("[mcoputils] Error: Cannot create directory \"%s\".\n", tmp_dir);
+ return 1;
+ }
+ result = stat(tmp_dir, &stat_buf);
+ }
+ if ((result == -1) || (!S_ISDIR(stat_buf.st_mode)))
+ {
+ arts_warning("[mcoputils] Error: \"%s\" is not a directory.\n", tmp_dir);
+ return 1;
+ }
+
+ if (stat_buf.st_uid != getuid())
+ {
+ arts_warning("[mcoputils] Error: \"%s\" is owned by uid %d instead of uid %d.\n", tmp_dir, stat_buf.st_uid, getuid());
+ return 1;
+ }
+ return 0;
+}
+
+
+// 0: Link not ok
+// != 0: location of mcop directory
+static char *locate_mcop_dir()
+{
+ struct passwd *pw_ent;
+ string kde_tmp_dir;
+ string user_tmp_dir;
+ int uid = getuid();
+ const char *home_dir = getenv("HOME");
+ const char *kde_home = uid ? getenv("TDEHOME") : getenv("TDEROOTHOME");
+ const char *kde_prefix = "/socket-";
+ const char *tmp;
+ char *tmp_buf;
+ int result;
+ struct stat stat_buf;
+
+ tmp = getenv("TDETMP");
+ if (!tmp || !tmp[0])
+ tmp = getenv("TMPDIR");
+ if (!tmp || !tmp[0])
+ tmp = "/tmp";
+
+ kde_tmp_dir = "";
+
+ pw_ent = getpwuid(uid);
+ if (!pw_ent)
+ {
+ arts_warning("[mcoputils] Error: Can not find password entry for uid %d.\n", getuid());
+ return 0;
+ }
+
+ user_tmp_dir = string(tmp) + "/tdesocket-" + string(pw_ent->pw_name);
+
+ if (!kde_home || !kde_home[0])
+ {
+ kde_home = "~/.trinity/";
+ }
+
+ if (kde_home[0] == '~')
+ {
+ if (uid == 0)
+ {
+ home_dir = pw_ent->pw_dir ? pw_ent->pw_dir : "/root";
+ }
+ if (!home_dir || !home_dir[0])
+ {
+ arts_fatal("Aborting. $HOME not set!");
+ }
+ kde_home++;
+ kde_tmp_dir = string(home_dir);
+ }
+ kde_tmp_dir += kde_home;
+
+ /** Strip trailing '/' **/
+ if ( kde_tmp_dir[kde_tmp_dir.length()-1] == '/')
+ kde_tmp_dir.resize(kde_tmp_dir.length()-1);
+
+ result = stat(kde_tmp_dir.c_str(), &stat_buf);
+ if (result == -1)
+ {
+ return 0;
+ }
+
+ kde_tmp_dir += kde_prefix;
+ {
+ char buf[1024];
+ if (gethostname(buf, sizeof(buf)-1) != 0)
+ {
+ arts_fatal("Aborting. Could not determine hostname or hostname too long.");
+ }
+ buf[sizeof(buf)-1] = '\0';
+ kde_tmp_dir += buf;
+ }
+
+ result = lstat(kde_tmp_dir.c_str(), &stat_buf);
+ if ((result == 0) && (S_ISDIR(stat_buf.st_mode)))
+ {
+ /* $TDEHOME/socket-$HOSTNAME is a normal directory. Do nothing. */
+ tmp_buf = strdup(kde_tmp_dir.c_str());
+ return tmp_buf;
+ }
+
+ if ((result == -1) && (errno == ENOENT))
+ {
+ // Link mising...
+ return 0;
+ }
+ if ((result == -1) || (!S_ISLNK(stat_buf.st_mode)))
+ {
+ arts_warning("[mcoputils] Error: \"%s\" is not a link or a directory.\n", kde_tmp_dir.c_str());
+ return 0;
+ }
+
+ /* kde_tmp_dir is a link. Check whether it points to a valid directory. */
+ ssize_t size = 2048;
+ tmp_buf = NULL;
+ do {
+ size *= 2;
+ tmp_buf = (char *) realloc(tmp_buf, size);
+ if (!tmp_buf)
+ return 0;
+ result = readlink(kde_tmp_dir.c_str(), tmp_buf, size - 1);
+ if (result == -1)
+ {
+ arts_warning("[mcoputils] Error: \"%s\" could not be read.\n", kde_tmp_dir.c_str());
+ free(tmp_buf);
+ return 0;
+ }
+ } while(result == size - 1);
+ tmp_buf[result] = '\0';
+
+// printf("[mcoputils] Link \"%s\" points to \"%s\"\n", kde_tmp_dir.c_str(), tmp_buf);
+ if (strncmp(tmp_buf, user_tmp_dir.c_str(), user_tmp_dir.length()) != 0)
+ {
+ arts_warning("[mcoputils] Error: \"%s\" points to \"%s\" instead of \"%s\".\n", kde_tmp_dir.c_str(), tmp_buf, user_tmp_dir.c_str());
+ free(tmp_buf);
+ return 0;
+ }
+ result = check_tmp_dir(tmp_buf);
+ if (result == 0) return tmp_buf; /* Success */
+
+ free(tmp_buf);
+ return 0;
+}
+
+/* blatant and literal copy from lnusertemp to avoid tdelibs dependency */
+/* Copyright (c) 2000 Waldo Bastian <[email protected]>, released under LGPL */
+
+static
+int create_link(const char *file, const char *tmp_dir)
+{
+ int result;
+ result = check_tmp_dir(tmp_dir);
+ if (result)
+ {
+ return result;
+ }
+ result = symlink(tmp_dir, file);
+ if (result == -1)
+ {
+ fprintf(stderr, "[mcoputils] Error: Can not create link from \"%s\" to \"%s\"\n", file, tmp_dir);
+ return 1;
+ }
+ printf("[mcoputils] Created link from \"%s\" to \"%s\"\n", file, tmp_dir);
+ return 0;
+}
+
+
+static
+int build_link(string tmp_prefix, const char *kde_prefix)
+{
+ struct passwd *pw_ent;
+ string kde_tmp_dir;
+ string user_tmp_dir;
+ char *tmp_buf;
+ int uid = getuid();
+ const char *home_dir = getenv("HOME");
+ const char *kde_home = uid ? getenv("TDEHOME") : getenv("TDEROOTHOME");
+ int result;
+ struct stat stat_buf;
+
+ kde_tmp_dir = "";
+
+ pw_ent = getpwuid(uid);
+ if (!pw_ent)
+ {
+ fprintf(stderr, "[mcoputils] Error: Can not find password entry for uid %d.\n", getuid());
+ return 1;
+ }
+
+ user_tmp_dir = tmp_prefix + string(pw_ent->pw_name);
+
+ if (!kde_home || !kde_home[0])
+ {
+ kde_home = "~/.trinity/";
+ }
+
+ if (kde_home[0] == '~')
+ {
+ if (uid == 0)
+ {
+ home_dir = pw_ent->pw_dir ? pw_ent->pw_dir : "/root";
+ }
+ if (!home_dir || !home_dir[0])
+ {
+ fprintf(stderr, "[mcoputils] Aborting. $HOME not set!");
+ exit(255);
+ }
+ kde_home++;
+ kde_tmp_dir = string(home_dir);
+ }
+ kde_tmp_dir += kde_home;
+
+ /** Strip trailing '/' **/
+ if ( kde_tmp_dir[kde_tmp_dir.length()-1] == '/')
+ kde_tmp_dir.resize(kde_tmp_dir.length()-1);
+
+ result = stat(kde_tmp_dir.c_str(), &stat_buf);
+ if ((result == -1) && (errno == ENOENT))
+ {
+ result = mkdir(kde_tmp_dir.c_str(), 0700);
+ }
+ if (result == -1)
+ {
+ return 1;
+ }
+
+ kde_tmp_dir += kde_prefix;
+ {
+ char buf[1024];
+ if (gethostname(buf, sizeof(buf)-1) != 0)
+ {
+ arts_fatal("Aborting. Could not determine hostname or hostname too long.");
+ }
+ buf[sizeof(buf)-1] = '\0';
+ kde_tmp_dir += buf;
+ }
+
+ result = lstat(kde_tmp_dir.c_str(), &stat_buf);
+ if ((result == 0) && (S_ISDIR(stat_buf.st_mode)))
+ {
+ /* $TDEHOME/tmp is a normal directory. Do nothing. */
+ printf("[mcoputils] Directory \"%s\" already exists.\n", kde_tmp_dir.c_str());
+ return 0;
+ }
+ if ((result == -1) && (errno == ENOENT))
+ {
+ printf("[mcoputils] Creating link %s.\n", kde_tmp_dir.c_str());
+ result = create_link(kde_tmp_dir.c_str(), user_tmp_dir.c_str());
+ if (result == 0) return 0; /* Success */
+ unlink(kde_tmp_dir.c_str());
+ user_tmp_dir += "XXXXXX";
+ tmp_buf = strdup(user_tmp_dir.c_str());
+ if (mkdtemp(tmp_buf) == NULL)
+ return 1;
+ result = create_link(kde_tmp_dir.c_str(), tmp_buf);
+ free(tmp_buf);
+ return result;
+ }
+ if ((result == -1) || (!S_ISLNK(stat_buf.st_mode)))
+ {
+ fprintf(stderr, "[mcoputils] Error: \"%s\" is not a link or a directory.\n", kde_tmp_dir.c_str());
+ return 1;
+ }
+ /* kde_tmp_dir is a link. Check whether it points to a valid directory. */
+ ssize_t size = 2048;
+ tmp_buf = NULL;
+ do {
+ size *= 2;
+ tmp_buf = (char *) realloc(tmp_buf, size);
+ if (!tmp_buf)
+ return 0;
+ result = readlink(kde_tmp_dir.c_str(), tmp_buf, size - 1);
+ if (result == -1)
+ {
+ arts_warning("[mcoputils] Error: \"%s\" could not be read.\n", kde_tmp_dir.c_str());
+ free(tmp_buf);
+ return 0;
+ }
+ } while(result == size - 1);
+ tmp_buf[result] = '\0';
+
+ printf("[mcoputils] Link \"%s\" points to \"%s\"\n", kde_tmp_dir.c_str(), tmp_buf);
+ if (strncmp(tmp_buf, user_tmp_dir.c_str(), user_tmp_dir.length()) != 0)
+ {
+ fprintf(stderr, "[mcoputils] Error: \"%s\" points to \"%s\" instead of \"%s\".\n", kde_tmp_dir.c_str(), tmp_buf, user_tmp_dir.c_str());
+ free(tmp_buf);
+ unlink(kde_tmp_dir.c_str());
+ printf("[mcoputils] Creating link %s.\n", kde_tmp_dir.c_str());
+ result = create_link(kde_tmp_dir.c_str(), user_tmp_dir.c_str());
+ if (result == 0) return 0; /* Success */
+ unlink(kde_tmp_dir.c_str());
+ user_tmp_dir += "XXXXXX";
+ tmp_buf = strdup(user_tmp_dir.c_str());
+ if (mkdtemp(tmp_buf) == NULL)
+ return 1;
+ result = create_link(kde_tmp_dir.c_str(), tmp_buf);
+ free(tmp_buf);
+ return result;
+ }
+ result = check_tmp_dir(tmp_buf);
+ free(tmp_buf);
+ if (result == 0) return 0; /* Success */
+ unlink(kde_tmp_dir.c_str());
+ user_tmp_dir += "XXXXXX";
+ tmp_buf = strdup(user_tmp_dir.c_str());
+ if (mkdtemp(tmp_buf) == NULL)
+ return 1;
+ result = create_link(kde_tmp_dir.c_str(), tmp_buf);
+ free(tmp_buf);
+ return result;
+}
+
+string MCOPUtils::createFilePath(string name)
+{
+ static char *mcop_dir = 0;
+ if (!mcop_dir)
+ {
+ mcop_dir = locate_mcop_dir();
+ }
+ if (!mcop_dir)
+ {
+
+ const char *tmp = 0;
+ tmp = getenv("TDETMP");
+ if (!tmp || !tmp[0])
+ tmp = getenv("TMPDIR");
+ if (!tmp || !tmp[0])
+ tmp = "/tmp";
+
+ build_link(string(tmp) + "/tdesocket-", "/socket-");
+ mcop_dir = locate_mcop_dir();
+ }
+ if (!mcop_dir)
+ {
+ arts_fatal("can't create mcop directory");
+ }
+ string tmpdir = mcop_dir;
+ return tmpdir+"/"+name;
+}
+
+/*
+ * try to figure out full hostname - this is important as every client which
+ * tries to connect objects located here will need to be able to resolve that
+ * correctly
+ */
+string MCOPUtils::getFullHostname()
+{
+ char buffer[1024];
+ string result;
+ struct hostent *hp;
+
+ if(gethostname(buffer,1024) == 0) {
+ buffer[1023] = '\0';
+ result = buffer;
+ } else
+ return "localhost";
+
+ /*
+ * if gethostname() isn't a FQDN (guess that by checking whether it
+ * contains a dot), try to look it up to ensure it is
+ */
+ if(result.find('.') == string::npos && (hp = gethostbyname(buffer)) != 0)
+ result = hp->h_name;
+
+ return result;
+}
+
+string MCOPUtils::readConfigEntry(const string& key, const string& defaultValue)
+{
+ const char *home = getenv("HOME");
+ if(home)
+ {
+ string rcname = home + string("/.mcoprc");
+
+ MCOPConfig config(rcname);
+ return config.readEntry(key,defaultValue);
+ }
+ return defaultValue;
+}
+
+/** IID generation **/
+
+unsigned long MCOPUtils::makeIID(const string& interfaceName)
+{
+ static map<string, unsigned long> *iidmapobj = 0;
+ static unsigned long nextiid = 1;
+
+ if(!iidmapobj) iidmapobj = new map<string,unsigned long>;
+
+ map<string,unsigned long>& iidmap = *iidmapobj;
+
+ if(iidmap.find(interfaceName) == iidmap.end()) {
+ iidmap[interfaceName] = nextiid++;
+ return nextiid-1; // no need to lookup in the map thrice
+ }
+
+ return iidmap[interfaceName];
+}
+
+static vector<string> *readPath(const string& name, const string& defaultValue)
+{
+ vector<string> *result = 0;
+ const char *home = getenv("HOME");
+
+ if(home)
+ {
+ string rcname = home + string("/.mcoprc");
+
+ MCOPConfig config(rcname);
+ result = config.readListEntry(name);
+ }
+ if(!result)
+ result = new vector<string>;
+ if(result->empty())
+ result->push_back(defaultValue);
+
+ return result;
+}
+
+const vector<string> *MCOPUtils::extensionPath()
+{
+ static vector<string> *result = 0;
+
+ if(!result) result = readPath("ExtensionPath", EXTENSION_DIR);
+ return result;
+}
+
+const vector<string> *MCOPUtils::traderPath()
+{
+ static vector<string> *result = 0;
+
+ if(!result)
+ {
+ result = readPath("TraderPath", TRADER_DIR);
+
+ const char *home = getenv("HOME");
+ if(home) result->push_back(home + string("/.mcop/trader-cache"));
+ }
+ return result;
+}
+
+string MCOPUtils::mcopDirectory()
+{
+ static bool initialized = false;
+ static string mcopDirectory;
+
+ if(initialized)
+ return mcopDirectory;
+ initialized = true;
+
+ const char *home = getenv("HOME");
+ arts_return_val_if_fail(home != 0, "");
+
+ mcopDirectory = home + string("/.mcop");
+
+ mkdir(home,0755);
+ if(mkdir(mcopDirectory.c_str(),0755) != 0)
+ {
+ string why = strerror(errno);
+
+ struct stat st;
+ stat(mcopDirectory.c_str(),&st);
+ if(!S_ISDIR(st.st_mode))
+ {
+ arts_warning("[mcoputils] can't create directory %s (%s)",
+ mcopDirectory.c_str(), why.c_str());
+
+ mcopDirectory = "";
+ }
+ }
+ return mcopDirectory;
+}
+
+bool MCOPUtils::tokenize(const string& line, string& key,vector<string>& values)
+{
+ string value;
+ enum { sKey, sValue, sValueQuoted, sValueQuotedEscaped, sBad } state;
+
+ state = sKey;
+ for(string::const_iterator i = line.begin(); i != line.end(); i++)
+ {
+ char c = *i;
+ unsigned char uc = static_cast<unsigned char>(c);
+
+ arts_assert(c != '\n');
+
+ if(state == sKey)
+ {
+ if(c == ' ' || c == '\t')
+ ; // ignore
+ else if(isalnum(c))
+ key += c;
+ else if(c == '=')
+ state = sValue;
+ else
+ state = sBad;
+ }
+ else if(state == sValue)
+ {
+ if(c == ' ' || c == '\t')
+ ; // ignore
+ else if(c == '"')
+ state = sValueQuoted;
+ else if(c == ',')
+ {
+ values.push_back(value);
+ value = "";
+ }
+ else if(uc > 32 && uc < 128)
+ value += c;
+ else
+ state = sBad;
+ }
+ else if(state == sValueQuoted)
+ {
+ if(c == '"')
+ state = sValue;
+ else if(c == '\\')
+ state = sValueQuotedEscaped;
+ else
+ value += c;
+ }
+ else if(state == sValueQuotedEscaped)
+ {
+ value += c;
+ state = sValueQuoted;
+ }
+ }
+ if(state == sValue)
+ values.push_back(value);
+
+ return(state != sBad);
+}
generated by cgit v1.2.1 (git 2.18.0) at 2024-12-17 17:55:27 +0000