io.antmedia.security.AcceptOnlyStreamsInDataStore Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ant-media-server Show documentation

Ant Media Server supports RTMP, RTSP, MP4, HLS, WebRTC, Adaptive Streaming, etc.

There is a newer version: 2.10.0

package io.antmedia.security;

import java.util.Map;

import org.red5.server.api.Red5;
import org.red5.server.api.scope.IScope;
import org.red5.server.api.stream.IStreamPublishSecurity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.antmedia.datastore.db.DataStoreFactory;
import io.antmedia.datastore.db.DataStore;
import io.antmedia.datastore.db.types.Broadcast;

public class AcceptOnlyStreamsInDataStore implements IStreamPublishSecurity  {
	
	private DataStoreFactory dataStoreFactory;
	private DataStore dataStore;
	
	private boolean enabled = true;
	
	public static final String BEAN_NAME = "acceptOnlyStreamsInDataStore"; 
	
	protected static Logger logger = LoggerFactory.getLogger(AcceptOnlyStreamsInDataStore.class);

	@Override
	public boolean isPublishAllowed(IScope scope, String name, String mode, Map queryParams) {
		
		boolean result = false;
		if (enabled) {
			Broadcast broadcast = getDatastore().get(name);
			if (broadcast != null) 
			{
				result = true;
			}
			else {
				logger.info("No stream in data store not allowing the stream {}", name);
				Red5.getConnectionLocal().close();
			}
		}
		else {
			logger.info("AcceptOnlyStreamsInDataStore is not activated. Accepting all streams {}", name);
			result = true;
		}
		
		
		return result;
	}
	
	
	public DataStore getDatastore() {
		if (dataStore == null) {
			dataStore = dataStoreFactory.getDataStore();
		}
		return dataStore;
	}
	
	
	public void setDataStore(DataStore dataStore) {
		this.dataStore = dataStore;
	}


	public boolean isEnabled() {
		return enabled;
	}


	public void setEnabled(boolean enabled) {
		this.enabled = enabled;
	}

	public DataStoreFactory getDataStoreFactory() {
		return dataStoreFactory;
	}


	public void setDataStoreFactory(DataStoreFactory dataStoreFactory) {
		this.dataStoreFactory = dataStoreFactory;
	}

	
}