119 lines
2.5 KiB
C++
119 lines
2.5 KiB
C++
|
|
#include "distrib.h"
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/ip.h>
|
|
using namespace std;
|
|
|
|
int distrib::readHostFile(const char * filename)
|
|
{
|
|
ifstream ifs(filename);
|
|
|
|
if ( ! ifs.is_open() )
|
|
return 1;
|
|
|
|
string host;
|
|
while ( ! ifs.eof() )
|
|
{
|
|
ifs >> host;
|
|
m_hosts.push_back(host);
|
|
}
|
|
|
|
ifs.close();
|
|
return 0;
|
|
}
|
|
|
|
int distrib::connect()
|
|
{
|
|
int err = 0;
|
|
for (int i = 0, sz = m_hosts.size(); i < sz; i++)
|
|
{
|
|
err += connect(m_hosts[i]);
|
|
}
|
|
return err;
|
|
}
|
|
|
|
int distrib::connect(const string & host)
|
|
{
|
|
int id = fork();
|
|
if (id < 0) /* check for fork() error */
|
|
{
|
|
cerr << "Error forking: " << id << endl;
|
|
return 1;
|
|
}
|
|
else if (id > 0) /* in the parent */
|
|
{
|
|
m_children.push_back(id);
|
|
}
|
|
else /* in the child */
|
|
{
|
|
char server_port_str[15];
|
|
sprintf(server_port_str, "%d", m_serverport);
|
|
execlp("ssh", "ssh", host.c_str(),
|
|
"fart", "--host", m_servername.c_str(),
|
|
"--port", server_port_str,
|
|
(char *) NULL);
|
|
|
|
/* we should not get here */
|
|
cerr << "Error " << errno << " with execlp()!" << endl;
|
|
exit(33);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int distrib::startServer()
|
|
{
|
|
connect();
|
|
|
|
char hostname[1000];
|
|
gethostname(&hostname[0], 1000);
|
|
m_servername = hostname;
|
|
|
|
m_listen_socket = socket(PF_INET, SOCK_STREAM, 0);
|
|
if ( m_listen_socket == -1 )
|
|
{
|
|
cerr << "Error " << errno << " creating listen socket!" << endl;
|
|
return 1;
|
|
}
|
|
|
|
if ( listen(m_listen_socket, 5) == -1 )
|
|
{
|
|
cerr << "Error " << errno << " when trying to listen!" << endl;
|
|
return 2;
|
|
}
|
|
|
|
struct sockaddr_in addr;
|
|
int addr_len = sizeof(struct sockaddr_in);
|
|
getsockname(m_listen_socket,
|
|
(struct sockaddr *) &addr,
|
|
(socklen_t *) &addr_len);
|
|
|
|
int ip_addr = ntohl(addr.sin_addr.s_addr);
|
|
m_serverport = ntohs(addr.sin_port);
|
|
cout << "Listening on "
|
|
<< (unsigned int) ((ip_addr >> 24) & 0xFF)
|
|
<< '.'
|
|
<< (unsigned int) ((ip_addr >> 16) & 0xFF)
|
|
<< '.'
|
|
<< (unsigned int) ((ip_addr >> 8) & 0xFF)
|
|
<< '.'
|
|
<< (unsigned int) (ip_addr & 0xFF)
|
|
<< ':'
|
|
<< m_serverport
|
|
<< endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int distrib::startClient(const char * server, int port)
|
|
{
|
|
return 0;
|
|
}
|