io.antmedia.security.ExpireStreamPublishSecurity 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 ExpireStreamPublishSecurity implements IStreamPublishSecurity {

	private DataStoreFactory dataStoreFactory;
	private DataStore dataStore;


	protected static Logger logger = LoggerFactory.getLogger(ExpireStreamPublishSecurity.class);

	@Override
	public boolean isPublishAllowed(IScope scope, String name, String mode, Map queryParams) {

		boolean result = false;

		Broadcast broadcast = getDatastore().get(name);
		if (broadcast != null) 
		{
			int expireDurationMS = broadcast.getExpireDurationMS();

			if (expireDurationMS != 0) 
			{
				if (System.currentTimeMillis() < (broadcast.getDate() + expireDurationMS)) {
					result = true;
				}
				else {
					logger.info("Not allowing the stream "+ broadcast.getStreamId() +" to publish. It is expired.");
				}
			}
			else {
				result = true;
			}
		}
		else {
			result = true;
		}
		
		if (!result) {
			Red5.getConnectionLocal().close();
		}

		return result;
	}


	public DataStore getDatastore() {
		if (dataStore == null) {
			dataStore = dataStoreFactory.getDataStore();
		}
		return dataStore;
	}


	public void setDataStore(DataStore dataStore) {
		this.dataStore = dataStore;
	}

	public DataStoreFactory getDataStoreFactory() {
		return dataStoreFactory;
	}


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

}