1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package org.apache.directory.shared.kerberos.components;
22
23
24 import java.nio.BufferOverflowException;
25 import java.nio.ByteBuffer;
26
27 import org.apache.directory.api.asn1.AbstractAsn1Object;
28 import org.apache.directory.api.asn1.EncoderException;
29 import org.apache.directory.api.asn1.ber.tlv.BerValue;
30 import org.apache.directory.api.asn1.ber.tlv.TLV;
31 import org.apache.directory.api.asn1.ber.tlv.UniversalTag;
32 import org.apache.directory.api.util.Strings;
33 import org.apache.directory.server.i18n.I18n;
34 import org.apache.directory.shared.kerberos.KerberosConstants;
35 import org.apache.directory.shared.kerberos.KerberosTime;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55 public class KrbSafeBody extends AbstractAsn1Object
56 {
57
58 private static final Logger log = LoggerFactory.getLogger( KrbSafeBody.class );
59
60
61 private static final boolean IS_DEBUG = log.isDebugEnabled();
62
63
64 private byte[] userData;
65
66
67 private KerberosTime timestamp;
68
69
70 private Integer usec;
71
72
73 private Integer seqNumber;
74
75
76 private HostAddress senderAddress;
77
78
79 private HostAddress recipientAddress;
80
81
82 private int userDataLen;
83 private int timestampLen;
84 private int usecLen;
85 private int seqNumberLen;
86 private int senderAddressLen;
87 private int recipientAddressLen;
88 private int krbSafeBodySeqLen;
89
90
91
92
93
94 public KrbSafeBody()
95 {
96 }
97
98
99
100
101
102 public byte[] getUserData()
103 {
104 return userData;
105 }
106
107
108
109
110
111 public void setUserData( byte[] userData )
112 {
113 this.userData = userData;
114 }
115
116
117
118
119
120 public KerberosTime getTimestamp()
121 {
122 return timestamp;
123 }
124
125
126
127
128
129 public void setTimestamp( KerberosTime timestamp )
130 {
131 this.timestamp = timestamp;
132 }
133
134
135
136
137
138 public int getUsec()
139 {
140 if ( usec == null )
141 {
142 return 0;
143 }
144
145 return usec;
146 }
147
148
149
150
151
152 public void setUsec( int usec )
153 {
154 this.usec = usec;
155 }
156
157
158
159
160
161 public int getSeqNumber()
162 {
163 if ( seqNumber == null )
164 {
165 return 0;
166 }
167
168 return seqNumber;
169 }
170
171
172
173
174
175 public void setSeqNumber( int seqNumber )
176 {
177 this.seqNumber = seqNumber;
178 }
179
180
181
182
183
184 public HostAddress getSenderAddress()
185 {
186 return senderAddress;
187 }
188
189
190
191
192
193 public void setSenderAddress( HostAddress senderAddress )
194 {
195 this.senderAddress = senderAddress;
196 }
197
198
199
200
201
202 public HostAddress getRecipientAddress()
203 {
204 return recipientAddress;
205 }
206
207
208
209
210
211 public void setRecipientAddress( HostAddress recipientAddress )
212 {
213 this.recipientAddress = recipientAddress;
214 }
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248 @Override
249 public int computeLength()
250 {
251 userDataLen = 1 + TLV.getNbBytes( userData.length ) + userData.length;
252 krbSafeBodySeqLen = 1 + TLV.getNbBytes( userDataLen ) + userDataLen;
253
254 senderAddressLen = senderAddress.computeLength();
255 krbSafeBodySeqLen += 1 + TLV.getNbBytes( senderAddressLen ) + senderAddressLen;
256
257 if ( timestamp != null )
258 {
259 timestampLen = timestamp.getBytes().length;
260 timestampLen = 1 + TLV.getNbBytes( timestampLen ) + timestampLen;
261 krbSafeBodySeqLen += 1 + TLV.getNbBytes( timestampLen ) + timestampLen;
262 }
263
264 if ( usec != null )
265 {
266 usecLen = BerValue.getNbBytes( usec );
267 usecLen = 1 + TLV.getNbBytes( usecLen ) + usecLen;
268 krbSafeBodySeqLen += 1 + TLV.getNbBytes( usecLen ) + usecLen;
269 }
270
271 if ( seqNumber != null )
272 {
273 seqNumberLen = BerValue.getNbBytes( seqNumber );
274 seqNumberLen = 1 + TLV.getNbBytes( seqNumberLen ) + seqNumberLen;
275 krbSafeBodySeqLen += 1 + TLV.getNbBytes( seqNumberLen ) + seqNumberLen;
276 }
277
278 if ( recipientAddress != null )
279 {
280 recipientAddressLen = recipientAddress.computeLength();
281 krbSafeBodySeqLen += 1 + TLV.getNbBytes( recipientAddressLen ) + recipientAddressLen;
282 }
283
284 return 1 + TLV.getNbBytes( krbSafeBodySeqLen ) + krbSafeBodySeqLen;
285 }
286
287
288
289
290
291 @Override
292 public ByteBuffer encode( ByteBuffer buffer ) throws EncoderException
293 {
294 if ( buffer == null )
295 {
296 throw new EncoderException( I18n.err( I18n.ERR_148 ) );
297 }
298
299 try
300 {
301 buffer.put( UniversalTag.SEQUENCE.getValue() );
302 buffer.put( TLV.getBytes( krbSafeBodySeqLen ) );
303
304
305 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_USER_DATA_TAG );
306 buffer.put( TLV.getBytes( userDataLen ) );
307 BerValue.encode( buffer, userData );
308
309 if ( timestamp != null )
310 {
311
312 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_TIMESTAMP_TAG );
313 buffer.put( TLV.getBytes( timestampLen ) );
314
315
316 buffer.put( ( byte ) UniversalTag.GENERALIZED_TIME.getValue() );
317 buffer.put( ( byte ) 0x0F );
318 buffer.put( timestamp.getBytes() );
319 }
320
321 if ( usec != null )
322 {
323
324 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_USEC_TAG );
325 buffer.put( TLV.getBytes( usecLen ) );
326 BerValue.encode( buffer, usec );
327 }
328
329 if ( seqNumber != null )
330 {
331
332 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_SEQ_NUMBER_TAG );
333 buffer.put( TLV.getBytes( seqNumberLen ) );
334 BerValue.encode( buffer, seqNumber );
335 }
336
337
338 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_SENDER_ADDRESS_TAG );
339 buffer.put( TLV.getBytes( senderAddressLen ) );
340 senderAddress.encode( buffer );
341
342 if ( recipientAddress != null )
343 {
344
345 buffer.put( ( byte ) KerberosConstants.KRB_SAFE_BODY_RECIPIENT_ADDRESS_TAG );
346 buffer.put( TLV.getBytes( recipientAddressLen ) );
347 recipientAddress.encode( buffer );
348 }
349 }
350 catch ( BufferOverflowException boe )
351 {
352 log.error( I18n.err( I18n.ERR_735_CANNOT_ENCODE_KRBSAFEBODY, 1 + TLV.getNbBytes( krbSafeBodySeqLen )
353 + krbSafeBodySeqLen, buffer.capacity() ) );
354 throw new EncoderException( I18n.err( I18n.ERR_138 ) );
355 }
356
357 if ( IS_DEBUG )
358 {
359 log.debug( "KrbSafeBody encoding : {}", Strings.dumpBytes( buffer.array() ) );
360 log.debug( "KrbSafeBody initial value : {}", toString() );
361 }
362
363 return buffer;
364 }
365
366
367
368
369
370 public String toString()
371 {
372 StringBuilder sb = new StringBuilder();
373
374 sb.append( "KRB-SAFE-BODY : {\n" );
375 sb.append( " user-data: " ).append( Strings.dumpBytes( userData ) ).append( '\n' );
376
377 if ( timestamp != null )
378 {
379 sb.append( " timestamp: " ).append( timestamp.getDate() ).append( '\n' );
380 }
381
382 if ( usec != null )
383 {
384 sb.append( " usec: " ).append( usec ).append( '\n' );
385 }
386
387 if ( seqNumber != null )
388 {
389 sb.append( " seq-number: " ).append( seqNumber ).append( '\n' );
390 }
391
392 sb.append( " s-address: " ).append( senderAddress ).append( '\n' );
393
394 if ( recipientAddress != null )
395 {
396 sb.append( " r-address: " ).append( recipientAddress ).append( '\n' );
397 }
398
399 sb.append( "}\n" );
400
401 return sb.toString();
402 }
403 }