1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
#include "clSSHAgent.hpp"
#include "file_logger.h"
#include "fileutils.h"
#include "procutils.h"
#include <wx/filename.h>
#include <wx/utils.h>
#ifndef __WXMSW__
static wxString& AddQuotesIfNeeded(wxString& str)
{
if(str.Contains(" ")) {
str.Prepend("\"").Append("\"");
}
return str;
}
#endif
clSSHAgent::clSSHAgent() { Start(); }
clSSHAgent::~clSSHAgent() { Stop(); }
void clSSHAgent::Start()
{
wxFileName sshAgent;
if(!FileUtils::FindExe("ssh-agent", sshAgent)) {
clDEBUG() << "Could not find ssh-agent executable";
return;
}
clDEBUG() << "Found ssh-agent:" << sshAgent;
// Check if an instance of ssh-agent is already running
#ifdef __WXMSW__
PidVec_t P = ProcUtils::PS("ssh-agent");
if(P.empty()) {
clDEBUG() << "Could not find a running instance of ssh-agent, starting one...";
::wxExecute(sshAgent.GetFullPath());
} else {
clDEBUG() << "Found ssh-agent running at pid:" << P.begin()->pid;
}
#else
wxFileName fnSocketPath(clStandardPaths::Get().GetUserDataDir(), "ssh-agent.");
fnSocketPath.AppendDir("tmp");
fnSocketPath.SetFullName(fnSocketPath.GetFullName() + (wxString() << ::wxGetProcessId()));
fnSocketPath.Mkdir(wxS_DIR_DEFAULT, wxPATH_MKDIR_FULL);
wxString socketPath = fnSocketPath.GetFullPath();
AddQuotesIfNeeded(socketPath);
wxString command = sshAgent.GetFullPath();
AddQuotesIfNeeded(command);
command << " -D -a " << socketPath;
m_sshAgentPID = ::wxExecute(command);
clDEBUG() << "Starting ssh-agent:" << command << ". pid:" << m_sshAgentPID;
// Call it on the next event loop, this will give the ssh-agent time to start
if(m_sshAgentPID != wxNOT_FOUND) {
wxString SSH_AUTH_SOCK;
wxString SSH_AGENT_PID;
SSH_AGENT_PID << m_sshAgentPID;
SSH_AUTH_SOCK = fnSocketPath.GetFullPath();
::wxSetEnv("SSH_AUTH_SOCK", SSH_AUTH_SOCK);
::wxSetEnv("SSH_AGENT_PID", SSH_AGENT_PID);
clDEBUG() << "SSH_AUTH_SOCK is set to:" << SSH_AUTH_SOCK;
clDEBUG() << "SSH_AGENT_PID is set to:" << SSH_AGENT_PID;
// Run ssh-add
}
#endif
// Execute ssh-add
sshAgent.SetFullName("ssh-add");
::wxExecute(sshAgent.GetFullPath());
}
void clSSHAgent::Stop()
{
#ifdef __WXGTK__
if(m_sshAgentPID != wxNOT_FOUND) {
wxKill(m_sshAgentPID, wxSIGTERM);
clDEBUG() << "Terminated ssh-agent:" << m_sshAgentPID;
m_sshAgentPID = wxNOT_FOUND;
}
#endif
}
|
