92 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| /*
 | |
|  * Pterodactyl - Panel
 | |
|  * Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
 | |
|  *
 | |
|  * Permission is hereby granted, free of charge, to any person obtaining a copy
 | |
|  * of this software and associated documentation files (the "Software"), to deal
 | |
|  * in the Software without restriction, including without limitation the rights
 | |
|  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 | |
|  * copies of the Software, and to permit persons to whom the Software is
 | |
|  * furnished to do so, subject to the following conditions:
 | |
|  *
 | |
|  * The above copyright notice and this permission notice shall be included in all
 | |
|  * copies or substantial portions of the Software.
 | |
|  *
 | |
|  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | |
|  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | |
|  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 | |
|  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | |
|  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 | |
|  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 | |
|  * SOFTWARE.
 | |
|  */
 | |
| 
 | |
| namespace Pterodactyl\Http\Middleware\Daemon;
 | |
| 
 | |
| use Closure;
 | |
| use Illuminate\Http\Request;
 | |
| use Symfony\Component\HttpKernel\Exception\HttpException;
 | |
| use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
 | |
| use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
 | |
| use Symfony\Component\HttpKernel\Exception\AccessDeniedHttpException;
 | |
| 
 | |
| class DaemonAuthenticate
 | |
| {
 | |
|     /**
 | |
|      * @var \Pterodactyl\Contracts\Repository\NodeRepositoryInterface
 | |
|      */
 | |
|     private $repository;
 | |
| 
 | |
|     /**
 | |
|      * Daemon routes that this middleware should be skipped on.
 | |
|      *
 | |
|      * @var array
 | |
|      */
 | |
|     protected $except = [
 | |
|         'daemon.configuration',
 | |
|     ];
 | |
| 
 | |
|     /**
 | |
|      * DaemonAuthenticate constructor.
 | |
|      *
 | |
|      * @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $repository
 | |
|      */
 | |
|     public function __construct(NodeRepositoryInterface $repository)
 | |
|     {
 | |
|         $this->repository = $repository;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Check if a request from the daemon can be properly attributed back to a single node instance.
 | |
|      *
 | |
|      * @param \Illuminate\Http\Request $request
 | |
|      * @param \Closure                 $next
 | |
|      * @return mixed
 | |
|      *
 | |
|      * @throws \Symfony\Component\HttpKernel\Exception\HttpException
 | |
|      */
 | |
|     public function handle(Request $request, Closure $next)
 | |
|     {
 | |
|         if (in_array($request->route()->getName(), $this->except)) {
 | |
|             return $next($request);
 | |
|         }
 | |
| 
 | |
|         $token = $request->bearerToken();
 | |
| 
 | |
|         if (is_null($token)) {
 | |
|             throw new HttpException(401, null, null, ['WWW-Authenticate' => 'Bearer']);
 | |
|         }
 | |
| 
 | |
|         try {
 | |
|             $node = $this->repository->findFirstWhere([['daemonSecret', '=', $token]]);
 | |
|         } catch (RecordNotFoundException $exception) {
 | |
|             throw new AccessDeniedHttpException;
 | |
|         }
 | |
| 
 | |
|         $request->attributes->set('node', $node);
 | |
| 
 | |
|         return $next($request);
 | |
|     }
 | |
| }
 | 
