map = new HashMap<>();
266 | for (int i = 0; i < request.length; i++) {
267 | log.trace("Request {}: {}", i, request[i]);
268 | if (request[i].startsWith("GET ") || request[i].startsWith("POST ") || request[i].startsWith("PUT ")) {
269 | // "GET /chat/room1?id=publisher1 HTTP/1.1"
270 | // split it on space
271 | String requestPath = request[i].split("\\s+")[1];
272 | // get the path data for handShake
273 | int start = requestPath.indexOf('/');
274 | int end = requestPath.length();
275 | int ques = requestPath.indexOf('?');
276 | if (ques > 0) {
277 | end = ques;
278 | }
279 | log.trace("Request path: {} to {} ques: {}", start, end, ques);
280 | String path = requestPath.substring(start, end).trim();
281 | log.trace("Client request path: {}", path);
282 | conn.setPath(path);
283 | // check for '?' or included query string
284 | if (ques > 0) {
285 | // parse any included query string
286 | String qs = requestPath.substring(ques).trim();
287 | log.trace("Request querystring: {}", qs);
288 | map.put(Constants.URI_QS_PARAMETERS, parseQuerystring(qs));
289 | }
290 | // get the manager
291 | WebSocketPlugin plugin = (WebSocketPlugin) PluginRegistry.getPlugin("WebSocketPlugin");
292 | if (plugin != null) {
293 | log.trace("Found plugin");
294 | WebSocketScopeManager manager = plugin.getManager(path);
295 | log.trace("Manager was found? : {}", manager);
296 | // only check that the application is enabled, not the room or sub levels
297 | if (manager != null && manager.isEnabled(path)) {
298 | log.trace("Path enabled: {}", path);
299 | } else {
300 | // invalid scope or its application is not enabled, send disconnect message
301 | conn.close(1002, build400Response(conn));
302 | throw new WebSocketException("Handshake failed, path not enabled");
303 | }
304 | } else {
305 | log.warn("Plugin lookup failed");
306 | conn.close(1002, build400Response(conn));
307 | throw new WebSocketException("Handshake failed, missing plugin");
308 | }
309 | } else if (request[i].contains(Constants.WS_HEADER_KEY)) {
310 | map.put(Constants.WS_HEADER_KEY, extractHeaderValue(request[i]));
311 | } else if (request[i].contains(Constants.WS_HEADER_VERSION)) {
312 | map.put(Constants.WS_HEADER_VERSION, extractHeaderValue(request[i]));
313 | } else if (request[i].contains(Constants.WS_HEADER_EXTENSIONS)) {
314 | map.put(Constants.WS_HEADER_EXTENSIONS, extractHeaderValue(request[i]));
315 | } else if (request[i].contains(Constants.WS_HEADER_PROTOCOL)) {
316 | map.put(Constants.WS_HEADER_PROTOCOL, extractHeaderValue(request[i]));
317 | } else if (request[i].contains(Constants.HTTP_HEADER_HOST)) {
318 | // get the host data
319 | host = extractHeaderValue(request[i]);
320 | conn.setHost(host);
321 | } else if (request[i].contains(Constants.HTTP_HEADER_ORIGIN)) {
322 | // get the origin data
323 | origin = extractHeaderValue(request[i]);
324 | conn.setOrigin(origin);
325 | } else if (request[i].contains(Constants.HTTP_HEADER_USERAGENT)) {
326 | map.put(Constants.HTTP_HEADER_USERAGENT, extractHeaderValue(request[i]));
327 | } else if (request[i].startsWith(Constants.WS_HEADER_GENERIC_PREFIX)) {
328 | map.put(getHeaderName(request[i]), extractHeaderValue(request[i]));
329 | }
330 | }
331 | // policy checking
332 | boolean validOrigin = true;
333 | if (conn.isSameOriginPolicy()) {
334 | // if SOP / origin validation is enabled, verify the origin
335 | String trimmedHost = host;
336 | // strip protocol if its there
337 | if (host.startsWith("http")) {
338 | trimmedHost = host.substring(host.indexOf("//") + 1);
339 | }
340 | // chop off port etc..
341 | int colonIndex = trimmedHost.indexOf(':');
342 | if (colonIndex > 0) {
343 | trimmedHost = trimmedHost.substring(0, colonIndex);
344 | }
345 | log.debug("Trimmed host: {}", trimmedHost);
346 | validOrigin = origin.contains(trimmedHost);
347 | log.debug("Same Origin? {}", validOrigin);
348 | }
349 | if (conn.isCrossOriginPolicy()) {
350 | // if CORS is enabled
351 | validOrigin = conn.isValidOrigin(origin);
352 | log.debug("Origin {} valid? {}", origin, validOrigin);
353 | }
354 | if (!validOrigin) {
355 | conn.close(1008, build403Response(conn));
356 | throw new WebSocketException(String.format("Policy failure - SOP enabled: %b CORS enabled: %b", WebSocketTransport.isSameOriginPolicy(), WebSocketTransport.isCrossOriginPolicy()));
357 | } else {
358 | log.debug("Origin is valid");
359 | }
360 | return map;
361 | }
362 |
363 | /**
364 | * Returns the trimmed header name.
365 | *
366 | * @param requestHeader
367 | * @return value
368 | */
369 | private String getHeaderName(String requestHeader) {
370 | return requestHeader.substring(0, requestHeader.indexOf(':')).trim();
371 | }
372 |
373 | /**
374 | * Returns the trimmed header value.
375 | *
376 | * @param requestHeader
377 | * @return value
378 | */
379 | private String extractHeaderValue(String requestHeader) {
380 | return requestHeader.substring(requestHeader.indexOf(':') + 1).trim();
381 | }
382 |
383 | /**
384 | * Build a handshake response based on the given client key.
385 | *
386 | * @param clientKey
387 | * @return response
388 | * @throws WebSocketException
389 | */
390 | private HandshakeResponse buildHandshakeResponse(WebSocketConnection conn, String clientKey) throws WebSocketException {
391 | if (log.isDebugEnabled()) {
392 | log.debug("buildHandshakeResponse: {} client key: {}", conn, clientKey);
393 | }
394 | byte[] accept;
395 | try {
396 | // performs the accept creation routine from RFC6455 @see RFC6455
397 | // concatenate the key and magic string, then SHA1 hash and base64 encode
398 | MessageDigest md = MessageDigest.getInstance("SHA1");
399 | accept = Base64.encode(md.digest((clientKey + Constants.WEBSOCKET_MAGIC_STRING).getBytes()));
400 | } catch (NoSuchAlgorithmException e) {
401 | throw new WebSocketException("Algorithm is missing");
402 | }
403 | // make up reply data...
404 | IoBuffer buf = IoBuffer.allocate(308);
405 | buf.setAutoExpand(true);
406 | buf.put("HTTP/1.1 101 Switching Protocols".getBytes());
407 | buf.put(Constants.CRLF);
408 | buf.put("Upgrade: websocket".getBytes());
409 | buf.put(Constants.CRLF);
410 | buf.put("Connection: Upgrade".getBytes());
411 | buf.put(Constants.CRLF);
412 | buf.put("Server: Red5".getBytes());
413 | buf.put(Constants.CRLF);
414 | buf.put("Sec-WebSocket-Version-Server: 13".getBytes());
415 | buf.put(Constants.CRLF);
416 | buf.put(String.format("Sec-WebSocket-Origin: %s", conn.getOrigin()).getBytes());
417 | buf.put(Constants.CRLF);
418 | buf.put(String.format("Sec-WebSocket-Location: %s", conn.getHost()).getBytes());
419 | buf.put(Constants.CRLF);
420 | // send back extensions if enabled
421 | if (conn.hasExtensions()) {
422 | buf.put(String.format("Sec-WebSocket-Extensions: %s", conn.getExtensionsAsString()).getBytes());
423 | buf.put(Constants.CRLF);
424 | }
425 | // send back protocol if enabled
426 | if (conn.hasProtocol()) {
427 | buf.put(String.format("Sec-WebSocket-Protocol: %s", conn.getProtocol()).getBytes());
428 | buf.put(Constants.CRLF);
429 | }
430 | buf.put(String.format("Sec-WebSocket-Accept: %s", new String(accept)).getBytes());
431 | buf.put(Constants.CRLF);
432 | buf.put(Constants.CRLF);
433 | // if any bytes follow this crlf, the follow-up data will be corrupted
434 | if (log.isTraceEnabled()) {
435 | log.trace("Handshake response size: {}", buf.limit());
436 | }
437 | return new HandshakeResponse(buf);
438 | }
439 |
440 | /**
441 | * Build an HTTP 400 "Bad Request" response.
442 | *
443 | * @return response
444 | * @throws WebSocketException
445 | */
446 | private HandshakeResponse build400Response(WebSocketConnection conn) throws WebSocketException {
447 | if (log.isDebugEnabled()) {
448 | log.debug("build400Response: {}", conn);
449 | }
450 | // make up reply data...
451 | IoBuffer buf = IoBuffer.allocate(32);
452 | buf.setAutoExpand(true);
453 | buf.put("HTTP/1.1 400 Bad Request".getBytes());
454 | buf.put(Constants.CRLF);
455 | buf.put("Sec-WebSocket-Version-Server: 13".getBytes());
456 | buf.put(Constants.CRLF);
457 | buf.put(Constants.CRLF);
458 | if (log.isTraceEnabled()) {
459 | log.trace("Handshake error response size: {}", buf.limit());
460 | }
461 | return new HandshakeResponse(buf);
462 | }
463 |
464 | /**
465 | * Build an HTTP 403 "Forbidden" response.
466 | *
467 | * @return response
468 | * @throws WebSocketException
469 | */
470 | private HandshakeResponse build403Response(WebSocketConnection conn) throws WebSocketException {
471 | if (log.isDebugEnabled()) {
472 | log.debug("build403Response: {}", conn);
473 | }
474 | // make up reply data...
475 | IoBuffer buf = IoBuffer.allocate(32);
476 | buf.setAutoExpand(true);
477 | buf.put("HTTP/1.1 403 Forbidden".getBytes());
478 | buf.put(Constants.CRLF);
479 | buf.put("Sec-WebSocket-Version-Server: 13".getBytes());
480 | buf.put(Constants.CRLF);
481 | buf.put(Constants.CRLF);
482 | if (log.isTraceEnabled()) {
483 | log.trace("Handshake error response size: {}", buf.limit());
484 | }
485 | return new HandshakeResponse(buf);
486 | }
487 |
488 | /**
489 | * Decode the in buffer according to the Section 5.2. RFC 6455. If there are multiple websocket dataframes in the buffer, this will parse all and return one complete decoded buffer.
490 | *
491 | *
492 | * 0 1 2 3
493 | * 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
494 | * +-+-+-+-+-------+-+-------------+-------------------------------+
495 | * |F|R|R|R| opcode|M| Payload len | Extended payload length |
496 | * |I|S|S|S| (4) |A| (7) | (16/64) |
497 | * |N|V|V|V| |S| | (if payload len==126/127) |
498 | * | |1|2|3| |K| | |
499 | * +-+-+-+-+-------+-+-------------+ - - - - - - - - - - - - - - - +
500 | * | Extended payload length continued, if payload len == 127 |
501 | * + - - - - - - - - - - - - - - - +-------------------------------+
502 | * | |Masking-key, if MASK set to 1 |
503 | * +-------------------------------+-------------------------------+
504 | * | Masking-key (continued) | Payload Data |
505 | * +-------------------------------- - - - - - - - - - - - - - - - +
506 | * : Payload Data continued ... :
507 | * + - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - +
508 | * | Payload Data continued ... |
509 | * +---------------------------------------------------------------+
510 | *
511 | *
512 | * @param in
513 | * @param session
514 | */
515 | public static void decodeIncommingData(IoBuffer in, IoSession session) {
516 | log.trace("Decoding: {}", in);
517 | // get decoder state
518 | DecoderState decoderState = (DecoderState) session.getAttribute(DECODER_STATE_KEY);
519 | if (decoderState.fin == Byte.MIN_VALUE) {
520 | byte frameInfo = in.get();
521 | // get FIN (1 bit)
522 | //log.debug("frameInfo: {}", Integer.toBinaryString((frameInfo & 0xFF) + 256));
523 | decoderState.fin = (byte) ((frameInfo >>> 7) & 1);
524 | log.trace("FIN: {}", decoderState.fin);
525 | // the next 3 bits are for RSV1-3 (not used here at the moment)
526 | // get the opcode (4 bits)
527 | decoderState.opCode = (byte) (frameInfo & 0x0f);
528 | log.trace("Opcode: {}", decoderState.opCode);
529 | // opcodes 3-7 and b-f are reserved for non-control frames
530 | }
531 | if (decoderState.mask == Byte.MIN_VALUE) {
532 | byte frameInfo2 = in.get();
533 | // get mask bit (1 bit)
534 | decoderState.mask = (byte) ((frameInfo2 >>> 7) & 1);
535 | log.trace("Mask: {}", decoderState.mask);
536 | // get payload length (7, 7+16, 7+64 bits)
537 | decoderState.frameLen = (frameInfo2 & (byte) 0x7F);
538 | log.trace("Payload length: {}", decoderState.frameLen);
539 | if (decoderState.frameLen == 126) {
540 | decoderState.frameLen = in.getUnsignedShort();
541 | log.trace("Payload length updated: {}", decoderState.frameLen);
542 | } else if (decoderState.frameLen == 127) {
543 | long extendedLen = in.getLong();
544 | if (extendedLen >= Integer.MAX_VALUE) {
545 | log.error("Data frame is too large for this implementation. Length: {}", extendedLen);
546 | } else {
547 | decoderState.frameLen = (int) extendedLen;
548 | }
549 | log.trace("Payload length updated: {}", decoderState.frameLen);
550 | }
551 | }
552 | // ensure enough bytes left to fill payload, if masked add 4 additional bytes
553 | if (decoderState.frameLen + (decoderState.mask == 1 ? 4 : 0) > in.remaining()) {
554 | log.info("Not enough data available to decode, socket may be closed/closing");
555 | } else {
556 | // if the data is masked (xor'd)
557 | if (decoderState.mask == 1) {
558 | // get the mask key
559 | byte maskKey[] = new byte[4];
560 | for (int i = 0; i < 4; i++) {
561 | maskKey[i] = in.get();
562 | }
563 | /* now un-mask frameLen bytes as per Section 5.3 RFC 6455
564 | Octet i of the transformed data ("transformed-octet-i") is the XOR of
565 | octet i of the original data ("original-octet-i") with octet at index
566 | i modulo 4 of the masking key ("masking-key-octet-j"):
567 | j = i MOD 4
568 | transformed-octet-i = original-octet-i XOR masking-key-octet-j
569 | */
570 | decoderState.payload = new byte[decoderState.frameLen];
571 | for (int i = 0; i < decoderState.frameLen; i++) {
572 | byte maskedByte = in.get();
573 | decoderState.payload[i] = (byte) (maskedByte ^ maskKey[i % 4]);
574 | }
575 | } else {
576 | decoderState.payload = new byte[decoderState.frameLen];
577 | in.get(decoderState.payload);
578 | }
579 | // if FIN == 0 we have fragments
580 | if (decoderState.fin == 0) {
581 | // store the fragment and continue
582 | IoBuffer fragments = (IoBuffer) session.getAttribute(DECODED_MESSAGE_FRAGMENTS_KEY);
583 | if (fragments == null) {
584 | fragments = IoBuffer.allocate(decoderState.frameLen);
585 | fragments.setAutoExpand(true);
586 | session.setAttribute(DECODED_MESSAGE_FRAGMENTS_KEY, fragments);
587 | // store message type since following type may be a continuation
588 | MessageType messageType = MessageType.CLOSE;
589 | switch (decoderState.opCode) {
590 | case 0: // continuation
591 | messageType = MessageType.CONTINUATION;
592 | break;
593 | case 1: // text
594 | messageType = MessageType.TEXT;
595 | break;
596 | case 2: // binary
597 | messageType = MessageType.BINARY;
598 | break;
599 | case 9: // ping
600 | messageType = MessageType.PING;
601 | break;
602 | case 0xa: // pong
603 | messageType = MessageType.PONG;
604 | break;
605 | }
606 | session.setAttribute(DECODED_MESSAGE_TYPE_KEY, messageType);
607 | }
608 | fragments.put(decoderState.payload);
609 | // remove decoder state
610 | session.removeAttribute(DECODER_STATE_KEY);
611 | } else {
612 | // create a message
613 | WSMessage message = new WSMessage();
614 | // check for previously set type from the first fragment (if we have fragments)
615 | MessageType messageType = (MessageType) session.getAttribute(DECODED_MESSAGE_TYPE_KEY);
616 | if (messageType == null) {
617 | switch (decoderState.opCode) {
618 | case 0: // continuation
619 | messageType = MessageType.CONTINUATION;
620 | break;
621 | case 1: // text
622 | messageType = MessageType.TEXT;
623 | break;
624 | case 2: // binary
625 | messageType = MessageType.BINARY;
626 | break;
627 | case 9: // ping
628 | messageType = MessageType.PING;
629 | break;
630 | case 0xa: // pong
631 | messageType = MessageType.PONG;
632 | break;
633 | case 8: // close
634 | messageType = MessageType.CLOSE;
635 | // handler or listener should close upon receipt
636 | break;
637 | default:
638 | // TODO throw ex?
639 | log.info("Unhandled opcode: {}", decoderState.opCode);
640 | }
641 | }
642 | // set message type
643 | message.setMessageType(messageType);
644 | // check for fragments and piece them together, otherwise just send the single completed frame
645 | IoBuffer fragments = (IoBuffer) session.removeAttribute(DECODED_MESSAGE_FRAGMENTS_KEY);
646 | if (fragments != null) {
647 | fragments.put(decoderState.payload);
648 | fragments.flip();
649 | message.setPayload(fragments);
650 | } else {
651 | // add the payload
652 | message.addPayload(decoderState.payload);
653 | }
654 | // set the message on the session
655 | session.setAttribute(DECODED_MESSAGE_KEY, message);
656 | // remove decoder state
657 | session.removeAttribute(DECODER_STATE_KEY);
658 | // remove type
659 | session.removeAttribute(DECODED_MESSAGE_TYPE_KEY);
660 | }
661 | }
662 | }
663 |
664 | /**
665 | * Returns a map of key / value pairs from a given querystring.
666 | *
667 | * @param query
668 | * @return k/v map
669 | */
670 | public static Map parseQuerystring(String query) {
671 | String[] params = query.split("&");
672 | Map map = new HashMap();
673 | for (String param : params) {
674 | String[] nameValue = param.split("=");
675 | map.put(nameValue[0], nameValue[1]);
676 | }
677 | return map;
678 | }
679 |
680 | }
681 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/codec/WebSocketEncoder.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.codec;
20 |
21 | import org.apache.mina.core.buffer.IoBuffer;
22 | import org.apache.mina.core.session.IoSession;
23 | import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
24 | import org.apache.mina.filter.codec.ProtocolEncoderOutput;
25 | import org.red5.net.websocket.Constants;
26 | import org.red5.net.websocket.WebSocketConnection;
27 | import org.red5.net.websocket.model.HandshakeRequest;
28 | import org.red5.net.websocket.model.HandshakeResponse;
29 | import org.red5.net.websocket.model.Packet;
30 | import org.slf4j.Logger;
31 | import org.slf4j.LoggerFactory;
32 |
33 | /**
34 | * Encodes incoming buffers in a manner that makes the receiving client type transparent to the encoders further up in the filter chain. If the receiving client is a native client then the buffer contents are simply passed through. If the receiving client is a websocket, it will encode the buffer contents in to WebSocket DataFrame before passing it along the filter chain.
35 | *
36 | * Note: you must wrap the IoBuffer you want to send around a WebSocketCodecPacket instance.
37 | *
38 | * @author Dhruv Chopra
39 | * @author Paul Gregoire
40 | */
41 | public class WebSocketEncoder extends ProtocolEncoderAdapter {
42 |
43 | private static final Logger log = LoggerFactory.getLogger(WebSocketEncoder.class);
44 |
45 | @Override
46 | public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
47 | WebSocketConnection conn = (WebSocketConnection) session.getAttribute(Constants.CONNECTION);
48 | IoBuffer resultBuffer;
49 | if (message instanceof Packet) {
50 | Packet packet = (Packet) message;
51 | // if the connection is not native / direct, add websocket encoding
52 | resultBuffer = conn.isWebConnection() ? encodeOutgoingData(packet) : packet.getData();
53 | } else if (message instanceof HandshakeResponse) {
54 | HandshakeResponse resp = (HandshakeResponse) message;
55 | resultBuffer = resp.getResponse();
56 | } else if (message instanceof HandshakeRequest) {
57 | HandshakeRequest req = (HandshakeRequest) message;
58 | resultBuffer = req.getRequest();
59 | } else {
60 | throw new Exception("message not a websocket type");
61 | }
62 | out.write(resultBuffer);
63 | }
64 |
65 | // Encode the in buffer according to the Section 5.2. RFC 6455
66 | public static IoBuffer encodeOutgoingData(Packet packet) {
67 | log.debug("encode outgoing: {}", packet);
68 | // get the payload data
69 | IoBuffer data = packet.getData();
70 | // get the frame length based on the byte count
71 | int frameLen = data.limit();
72 | // start with frame length + 2b (header info)
73 | IoBuffer buffer = IoBuffer.allocate(frameLen + 2, false);
74 | buffer.setAutoExpand(true);
75 | // set the proper flags / opcode for the data
76 | byte frameInfo = (byte) (1 << 7);
77 | switch (packet.getType()) {
78 | case TEXT:
79 | if (log.isTraceEnabled()) {
80 | log.trace("Encoding text frame \r\n{}", new String(packet.getData().array()));
81 | }
82 | frameInfo = (byte) (frameInfo | 1);
83 | break;
84 | case BINARY:
85 | log.trace("Encoding binary frame");
86 | frameInfo = (byte) (frameInfo | 2);
87 | break;
88 | case CLOSE:
89 | frameInfo = (byte) (frameInfo | 8);
90 | break;
91 | case CONTINUATION:
92 | frameInfo = (byte) (frameInfo | 0);
93 | break;
94 | case PING:
95 | log.trace("ping out");
96 | frameInfo = (byte) (frameInfo | 9);
97 | break;
98 | case PONG:
99 | log.trace("pong out");
100 | frameInfo = (byte) (frameInfo | 0xa);
101 | break;
102 | default:
103 | break;
104 | }
105 | buffer.put(frameInfo);
106 | // set the frame length
107 | log.trace("Frame length {} ", frameLen);
108 | if (frameLen <= 125) {
109 | buffer.put((byte) ((byte) frameLen & (byte) 0x7F));
110 | } else if (frameLen > 125 && frameLen <= 65535) {
111 | buffer.put((byte) ((byte) 126 & (byte) 0x7F));
112 | buffer.putShort((short) frameLen);
113 | } else {
114 | buffer.put((byte) ((byte) 127 & (byte) 0x7F));
115 | buffer.putLong((int) frameLen);
116 | }
117 | buffer.put(data);
118 | buffer.flip();
119 | if (log.isTraceEnabled()) {
120 | log.trace("Encoded: {}", buffer);
121 | }
122 | return buffer;
123 | }
124 |
125 | }
126 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/codec/extension/PerMessageDeflateExt.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.codec.extension;
20 |
21 | /**
22 | * Extension to WebSocket which provides per-message deflate.
23 | *
24 | * @see IETF Draft
25 | * @see Configuration and optimization
26 | *
27 | * @author Paul Gregoire
28 | */
29 | public class PerMessageDeflateExt implements WebSocketExtension {
30 |
31 | private static final String id = "permessage-deflate";
32 |
33 | /** {@inheritDoc} */
34 | public String getId() {
35 | return id;
36 | }
37 |
38 | }
39 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/codec/extension/WebSocketExtension.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.codec.extension;
20 |
21 | /**
22 | * Common interface for WebSocket extensions.
23 | *
24 | * @author Paul Gregoire
25 | */
26 | public interface WebSocketExtension {
27 |
28 | /**
29 | * Returns the extensions identifying string.
30 | *
31 | * @return id
32 | */
33 | String getId();
34 |
35 | }
36 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/listener/DefaultWebSocketDataListener.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.listener;
20 |
21 | import java.io.UnsupportedEncodingException;
22 | import java.util.Set;
23 |
24 | import org.red5.net.websocket.WebSocketConnection;
25 | import org.red5.net.websocket.WebSocketPlugin;
26 | import org.red5.net.websocket.WebSocketScope;
27 | import org.red5.net.websocket.WebSocketScopeManager;
28 | import org.red5.net.websocket.model.WSMessage;
29 | import org.red5.server.plugin.PluginRegistry;
30 | import org.slf4j.Logger;
31 | import org.slf4j.LoggerFactory;
32 |
33 | /**
34 | * Default WebSocket data listener. In this default implementation, all messages are echoed back to every connection in the current scope.
35 | *
36 | * @author Paul Gregoire (mondain@gmail.com)
37 | */
38 | public class DefaultWebSocketDataListener extends WebSocketDataListener {
39 |
40 | private static final Logger log = LoggerFactory.getLogger(DefaultWebSocketDataListener.class);
41 |
42 | @Override
43 | public void onWSConnect(WebSocketConnection conn) {
44 | log.info("Connect: {}", conn);
45 | }
46 |
47 | @Override
48 | public void onWSDisconnect(WebSocketConnection conn) {
49 | log.info("Disconnect: {}", conn);
50 | }
51 |
52 | @Override
53 | public void onWSMessage(WSMessage message) {
54 | // assume we have text
55 | String msg = new String(message.getPayload().array());
56 | log.info("onWSMessage: {}", msg);
57 | // get the path
58 | String path = message.getPath();
59 | // just echo back the message
60 | WebSocketScopeManager manager = ((WebSocketPlugin) PluginRegistry.getPlugin("WebSocketPlugin")).getManager(path);
61 | if (manager != null) {
62 | // get the ws scope
63 | WebSocketScope wsScope = manager.getScope(path);
64 | Set conns = wsScope.getConns();
65 | for (WebSocketConnection conn : conns) {
66 | log.debug("Echoing to {}", conn);
67 | try {
68 | conn.send(msg);
69 | } catch (UnsupportedEncodingException e) {
70 | log.warn("Encoding issue with the message data: {}", message, e);
71 | }
72 | }
73 | } else {
74 | log.info("No manager found for path: {}", path);
75 | }
76 | }
77 |
78 | @Override
79 | public void stop() {
80 | log.info("Stop");
81 | }
82 |
83 | }
84 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/listener/IWebSocketDataListener.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.listener;
20 |
21 | import org.red5.net.websocket.WebSocketConnection;
22 | import org.red5.net.websocket.model.WSMessage;
23 |
24 | /**
25 | * Listener for WebSocket events.
26 | */
27 | public interface IWebSocketDataListener {
28 |
29 | /**
30 | * Returns the protocol for which this listener is interested.
31 | *
32 | * @return protocol
33 | */
34 | public String getProtocol();
35 |
36 | /**
37 | * Sets the protocol for which this listener is interested.
38 | *
39 | * @param protocol
40 | */
41 | public void setProtocol(String protocol);
42 |
43 | /**
44 | * Dispatch message.
45 | *
46 | * @param message
47 | */
48 | public void onWSMessage(WSMessage message);
49 |
50 | /**
51 | * Connect a WebSocket client.
52 | *
53 | * @param conn
54 | * WebSocketConnection
55 | */
56 | public void onWSConnect(WebSocketConnection conn);
57 |
58 | /**
59 | * Disconnect WebSocket client.
60 | *
61 | * @param conn
62 | * WebSocketConnection
63 | */
64 | public void onWSDisconnect(WebSocketConnection conn);
65 |
66 | /**
67 | * Stops the listener.
68 | */
69 | public void stop();
70 |
71 | }
72 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/listener/IWebSocketScopeListener.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2016 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.listener;
20 |
21 | import org.red5.net.websocket.WebSocketScope;
22 |
23 | public interface IWebSocketScopeListener {
24 |
25 | void scopeCreated(WebSocketScope wsScope);
26 |
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/listener/WebSocketDataListener.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.listener;
20 |
21 | /**
22 | * Adapter class for WebSocket data listener interface.
23 | *
24 | * @author Paul Gregoire
25 | */
26 | public abstract class WebSocketDataListener implements IWebSocketDataListener {
27 |
28 | /**
29 | * The protocol which this listener is interested in handling.
30 | */
31 | protected String protocol = "undefined";
32 |
33 | /** {@inheritDoc} */
34 | @Override
35 | public String getProtocol() {
36 | return protocol;
37 | }
38 |
39 | /** {@inheritDoc} */
40 | @Override
41 | public void setProtocol(String protocol) {
42 | this.protocol = protocol;
43 | }
44 |
45 | }
46 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/ConnectionType.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | public enum ConnectionType {
22 |
23 | DIRECT, WEB;
24 |
25 | }
26 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/HandshakeRequest.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | import org.apache.mina.core.buffer.IoBuffer;
22 |
23 | /**
24 | * Wraps around a string that represents a websocket handshake request from the browser to the server.
25 | *
26 | * @author Paul Gregoire
27 | */
28 | public class HandshakeRequest {
29 |
30 | private final IoBuffer req;
31 |
32 | public HandshakeRequest(IoBuffer req) {
33 | this.req = req;
34 | req.flip();
35 | }
36 |
37 | public IoBuffer getRequest() {
38 | return req;
39 | }
40 |
41 | }
42 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/HandshakeResponse.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | import org.apache.mina.core.buffer.IoBuffer;
22 |
23 | /**
24 | * Wraps around a string that represents a websocket handshake response from the server to the browser.
25 | *
26 | * @author Dhruv Chopra
27 | * @author Paul Gregoire
28 | */
29 | public class HandshakeResponse {
30 |
31 | private final IoBuffer response;
32 |
33 | public HandshakeResponse(IoBuffer response) {
34 | this.response = response;
35 | response.flip();
36 | }
37 |
38 | public IoBuffer getResponse() {
39 | return response;
40 | }
41 |
42 | }
43 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/MessageType.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | public enum MessageType {
22 |
23 | CONTINUATION((byte) 0), TEXT((byte) 1), BINARY((byte) 2), CLOSE((byte) 8), PING((byte) 9), PONG((byte) 0xa);
24 |
25 | private byte type;
26 |
27 | MessageType(byte type) {
28 | this.type = type;
29 | }
30 |
31 | byte getType() {
32 | return type;
33 | }
34 |
35 | }
36 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/Packet.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | import org.apache.mina.core.buffer.IoBuffer;
22 |
23 | /**
24 | * Defines the class whose objects are understood by websocket encoder.
25 | *
26 | * @author Dhruv Chopra
27 | * @author Paul Gregoire
28 | */
29 | public class Packet {
30 |
31 | private final IoBuffer data;
32 |
33 | private final MessageType type;
34 |
35 | private Packet(byte[] buf) {
36 | this.data = IoBuffer.wrap(buf);
37 | this.type = MessageType.BINARY;
38 | }
39 |
40 | private Packet(byte[] buf, MessageType type) {
41 | this.data = IoBuffer.wrap(buf);
42 | this.type = type;
43 | }
44 |
45 | /**
46 | * Returns the data.
47 | *
48 | * @return data
49 | */
50 | public IoBuffer getData() {
51 | return data;
52 | }
53 |
54 | /**
55 | * Returns the message type.
56 | *
57 | * @return type
58 | */
59 | public MessageType getType() {
60 | return type;
61 | }
62 |
63 | /**
64 | * Builds the packet which just wraps the IoBuffer.
65 | *
66 | * @param buf
67 | * @return packet
68 | */
69 | public static Packet build(byte[] buf) {
70 | return new Packet(buf);
71 | }
72 |
73 | /**
74 | * Builds the packet which just wraps the IoBuffer.
75 | *
76 | * @param buffer
77 | * @return packet
78 | */
79 | public static Packet build(IoBuffer buffer) {
80 | if (buffer.hasArray()) {
81 | return new Packet(buffer.array());
82 | }
83 | byte[] buf = new byte[buffer.remaining()];
84 | buffer.get(buf);
85 | return new Packet(buf);
86 | }
87 |
88 | public static Packet build(MessageType type) {
89 | return new Packet(new byte[0], type);
90 | }
91 |
92 | public static Packet build(byte[] bytes, MessageType type) {
93 | return new Packet(bytes, type);
94 | }
95 |
96 | }
97 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/WSMessage.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | import java.io.UnsupportedEncodingException;
22 | import java.lang.ref.WeakReference;
23 |
24 | import org.apache.mina.core.buffer.IoBuffer;
25 | import org.red5.net.websocket.WebSocketConnection;
26 |
27 | /**
28 | * Represents incoming WebSocket data which has been decoded.
29 | *
30 | * @author Paul Gregoire (mondain@gmail.com)
31 | */
32 | public class WSMessage {
33 |
34 | // message type
35 | private MessageType messageType;
36 |
37 | // the originating connection for this message
38 | private WeakReference connection;
39 |
40 | // payload
41 | private IoBuffer payload;
42 |
43 | // the path on which this message originated
44 | private String path;
45 |
46 | // creation time
47 | private long timeStamp = System.currentTimeMillis();
48 |
49 | /**
50 | * Returns the payload data as a UTF8 string.
51 | *
52 | * @return string
53 | * @throws UnsupportedEncodingException
54 | */
55 | public String getMessageAsString() throws UnsupportedEncodingException {
56 | return new String(payload.array(), "UTF8").trim();
57 | }
58 |
59 | public MessageType getMessageType() {
60 | return messageType;
61 | }
62 |
63 | public void setMessageType(MessageType messageType) {
64 | this.messageType = messageType;
65 | }
66 |
67 | public WebSocketConnection getConnection() {
68 | return connection.get();
69 | }
70 |
71 | public void setConnection(WebSocketConnection connection) {
72 | this.connection = new WeakReference(connection);
73 | // set the connections path on the message
74 | setPath(connection.getPath());
75 | }
76 |
77 | /**
78 | * Returns the payload.
79 | *
80 | * @return payload
81 | */
82 | public IoBuffer getPayload() {
83 | return payload.flip();
84 | }
85 |
86 | public void setPayload(IoBuffer payload) {
87 | this.payload = payload;
88 | }
89 |
90 | /**
91 | * Adds additional payload data.
92 | *
93 | * @param additionalPayload
94 | */
95 | public void addPayload(IoBuffer additionalPayload) {
96 | if (payload == null) {
97 | payload = IoBuffer.allocate(additionalPayload.remaining());
98 | payload.setAutoExpand(true);
99 | }
100 | this.payload.put(additionalPayload);
101 | }
102 |
103 | /**
104 | * Adds additional payload data.
105 | *
106 | * @param additionalPayload
107 | */
108 | public void addPayload(byte[] additionalPayload) {
109 | if (payload == null) {
110 | payload = IoBuffer.allocate(additionalPayload.length);
111 | payload.setAutoExpand(true);
112 | }
113 | this.payload.put(additionalPayload);
114 | }
115 |
116 | public boolean isPayloadComplete() {
117 | return !payload.hasRemaining();
118 | }
119 |
120 | public long getTimeStamp() {
121 | return timeStamp;
122 | }
123 |
124 | public String getPath() {
125 | return path;
126 | }
127 |
128 | public void setPath(String path) {
129 | this.path = path;
130 | }
131 |
132 | @Override
133 | public String toString() {
134 | return "WSMessage [messageType=" + messageType + ", timeStamp=" + timeStamp + ", path=" + path + ", payload=" + payload + "]";
135 | }
136 |
137 | }
138 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/model/WebSocketEvent.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.model;
20 |
21 | /**
22 | * WebSocket event enumeration.
23 | *
24 | * @author Paul Gregoire
25 | */
26 | public enum WebSocketEvent {
27 |
28 | SCOPE_CREATED, SCOPE_ADDED, SCOPE_REMOVED;
29 |
30 | }
31 |
--------------------------------------------------------------------------------
/src/main/java/org/red5/net/websocket/util/IdGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * RED5 Open Source Flash Server - https://github.com/red5
3 | *
4 | * Copyright 2006-2015 by respective authors (see below). All rights reserved.
5 | *
6 | * Licensed under the Apache License, Version 2.0 (the "License");
7 | * you may not use this file except in compliance with the License.
8 | * You may obtain a copy of the License at
9 | *
10 | * http://www.apache.org/licenses/LICENSE-2.0
11 | *
12 | * Unless required by applicable law or agreed to in writing, software
13 | * distributed under the License is distributed on an "AS IS" BASIS,
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | * See the License for the specific language governing permissions and
16 | * limitations under the License.
17 | */
18 |
19 | package org.red5.net.websocket.util;
20 |
21 | import java.util.concurrent.ThreadLocalRandom;
22 |
23 | import org.bouncycastle.crypto.digests.SHA1Digest;
24 | import org.bouncycastle.crypto.prng.DigestRandomGenerator;
25 |
26 | /**
27 | * Id generator.
28 | *
29 | * @author Paul Gregoire
30 | */
31 | public class IdGenerator {
32 |
33 | private static final DigestRandomGenerator random = new DigestRandomGenerator(new SHA1Digest());
34 |
35 | /**
36 | * Returns a cryptographically generated id.
37 | *
38 | * @return id
39 | */
40 | public static final long generateId() {
41 | long id = 0;
42 | // add new seed material from current time
43 | random.addSeedMaterial(ThreadLocalRandom.current().nextLong());
44 | // get a new id
45 | byte[] bytes = new byte[16];
46 | // get random bytes
47 | random.nextBytes(bytes);
48 | for (int i = 0; i < bytes.length; i++) {
49 | id += ((long) bytes[i] & 0xffL) << (8 * i);
50 | }
51 | //System.out.println("Id: " + id);
52 | return id;
53 | }
54 |
55 | }
56 |
--------------------------------------------------------------------------------
/src/main/resources/logback-websocket.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 | websocket
4 |
5 |
6 | log/websocket.log
7 | false
8 | UTF-8
9 | false
10 | true
11 |
12 |
13 | %date [%thread] %-5level %logger{35} - %msg%n
14 |
15 |
16 |
17 |
18 |
19 |
20 |
21 |
22 |
--------------------------------------------------------------------------------
/src/test/resources/logback-test.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 | [%p] [%thread] %logger - %msg%n
6 |
7 |
8 |
9 |
10 | target/test.log
11 | false
12 |
13 | %d{ISO8601} [%thread] %-5level %logger{35} - %msg%n
14 |
15 |
16 |
17 |
18 |
19 |
20 |
21 |
22 |
23 |
24 |
--------------------------------------------------------------------------------