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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
// $Id$
#include "HTBP_ID_Requestor.h"
#include "HTBP_Environment.h"
#include "ace/OS_NS_stdio.h"
#include "ace/OS_NS_string.h"
#include "ace/SOCK_Stream.h"
#include "ace/SOCK_Connector.h"
#include "ace/UUID.h"
ACE_RCSID(HTBP,
ACE_HTBP_ID_Requestor,
"$Id$")
ACE_BEGIN_VERSIONED_NAMESPACE_DECL
ACE_TString ACE::HTBP::ID_Requestor::htid_;
ACE_SYNCH_MUTEX ACE::HTBP::ID_Requestor::htid_lock_;
ACE::HTBP::ID_Requestor::ID_Requestor (ACE::HTBP::Environment *env)
: port_ (0),
host_ (),
url_()
{
if (env)
{
env->get_htid_url(url_);
int via_proxy = 0;
env->get_htid_via_proxy (via_proxy);
if (via_proxy && env->get_proxy_host (host_) == 0)
env->get_proxy_port (port_);
}
}
int
ACE::HTBP::ID_Requestor::connect_to_server (ACE_SOCK_Stream *cli_stream)
{
if (port_ == 0 || host_.length() == 0)
{
int host_start = url_.find (ACE_TEXT("http://")) + 7;
int port_sep = 0;
int sep = 0;
if (host_start == -1)
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT("ACE::HTBP::ID_Requestor::")
ACE_TEXT("connect_to_server: ")
ACE_TEXT("invalid URL: \"%s\"\n"),
url_.c_str()),
-1);
port_sep = url_.find (ACE_TEXT(":"),(size_t)host_start);
sep = url_.find (ACE_TEXT("/"),(size_t)host_start);
if (sep == -1 || sep == host_start +1)
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT("ACE::HTBP::ID_Requestor::")
ACE_TEXT("connect_to_server: ")
ACE_TEXT("invalid URL: \"%s\"\n"),
url_.c_str()),
-1);
if (port_sep == -1)
{
port_sep = sep;
port_ = 80; // should be a default?
}
host_ = url_.substr(host_start,port_sep - host_start);
}
ACE_INET_Addr remote_addr (static_cast<u_short> (port_), host_.c_str());
ACE_SOCK_Connector con;
if (con.connect (*cli_stream,
remote_addr) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT("ACE::HTBP::ID_Requestor::connect_to_server: ")
ACE_TEXT("%p\n"),
ACE_TEXT("socket connect")),
-1);
return 0;
}
int
ACE::HTBP::ID_Requestor::send_request (ACE_SOCK_Stream *cli_stream)
{
char *buffer;
ACE_NEW_RETURN (buffer, char[this->url_.length()+16],-1);
ACE_OS::sprintf (buffer,"GET %s HTTP/1.0\n\n",
ACE_TEXT_ALWAYS_CHAR(url_.c_str()));
int result = cli_stream->send_n (buffer,ACE_OS::strlen(buffer));
delete [] buffer;
if (result == -1)
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT("ACE::HTBP::ID_Requestor::send_request %p\n"),
ACE_TEXT("socket send")), -1);
return 0;
}
ACE_TCHAR *
ACE::HTBP::ID_Requestor::get_HTID ()
{
if (ACE::HTBP::ID_Requestor::htid_.length() != 0)
return ACE::HTBP::ID_Requestor::htid_.rep();
ACE_Guard<ACE_SYNCH_MUTEX> guard (ACE::HTBP::ID_Requestor::htid_lock_);
if (ACE::HTBP::ID_Requestor::htid_.length() != 0)
return ACE::HTBP::ID_Requestor::htid_.rep();
ACE_SOCK_Stream cli_stream;
ACE_TCHAR * htid = 0;
if (this->url_.length() == 0 ||
this->connect_to_server (&cli_stream) == -1 ||
this->send_request (&cli_stream) == -1)
{
ACE_Utils::UUID_Generator gen;
ACE_Utils::UUID *uuid = gen.generateUUID ();
const ACE_CString *uuidstr = uuid->to_string();
ACE::HTBP::ID_Requestor::htid_ = ACE_TEXT_CHAR_TO_TCHAR (uuidstr->c_str());
delete uuid;
return ACE::HTBP::ID_Requestor::htid_.rep();
}
iovec recv_buf;
ssize_t result = cli_stream.recvv (&recv_buf);
cli_stream.close();
if (result > 0)
{
ACE_CString answer ((char *)recv_buf.iov_base,recv_buf.iov_len);
ssize_t start = answer.rfind (ACE_TEXT('\n'));
if (start == ACE_CString::npos)
start = 0;
else
start++;
ACE::HTBP::ID_Requestor::htid_ = ACE_TEXT_CHAR_TO_TCHAR(answer.substr (start).c_str());
htid = ACE::HTBP::ID_Requestor::htid_.rep();
}
return htid;
}
ACE_END_VERSIONED_NAMESPACE_DECL
|
