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
140
141
142
143
144
|
/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
using System;
using System.Collections;
using System.Text;
using log4net;
using Apache.Qpid.Client.Protocol;
using Apache.Qpid.Client.Security;
using Apache.Qpid.Client.State;
using Apache.Qpid.Framing;
using Apache.Qpid.Sasl;
namespace Apache.Qpid.Client.Handler
{
public class ConnectionStartMethodHandler : IStateAwareMethodListener
{
private static readonly ILog _log = LogManager.GetLogger(typeof(ConnectionStartMethodHandler));
public void MethodReceived(AMQStateManager stateManager, AMQMethodEvent evt)
{
ConnectionStartBody body = (ConnectionStartBody) evt.Method;
AMQProtocolSession ps = evt.ProtocolSession;
try
{
if ( body.Mechanisms == null )
{
throw new AMQException("mechanism not specified in ConnectionStart method frame");
}
string mechanisms = Encoding.UTF8.GetString(body.Mechanisms);
string selectedMechanism = ChooseMechanism(mechanisms);
if ( selectedMechanism == null )
{
throw new AMQException("No supported security mechanism found, passed: " + mechanisms);
}
byte[] saslResponse = DoAuthentication(selectedMechanism, ps);
if (body.Locales == null)
{
throw new AMQException("Locales is not defined in Connection Start method");
}
string allLocales = Encoding.ASCII.GetString(body.Locales);
string[] locales = allLocales.Split(' ');
string selectedLocale;
if (locales != null && locales.Length > 0)
{
selectedLocale = locales[0];
}
else
{
throw new AMQException("No locales sent from server, passed: " + locales);
}
stateManager.ChangeState(AMQState.CONNECTION_NOT_TUNED);
FieldTable clientProperties = new FieldTable();
clientProperties["product"] = "Apache.Qpid.NET";
clientProperties["version"] = "1.0";
clientProperties["platform"] = GetFullSystemInfo();
clientProperties["instance"] = ps.ClientID;
AMQFrame frame = ConnectionStartOkBody.CreateAMQFrame(
evt.ChannelId, clientProperties, selectedMechanism,
saslResponse, selectedLocale);
ps.WriteFrame(frame);
}
catch (Exception e)
{
throw new AMQException(_log, "Unable to decode data: " + e, e);
}
}
private string GetFullSystemInfo()
{
StringBuilder sysInfo = new StringBuilder();
// check if we're running on mono or .net
Type monoRuntime = Type.GetType("Mono.Runtime");
if ( monoRuntime != null )
sysInfo.Append("Mono");
else
sysInfo.Append(".NET");
sysInfo.Append(" ").Append(Environment.Version);
sysInfo.Append(", ").Append(Environment.OSVersion);
return sysInfo.ToString();
}
private string ChooseMechanism(string mechanisms)
{
return CallbackHandlerRegistry.Instance.ChooseMechanism(mechanisms);
}
private byte[] DoAuthentication(string selectedMechanism, AMQProtocolSession ps)
{
ISaslClient saslClient = Sasl.Sasl.CreateClient(
new string[] { selectedMechanism }, null, "AMQP", "localhost",
new Hashtable(), CreateCallbackHandler(selectedMechanism, ps)
);
if ( saslClient == null )
{
throw new AMQException("Client SASL configuration error: no SaslClient could be created for mechanism " +
selectedMechanism);
}
ps.SaslClient = saslClient;
try
{
return saslClient.HasInitialResponse ?
saslClient.EvaluateChallenge(new byte[0]) : null;
} catch ( Exception ex )
{
ps.SaslClient = null;
throw new AMQException("Unable to create SASL client", ex);
}
}
private IAMQCallbackHandler CreateCallbackHandler(string mechanism, AMQProtocolSession session)
{
Type type = CallbackHandlerRegistry.Instance.GetCallbackHandler(mechanism);
IAMQCallbackHandler handler =
(IAMQCallbackHandler)Activator.CreateInstance(type);
if ( handler == null )
throw new AMQException("Unable to create callback handler: " + mechanism);
handler.Initialize(session);
return handler;
}
}
}
|
