STREAM *STREAMSpawnFunction(BASIC_FUNC Func, void *Data, int Flags, const char *Config)
{
int to_fd, from_fd, *iptr;
pid_t pid=0;
STREAM *S=NULL;
char *Tempstr=NULL;
if (Flags & COMMS_BY_PTY)
{
pid=PseudoTTYSpawn(&to_fd, Tempstr, Flags, Config);
}
else
{
iptr=NULL;
if (Flags & COMMS_COMBINE_STDERR) iptr=(int *) COMMS_COMBINE_STDERR;
pid=PipeSpawnFunction(&to_fd, &from_fd, iptr, Func, Data, Flags, Config);
}
if (pid > 0) S=STREAMFromDualFD(from_fd, to_fd);
if (S)
{
STREAMSetFlushType(S,FLUSH_LINE,0,0);
Tempstr=FormatStr(Tempstr,"%d",pid);
STREAMSetValue(S,"PeerPID",Tempstr);
}
DestroyString(Tempstr);
return(S);
}
STREAM *STREAMSpawnCommand(const char *Command, const char *User, const char *Group, int Flags)
{
int to_fd, from_fd;
pid_t pid;
STREAM *S=NULL;
char *Tempstr=NULL;
if (Flags & SPAWN_TRUST_COMMAND) Tempstr=CopyStr(Tempstr,Command);
else Tempstr=MakeShellSafeString(Tempstr, Command, 0);
if (Flags & COMMS_BY_PTY)
{
pid=PseudoTTYSpawn(&to_fd,Tempstr,User,Group,Flags);
if (pid > 0) S=STREAMFromFD(to_fd);
}
else
{
if (Flags & COMMS_COMBINE_STDERR)
{
pid=PipeSpawn(&to_fd, &from_fd, COMMS_COMBINE_STDERR, Tempstr,User,Group);
}
else pid=PipeSpawn(&to_fd, &from_fd, NULL, Tempstr,User,Group);
if (pid > 0) S=STREAMFromDualFD(from_fd, to_fd);
}
if (S)
{
STREAMSetFlushType(S,FLUSH_LINE,0,0);
Tempstr=FormatStr(Tempstr,"%d",pid);
STREAMSetValue(S,"PeerPID",Tempstr);
}
DestroyString(Tempstr);
return(S);
}
int STREAMProcessConnectHop(STREAM *S, char *HopURL, int LastHop)
{
int val, result=FALSE;
char *Token=NULL, *Token2=NULL;
char *Tempstr=NULL;
char *User=NULL, *Host=NULL,*Pass=NULL, *KeyFile=NULL;
int Port=0;
ParseConnectDetails(HopURL, &Token, &Host, &Token2, &User, &Pass, NULL);
Port=atoi(Token2);
val=MatchTokenFromList(Token,HopTypes,0);
switch (val)
{
case CONNECT_HOP_TCP:
if (S->in_fd==-1)
{
S->in_fd=ConnectToHost(Host,Port,0);
S->out_fd=S->in_fd;
if (S->in_fd > -1) result=TRUE;
}
break;
case CONNECT_HOP_HTTPTUNNEL:
result=DoHTTPProxyTunnel(S, Host, Port, S->Path, 0);
break;
case CONNECT_HOP_SSH:
case CONNECT_HOP_SSHTUNNEL:
result=ConnectHopSSH(S, val, Host, Port, User, Pass, S->Path);
break;
case CONNECT_HOP_SOCKS4:
result=ConnectHopSocks(S, val, Host, Port, User, Pass, S->Path);
break;
case CONNECT_HOP_SHELL_CMD:
break;
case CONNECT_HOP_TELNET:
if (Port > 0)
{
Tempstr=FormatStr(Tempstr,"telnet -8 %s %d ",Host, Port);
}
else Tempstr=FormatStr(Tempstr,"telnet -8 %s ",Host);
if (S->in_fd==-1)
{
PseudoTTYSpawn(& S->in_fd,Tempstr,0);
S->out_fd=S->in_fd;
if (S->in_fd > -1)
{
result=TRUE;
STREAMSetFlushType(S,FLUSH_LINE,0,0);
}
}
else
{
Tempstr=CatStr(Tempstr,";exit\n");
STREAMWriteLine(Tempstr,S);
result=TRUE;
}
if (StrLen(User) > 0)
{
Tempstr=MCopyStr(Tempstr,User,"\n",NULL);
STREAMExpectAndReply(S,"ogin:",Tempstr);
}
if (StrLen(Pass) > 0)
{
Tempstr=MCopyStr(Tempstr,Pass,"\n",NULL);
STREAMExpectAndReply(S,"assword:",Tempstr);
}
STREAMExpectSilence(S,2);
break;
}
DestroyString(Tempstr);
DestroyString(Token);
DestroyString(KeyFile);
DestroyString(Host);
DestroyString(User);
DestroyString(Pass);
STREAMFlush(S);
return(result);
}
int ConnectHopSSH(STREAM *S,int Type, char *Host, int Port, char *User, char *Pass, char *NextHop)
{
char *Tempstr=NULL, *KeyFile=NULL, *Token=NULL, *Token2=NULL;
STREAM *AuthS;
int result=FALSE, val;
unsigned int TunnelPort=0;
if (Type==CONNECT_HOP_SSHTUNNEL)
{
TunnelPort=(rand() % (0xFFFF - 9000)) +9000;
//Host will be Token, and port Token2
ParseConnectDetails(NextHop, NULL, &Token, &Token2, NULL, NULL, NULL);
Tempstr=FormatStr(Tempstr,"ssh -2 -N %s@%s -L %d:%s:%s ",User,Host,TunnelPort,Token,Token2);
}
else Tempstr=MCopyStr(Tempstr,"ssh -2 -T ",User,"@",Host, " ", NULL );
if (strncmp(Pass,"keyfile:",8)==0)
{
if (S->in_fd != -1)
{
Token=FormatStr(Token,".%d-%d",getpid(),time(NULL));
SendPublicKeyToRemote(S,Token,Pass+8);
KeyFile=CopyStr(KeyFile,Token);
}
Tempstr=MCatStr(Tempstr,"-i ",KeyFile," ",NULL);
}
if (Port > 0)
{
Token=FormatStr(Token," -p %d ",Port);
Tempstr=CatStr(Tempstr,Token);
}
if (Type==CONNECT_HOP_SSHTUNNEL)
{
Tempstr=CatStr(Tempstr, " 2> /dev/null");
AuthS=STREAMSpawnCommand(Tempstr,COMMS_BY_PTY);
STREAMSetValue(S,"HelperPID:SSH",STREAMGetValue(AuthS,"PeerPID"));
}
else if (S->in_fd==-1)
{
Tempstr=CatStr(Tempstr, " 2> /dev/null");
PseudoTTYSpawn(&S->in_fd,Tempstr,0);
S->out_fd=S->in_fd;
if (S->in_fd > -1)
{
result=TRUE;
STREAMSetFlushType(S,FLUSH_LINE,0,0);
}
AuthS=S;
}
else
{
if (StrLen(KeyFile)) Tempstr=MCatStr(Tempstr," ; rm -f ",KeyFile,NULL);
Tempstr=CatStr(Tempstr,"; exit\n");
STREAMWriteLine(Tempstr,S);
result=TRUE;
AuthS=S;
}
if ((StrLen(KeyFile)==0) && (StrLen(Pass) > 0))
{
Token=MCopyStr(Token,Pass,"\n",NULL);
for (val=0; val < 3; val++)
{
if (STREAMExpectAndReply(AuthS,"assword:",Token)) break;
}
}
STREAMSetTimeout(AuthS,100);
//STREAMExpectSilence(AuthS);
sleep(3);
if (Type==CONNECT_HOP_SSHTUNNEL)
{
S->in_fd=ConnectToHost("127.0.0.1",TunnelPort,0);
S->out_fd=S->in_fd;
if (S->in_fd > -1) result=TRUE;
}
DestroyString(Tempstr);
DestroyString(KeyFile);
DestroyString(Token2);
DestroyString(Token);
return(result);
}
