org.red5.server.Shutdown Maven / Gradle / Ivy

/*
 * RED5 Open Source Media Server - https://github.com/Red5/
 * 
 * Copyright 2006-2016 by respective authors (see below). All rights reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.red5.server;

import java.io.BufferedReader;
import java.io.File;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.RandomAccessFile;
import java.net.Socket;
import java.nio.file.Paths;

/**
 * Provides a means to cleanly shutdown an instance from the command line.
 *
 * @author The Red5 Project
 * @author Paul Gregoire ([email protected])
 */
public class Shutdown {

    /**
     * Connects to the given port (default: 9999) and invokes shutdown.
     * 
     * Arg 0 = port number
     * Arg 1 = token
     * 
     * 
     * @param args
     *            see args list
     */
    public static void main(String[] args) {
        String host = System.getProperty("red5.shutdown.host", "127.0.0.1");
        try (
                Socket clientSocket = new Socket(host, Integer.valueOf(args[0]));
                PrintWriter out = new PrintWriter(clientSocket.getOutputStream(), true);
                BufferedReader in = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
            ) {
            // send the token
            String token = "cafebeef";
            if (args.length > 1) {
                token = args[1];
            } else {
                // read the token from the file
            	token = getToken(token);
            }
            out.println(token);
            in.readLine(); //wait for server to stop
        } catch (Exception e) {
            System.err.printf("Exception connecting to %s%n", host);
            e.printStackTrace();
            System.exit(1);
        }
    }

	public static String getToken(String token) {
		File tokenFile = Paths.get("shutdown.token").toFile();
		try (RandomAccessFile raf = new RandomAccessFile(tokenFile, "r")) {
		    byte[] buf = new byte[36];
		    raf.readFully(buf);
		    token = new String(buf);
		    System.out.printf("Token loaded: %s%n", token);
		} catch (Exception e) {
		    e.printStackTrace();
		}
		return token;
	}
}