src/main/java/net/sourceforge/guacamole/net/basic/BasicGuacamoleTunnelServlet.java

   1 package net.sourceforge.guacamole.net.basic;
   2
   3 /*
   4  *  Guacamole - Clientless Remote Desktop
   5  *  Copyright (C) 2010  Michael Jumper
   6  *
   7  *  This program is free software: you can redistribute it and/or modify
   8  *  it under the terms of the GNU Affero General Public License as published by
   9  *  the Free Software Foundation, either version 3 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU Affero General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Affero General Public License
  18  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 import javax.servlet.ServletException;
  22 import javax.servlet.http.HttpServletRequest;
  23 import javax.servlet.http.HttpSession;
  24 import net.sourceforge.guacamole.GuacamoleException;
  25 import net.sourceforge.guacamole.net.InetGuacamoleSocket;
  26 import net.sourceforge.guacamole.protocol.GuacamoleConfiguration;
  27 import net.sourceforge.guacamole.properties.GuacamoleProperties;
  28 import net.sourceforge.guacamole.net.GuacamoleSocket;
  29 import net.sourceforge.guacamole.servlet.GuacamoleSession;
  30 import net.sourceforge.guacamole.net.GuacamoleTunnel;
  31 import net.sourceforge.guacamole.net.basic.properties.BasicGuacamoleProperties;
  32 import net.sourceforge.guacamole.protocol.ConfiguredGuacamoleSocket;
  33 import net.sourceforge.guacamole.servlet.GuacamoleTunnelServlet;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 public class BasicGuacamoleTunnelServlet extends GuacamoleTunnelServlet {
  38
  39     private Logger logger = LoggerFactory.getLogger(BasicGuacamoleTunnelServlet.class);
  40
  41     private AuthenticationProvider authProvider;
  42
  43     @Override
  44     public void init() throws ServletException {
  45
  46         // Get auth provider instance
  47         try {
  48             authProvider = GuacamoleProperties.getProperty(BasicGuacamoleProperties.AUTH_PROVIDER);
  49         }
  50         catch (GuacamoleException e) {
  51             logger.error("Error getting authentication provider from properties.", e);
  52             throw new ServletException(e);
  53         }
  54
  55     }
  56
  57     @Override
  58     protected GuacamoleTunnel doConnect(HttpServletRequest request) throws GuacamoleException {
  59
  60         HttpSession httpSession = request.getSession(true);
  61
  62         // Retrieve username and password from parms
  63         String username = request.getParameter("username");
  64         String password = request.getParameter("password");
  65
  66         // Get authorized config
  67         GuacamoleConfiguration config;
  68         try {
  69             config = authProvider.getAuthorizedConfiguration(username, password);
  70         }
  71         catch (GuacamoleException e) {
  72             logger.error("Error retrieving authorized configuration for user {}.", username);
  73             throw e;
  74         }
  75
  76         if (config == null) {
  77             logger.warn("Failed login from {} for user \"{}\".", request.getRemoteAddr(), username);
  78             throw new GuacamoleException("Invalid login");
  79         }
  80
  81         logger.info("Successful login from {} for user \"{}\".", request.getRemoteAddr(), username);
  82
  83         // Configure and connect socket
  84         String hostname = GuacamoleProperties.getProperty(GuacamoleProperties.GUACD_HOSTNAME);
  85         int port = GuacamoleProperties.getProperty(GuacamoleProperties.GUACD_PORT);
  86
  87         GuacamoleSocket socket = new ConfiguredGuacamoleSocket(
  88                 new InetGuacamoleSocket(hostname, port),
  89                 config
  90         );
  91
  92         // Associate socket with tunnel
  93         GuacamoleTunnel tunnel = new GuacamoleTunnel(socket);
  94
  95         // Attach tunnel to session
  96         GuacamoleSession session = new GuacamoleSession(httpSession);
  97         session.attachTunnel(tunnel);
  98
  99         return tunnel;
 100
 101     }
 102
 103 }
 104