001 /**
002 * GRANITE DATA SERVICES
003 * Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 * This file is part of the Granite Data Services Platform.
006 *
007 * Granite Data Services is free software; you can redistribute it and/or
008 * modify it under the terms of the GNU Lesser General Public
009 * License as published by the Free Software Foundation; either
010 * version 2.1 of the License, or (at your option) any later version.
011 *
012 * Granite Data Services is distributed in the hope that it will be useful,
013 * but WITHOUT ANY WARRANTY; without even the implied warranty of
014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015 * General Public License for more details.
016 *
017 * You should have received a copy of the GNU Lesser General Public
018 * License along with this library; if not, write to the Free Software
019 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020 * USA, or see <http://www.gnu.org/licenses/>.
021 */
022 package org.granite.gravity.websocket;
023
024 import java.io.IOException;
025 import java.nio.ByteBuffer;
026 import java.nio.channels.ClosedByInterruptException;
027 import java.nio.channels.ServerSocketChannel;
028 import java.nio.channels.SocketChannel;
029
030 import org.granite.logging.Logger;
031
032
033 public class PolicyFileServer implements Runnable {
034
035 private static final Logger log = Logger.getLogger(PolicyFileServer.class);
036
037
038 private int serverPort = 843;
039 private String[] allowDomains = {};
040 private String[] allowPorts = {};
041 private Thread policyServer = null;
042
043 public void setServerPort(int serverPort) {
044 this.serverPort = serverPort;
045 }
046
047 public void setAllowDomains(String[] domains) {
048 this.allowDomains = domains;
049 }
050
051 public void setAllowPorts(String[] ports) {
052 this.allowPorts = ports;
053 }
054
055 public void start() {
056 policyServer = new Thread(this, "FlashPolicyFileServer:" + serverPort);
057 policyServer.start();
058 }
059
060 public void stop() {
061 if (policyServer != null)
062 policyServer.interrupt();
063 }
064
065 public void run() {
066 ServerSocketChannel server = null;
067 try {
068 server = ServerSocketChannel.open();
069 server.socket().bind(new java.net.InetSocketAddress(serverPort));
070 log.info("Flash socket policy server started on port " + serverPort);
071 }
072 catch (IOException e) {
073 log.error(e, "Could not init flash socket policy server on port " + serverPort);
074 return;
075 }
076 while (true) {
077 SocketChannel socket = null;
078 try {
079 socket = server.accept();
080
081 ByteBuffer buf = ByteBuffer.allocate(100);
082 int size = socket.read(buf);
083 if (size == 23) {
084 byte[] req = new byte[size];
085 buf.get(req, 0, size);
086 String request = new String(req, "UTF-8");
087
088 log.info("Received policy file request %s", request);
089
090 String policyFile = "<?xml version=\"1.0\"?>\n"
091 + "<!DOCTYPE cross-domain-policy SYSTEM \"http://www.macromedia.com/xml/dtds/cross-domain-policy.dtd\">"
092 + "<cross-domain-policy>";
093 for (int i = 0; i < allowDomains.length; i++)
094 policyFile += " <allow-access-from domain=\"" + allowDomains[i] + "\" to-ports=\"" + allowPorts[i] + "\"/>";
095 policyFile += "</cross-domain-policy>";
096 byte[] bytes = policyFile.getBytes("UTF-8");
097
098 socket.write(ByteBuffer.wrap(bytes));
099 }
100 }
101 catch (ClosedByInterruptException e) {
102 log.info("Flash socket policy server stopped");
103 break;
104 }
105 catch (IOException e) {
106 log.error(e, "Could not send policy file");
107 }
108 finally {
109 if (socket != null) {
110 try {
111 socket.close();
112 }
113 catch (IOException e) {
114 log.error(e, "Could not close socket");
115 }
116 }
117 }
118 }
119 }
120
121 }