+/**
+ * \file lyxclient.C
+ * This file is part of LyX, the document processor.
+ * Licence details can be found in the file COPYING.
+ *
+ * \author João Luis M. Assirati
+ *
+ * Full author contact details are available in file CREDITS.
+ */
+
+#include <string>
+#include <vector>
+#include <map>
+#include <iostream>
+
+
+// getpid(), getppid()
+#include <sys/types.h>
+#include <unistd.h>
+
+// select()
+#include <sys/select.h>
+
+// opendir(), closedir(), readdir()
+#include <sys/types.h>
+#include <dirent.h>
+
+// stat()
+#include <sys/stat.h>
+
+// socket(), connect()
+#include <sys/socket.h>
+#include <sys/un.h>
+
+// fcntl()
+#include <fcntl.h>
+
+
+using std::string;
+using std::vector;
+using std::cout;
+using std::cerr;
+using std::cin;
+using std::endl;
+
+
+namespace support {
+
+string itoa(unsigned int i)
+{
+ string str;
+ if(!i) {
+ str = "0";
+ } else {
+ while((0 < i) && (i <= 9)) {
+ str = static_cast<char>('0' + i % 10) + str;
+ i /= 10;
+ }
+ }
+ return str;
+}
+
+
+bool prefixIs(string const & a, string const & pre)
+{
+ char const * p_a = a.c_str();
+ char const * p_pre = pre.c_str();
+ while ((*p_a != '\0') && (*p_pre != '\0') && (*p_a == *p_pre)) {
+ ++p_a;
+ ++p_pre;
+ }
+ if (*p_pre == '\0') return true;
+ return false;
+}
+
+
+// Parts stolen from lyx::support::DirList()
+// Returns pathnames begining with dir and ending with
+// pathname separator (/ in unix)
+vector<string> lyxSockets(string const & dir, string const & pid)
+{
+ vector<string> dirlist;
+ DIR * dirp = ::opendir(dir.c_str());
+ if (!dirp) {
+ cerr << "lyxclient: Could not read dir " << dir
+ << ": " << strerror(errno);
+ return dirlist;
+ }
+
+ dirent * dire;
+ while ((dire = ::readdir(dirp))) {
+ string const fil = dire->d_name;
+ if (prefixIs(fil, "lyx_tmpdir" + pid)) {
+ string lyxsocket = dir + '/' + fil + "/lyxsocket";
+ struct stat status;
+ // not checking if it is a socket -- just if it exists
+ if (!::stat(lyxsocket.c_str(), &status)) {
+ dirlist.push_back(lyxsocket);
+ }
+ }
+ }
+
+ ::closedir(dirp);
+ return dirlist;
+}
+
+namespace socktools {
+
+int connect(string const & name)
+{
+ int fd; // File descriptor for the socket
+ sockaddr_un addr; // Structure that hold the socket address
+
+ // char sun_path[108]
+ string::size_type len = name.size();
+ if(len > 107) {
+ cerr << "lyxclient: Socket address '" << name
+ << "' too long." << endl;
+ return -1;
+ }
+ // Synonims for AF_UNIX are AF_LOCAL and AF_FILE
+ addr.sun_family = AF_UNIX;
+ name.copy(addr.sun_path, 107);
+ addr.sun_path[len] = '\0';
+
+ if((fd = ::socket(PF_UNIX, SOCK_STREAM, 0))== -1) {
+ cerr << "lyxclient: Could not create socket: "
+ << strerror(errno) << endl;
+ return -1;
+ }
+ if(::connect(fd, (struct sockaddr *)&addr, SUN_LEN(&addr)) == -1) {
+ cerr << "lyxclient: Could not connect to socket " << name
+ << ": " << strerror(errno) << endl;
+ ::close(fd);
+ return -1;
+ }
+ if (::fcntl(fd, F_SETFL, O_NONBLOCK) == -1) {
+ cerr << "lyxclient: Could not set O_NONBLOCK for socket: "
+ << strerror(errno) << endl;
+ ::close(fd);
+ return -1;
+ }
+ return fd;
+}
+
+} // namespace socktools
+} // namespace support
+
+
+
+// Class IOWatch ------------------------------------------------------------
+class IOWatch
+{
+public:
+ IOWatch();
+ void clear();
+ void addfd(int);
+ bool wait(double);
+ bool wait();
+ bool isset(int fd);
+
+private:
+ fd_set des;
+ fd_set act;
+};
+
+IOWatch::IOWatch() {
+ clear();
+}
+void IOWatch::clear() {
+ FD_ZERO(&des);
+}
+void IOWatch::addfd(int fd) {
+ FD_SET(fd, &des);
+}
+bool IOWatch::wait(double timeout) {
+ timeval to;
+ to.tv_sec = static_cast<long int>(timeout);
+ to.tv_usec = static_cast<long int>((timeout - to.tv_sec)*1E6);
+ act = des;
+ return select(FD_SETSIZE, &act,
+ (fd_set *)0, (fd_set *)0, &to);
+}
+bool IOWatch::wait() {
+ act = des;
+ return select(FD_SETSIZE, &act,
+ (fd_set *)0, (fd_set *)0, (timeval *)0);
+}
+bool IOWatch::isset(int fd) {
+ return FD_ISSET(fd, &act);
+}
+// ~Class IOWatch ------------------------------------------------------------
+
+
+// Class LyXDataSocket -------------------------------------------------------
+// Modified LyXDataSocket class for use with the client
+class LyXDataSocket
+{
+public:
+ LyXDataSocket(string const &);
+ ~LyXDataSocket();
+ // File descriptor of the connection
+ int fd() const;
+ // Connection status
+ bool connected() const;
+ // Line buffered input from the socket
+ bool readln(string &);
+ // Write the string + '\n' to the socket
+ void writeln(string const &);
+
+private:
+ // File descriptor for the data socket
+ int fd_;
+ // True if the connection is up
+ bool connected_;
+ // buffer for input data
+ string buffer;
+};
+
+LyXDataSocket::LyXDataSocket(string const & address)
+{
+ if ((fd_ = support::socktools::connect(address)) == -1) {
+ connected_ = false;
+ } else {
+ connected_ = true;
+ }
+}
+
+LyXDataSocket::~LyXDataSocket()
+{
+ ::close(fd_);
+}
+
+int LyXDataSocket::fd() const
+{
+ return fd_;
+}
+
+bool LyXDataSocket::connected() const
+{
+ return connected_;
+}
+
+// Returns true if there was a complete line to input
+// A line is of the form <key>:<value>
+// A line not of this form will not be passed
+// The line read is splitted and stored in 'key' and 'value'
+bool LyXDataSocket::readln(string & line)
+{
+ int const charbuf_size = 100;
+ char charbuf[charbuf_size]; // buffer for the ::read() system call
+ int count;
+ string::size_type pos;
+
+ // read and store characters in buffer
+ while ((count = ::read(fd_, charbuf, charbuf_size - 1)) > 0) {
+ charbuf[count] = '\0'; // turn it into a c string
+ buffer += charbuf;
+ }
+
+ // Error conditions. The buffer must still be
+ // processed for lines read
+ if (count == 0) { // EOF -- connection closed
+ connected_ = false;
+ } else if ((count == -1) && (errno != EAGAIN)) { // IO error
+ cerr << "lyxclient: IO error." << endl;
+ connected_ = false;
+ }
+
+ // Cut a line from buffer
+ if ((pos = buffer.find('\n')) == string::npos)
+ return false; // No complete line stored
+ line = buffer.substr(0, pos);
+ buffer = buffer.substr(pos + 1);
+ return true;
+}
+
+// Write a line of the form <key>:<value> to the socket
+void LyXDataSocket::writeln(string const & line)
+{
+ string linen(line + '\n');
+ int size = linen.size();
+ int written = ::write(fd_, linen.c_str(), size);
+ if (written < size) { // Allways mean end of connection.
+ if ((written == -1) && (errno == EPIPE)) {
+ // The program will also receive a SIGPIPE
+ // that must be catched
+ cerr << "lyxclient: connection closed while writing."
+ << endl;
+ } else {
+ // Anything else, including errno == EAGAIN, must be
+ // considered IO error. EAGAIN should never happen
+ // when line is small
+ cerr << "lyxclient: IO error: " << strerror(errno);
+ }
+ connected_ = false;
+ }
+}
+// ~Class LyXDataSocket -------------------------------------------------------
+
+
+// Class CmdLineParser -------------------------------------------------------
+class CmdLineParser
+{
+public:
+ typedef int (*optfunc)(vector<char *> const & args);
+ std::map<string, optfunc> helper;
+ std::map<string, bool> isset;
+ bool parse(int, char * []);
+ vector<char *> nonopt;
+};
+
+bool CmdLineParser::parse(int argc, char * argv[])
+{
+ int opt = 1;
+ while(opt < argc) {
+ vector<char *> args;
+ if(helper[argv[opt]]) {
+ isset[argv[opt]] = true;
+ int arg = opt + 1;
+ while((arg < argc) && (!helper[argv[arg]])) {
+ args.push_back(argv[arg]);
+ ++arg;
+ }
+ int taken = helper[argv[opt]](args);
+ if(taken == -1) return false;
+ opt += 1 + taken;
+ } else {
+ if(argv[opt][0] == '-') {
+ if((argv[opt][1] == '-')
+ && (argv[opt][2]== '\0')) {
+ ++opt;
+ while(opt < argc) {
+ nonopt.push_back(argv[opt]);
+ ++opt;
+ }
+ return true;
+ } else {
+ cerr << "lyxclient: unknown option "
+ << argv[opt] << endl;
+ return false;
+ }
+ }
+ nonopt.push_back(argv[opt]);
+ ++opt;
+ }
+ }
+ return true;
+}
+// ~Class CmdLineParser -------------------------------------------------------
+
+
+
+namespace cmdline
+{
+void usage()
+{
+ cerr << "Usage: lyxclient [options]" << endl
+ << "Options are:" << endl
+ << " -a address set address of the lyx socket" << endl
+ << " -t directory set system temporary directory" << endl
+ << " -p pid select a running lyx by pid" << endl
+ << " -c command send a single command and quit" << endl
+ << " -g file row send a command to go to file and row" << endl
+ << " -n name set client name" << endl
+ << " -h name display this help end exit" << endl
+ << "If -a is not used, lyxclient will use the arguments of -t and -p to look for" << endl
+ << "a running lyx. If -t is not set, 'directory' defaults to /tmp. If -p is set," << endl
+ << "lyxclient will connect only to a lyx with the specified pid. Options -c and -g" << endl
+ << "cannot be set simultaneoulsly. If no -c or -g options are given, lyxclient" << endl
+ << "will read commands from standard input and disconnect when command read is BYE:"
+ << endl;
+}
+
+int h(vector<char *> const & arg)
+{
+ usage();
+ exit(0);
+}
+
+string clientName(support::itoa(::getppid()) + ">" + support::itoa(::getpid()));
+int n(vector<char *> const & arg)
+{
+ if(arg.size() < 1) {
+ cerr << "lyxclient: The option -n requires 1 argument."
+ << endl;
+ return -1;
+ }
+ clientName = arg[0];
+ return 1;
+}
+
+string singleCommand;
+int c(vector<char *> const & arg)
+{
+ if(arg.size() < 1) {
+ cerr << "lyxclient: The option -c requires 1 argument."
+ << endl;
+ return -1;
+ }
+ singleCommand = arg[0];
+ return 1;
+}
+
+int g(vector<char *> const & arg)
+{
+ if(arg.size() < 2) {
+ cerr << "lyxclient: The option -g requires 2 arguments."
+ << endl;
+ return -1;
+ }
+ singleCommand = "LYXCMD:server-goto-file-row "
+ + static_cast<string>(arg[0]) + ' '
+ + static_cast<string>(arg[1]);
+ return 2;
+}
+
+char * serverAddress;
+int a(vector<char *> const & arg)
+{
+ if(arg.size() < 1) {
+ cerr << "lyxclient: The option -a requires 1 argument."
+ << endl;
+ return -1;
+ }
+ serverAddress = arg[0];
+ return 1;
+}
+
+string mainTmp("/tmp");
+int t(vector<char *> const & arg)
+{
+ if(arg.size() < 1) {
+ cerr << "lyxclient: The option -t requires 1 argument."
+ << endl;
+ return -1;
+ }
+ mainTmp = arg[0];
+ return 1;
+}
+
+string serverPid; // Init to empty string
+int p(vector<char *> const & arg)
+{
+ if(arg.size() < 1) {
+ cerr << "lyxclient: The option -p requires 1 argument."
+ << endl;
+ return -1;
+ }
+ serverPid = arg[0];
+ return 1;
+}
+
+} // namespace cmdline
+
+using support::prefixIs;
+
+int main(int argc, char * argv[])
+{
+ CmdLineParser parser;
+ parser.helper["-h"] = cmdline::h;
+ parser.helper["-c"] = cmdline::c;
+ parser.helper["-g"] = cmdline::g;
+ parser.helper["-n"] = cmdline::n;
+ parser.helper["-a"] = cmdline::a;
+ parser.helper["-t"] = cmdline::t;
+ parser.helper["-p"] = cmdline::p;
+ // Command line failure conditions:
+ if((!parser.parse(argc, argv))
+ || (parser.isset["-c"] && parser.isset["-g"])
+ || (parser.isset["-a"] && parser.isset["-p"])) {
+ cmdline::usage();
+ return 1;
+ }
+
+ LyXDataSocket * server;
+
+ if(parser.isset["-a"]) {
+ server = new LyXDataSocket(cmdline::serverAddress);
+ if(!server->connected()) {
+ cerr << "lyxclient: " << "Could not connect to "
+ << cmdline::serverAddress << endl;
+ return 1;
+ }
+ } else {
+ // We have to look for an address.
+ // serverPid can be empty.
+ vector<string> addrs = support::lyxSockets(cmdline::mainTmp, cmdline::serverPid);
+ vector<string>::iterator addr = addrs.begin();
+ vector<string>::iterator end = addrs.end();
+ while (addr < end) {
+ server = new LyXDataSocket(*addr);
+ if(server->connected()) break;
+ cerr << "lyxclient: " << "Could not connect to "
+ << *addr << endl;
+ delete server;
+ ++addr;
+ }
+ if(addr == end) {
+ cerr << "lyxclient: No suitable server found." << endl;
+ return 1;
+ }
+ cerr << "lyxclient: " << "Connected to " << *addr << endl;
+ }
+
+ int const serverfd = server->fd();
+
+ IOWatch iowatch;
+ iowatch.addfd(serverfd);
+
+ // Used to read from server
+ string answer;
+
+ // Send greeting
+ server->writeln("HELLO:" + cmdline::clientName);
+ // wait at most 2 seconds until server responds
+ iowatch.wait(2.0);
+ if(iowatch.isset(serverfd) && server->readln(answer)) {
+ if(prefixIs(answer, "BYE:")) {
+ cerr << "lyxclient: Server disconnected." << endl;
+ cout << answer << endl;
+ return 1;
+ }
+ } else {
+ cerr << "lyxclient: No answer from server." << endl;
+ return 1;
+ }
+
+ if(parser.isset["-g"] || parser.isset["-c"]) {
+ server->writeln(cmdline::singleCommand);
+ iowatch.wait(2.0);
+ if(iowatch.isset(serverfd) && server->readln(answer)) {
+ cout << answer;
+ if(prefixIs(answer, "ERROR:")) return 1;
+ return 0;
+ } else {
+ cerr << "lyxclient: No answer from server." << endl;
+ return 1;
+ }
+ }
+
+ // Take commands from stdin
+ iowatch.addfd(0); // stdin
+ bool saidbye = false;
+ while((!saidbye) && server->connected()) {
+ iowatch.wait();
+ if(iowatch.isset(0)) {
+ string command;
+ cin >> command;
+ if(command == "BYE:") {
+ server->writeln("BYE:");
+ saidbye = true;
+ } else {
+ server->writeln("LYXCMD:" + command);
+ }
+ }
+ if(iowatch.isset(serverfd)) {
+ while(server->readln(answer))
+ cout << answer << endl;
+ }
+ }
+
+ return 0;
+}