All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.floragunn.searchguard.configuration.PrivilegesInterceptorImpl Maven / Gradle / Ivy

The newest version!
/*
 * Copyright 2017 by floragunn GmbH - All rights reserved
 * 
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed here is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * 
 * This software is free of charge for non-commercial and academic use.
 * For commercial use in a production environment you have to obtain a license
 * from https://floragunn.com
 * 
 */

package com.floragunn.searchguard.configuration;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.elasticsearch.ElasticsearchException;
import org.elasticsearch.action.ActionRequest;
import org.elasticsearch.action.CompositeIndicesRequest;
import org.elasticsearch.action.DocWriteRequest;
import org.elasticsearch.action.IndicesRequest.Replaceable;
import org.elasticsearch.action.admin.indices.create.CreateIndexRequest;
import org.elasticsearch.action.admin.indices.mapping.get.GetFieldMappingsIndexRequest;
import org.elasticsearch.action.admin.indices.mapping.get.GetFieldMappingsRequest;
import org.elasticsearch.action.admin.indices.refresh.RefreshRequest;
import org.elasticsearch.action.bulk.BulkRequest;
import org.elasticsearch.action.delete.DeleteRequest;
import org.elasticsearch.action.get.MultiGetRequest;
import org.elasticsearch.action.get.MultiGetRequest.Item;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.search.MultiSearchRequest;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.support.IndicesOptions;
import org.elasticsearch.action.support.replication.ReplicationRequest;
import org.elasticsearch.action.support.single.shard.SingleShardRequest;
import org.elasticsearch.action.termvectors.MultiTermVectorsRequest;
import org.elasticsearch.action.termvectors.TermVectorsRequest;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.client.Client;
import org.elasticsearch.cluster.metadata.IndexNameExpressionResolver;
import org.elasticsearch.cluster.service.ClusterService;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.threadpool.ThreadPool;

import com.floragunn.searchguard.configuration.PrivilegesEvaluator.IndexType;
import com.floragunn.searchguard.user.User;

public class PrivilegesInterceptorImpl extends PrivilegesInterceptor {

    private final static IndicesOptions DEFAULT_INDICES_OPTIONS = IndicesOptions.lenientExpandOpen();
    private static final String USER_TENANT = "__user__";
    private static final String EMPTY_STRING = "";

    protected final Logger log = LogManager.getLogger(this.getClass());
    
    public PrivilegesInterceptorImpl(IndexNameExpressionResolver resolver, ClusterService clusterService, Client client,
            ThreadPool threadPool) {
        super(resolver, clusterService, client, threadPool);
    }
    
    private boolean isTenantAllowed(final ActionRequest request, final String action, final User user, final Map tenants, final String requestedTenant) {
        
        if (!tenants.keySet().contains(requestedTenant)) {
            log.warn("Tenant {} is not allowed for user {}", requestedTenant, user.getName());
            return false;
        } else {
            // allowed, check read-write permissions
            boolean isBuildNumRequest = false;
            
            if(log.isDebugEnabled()) {
                log.debug("request "+request.getClass());
            }
            
            if (request instanceof IndexRequest) {

                final IndexRequest ir = ((IndexRequest) request);

                if (log.isDebugEnabled()) {
                    log.debug("type " + ir.type());
                    log.debug("id " + ir.id());
                    log.debug("source " + (ir.source() == null ? null : ir.source().utf8ToString()));
                }

                /*if (ir.type().equals("config") 
                        && Character.isDigit(ir.id().charAt(0)) 
                        && ir.source().toUtf8().contains("buildNum")) {
                    isBuildNumRequest = true;
                }*/
            }
            
            if (request instanceof UpdateRequest) {

                final UpdateRequest ir = ((UpdateRequest) request);

                if (log.isDebugEnabled()) {
                    log.debug("type " + ir.type());
                    log.debug("id " + ir.id());
                    log.debug("source " + (ir.doc() == null ? null : ir.doc().source()==null?null:ir.doc().source().utf8ToString()));
                }
            }
            
            if (!isBuildNumRequest && tenants.get(requestedTenant) == Boolean.FALSE 
                    && action.startsWith("indices:data/write")) {
                log.warn("Tenant {} is not allowed to write (user: {})", requestedTenant, user.getName());
                return false;
            }
        }
        
        return true;
    }
    
    /**
     * return Boolean.TRUE to prematurely deny request
     * return Boolean.FALSE to prematurely allow request
     * return null to go through original eval flow
     *
     */
    @Override
    public Boolean replaceKibanaIndex(final ActionRequest request, final String action, final User user, final Settings config, final Set requestedResolvedIndices, final Map tenants) { 
        
        final boolean enabled = config.getAsBoolean("searchguard.dynamic.kibana.multitenancy_enabled", true);
        
        if(!enabled) {
            return null;
        }
        
        //next two lines needs to be retrieved from configuration
        final String kibanaserverUsername = config.get("searchguard.dynamic.kibana.server_username","kibanaserver");
        final String kibanaIndexName = config.get("searchguard.dynamic.kibana.index",".kibana");

        String requestedTenant = user.getRequestedTenant();
        
        if(log.isDebugEnabled()) {
            log.debug("raw requestedTenant: '"+requestedTenant+"'");
        }
        
        if(requestedTenant == null || requestedTenant.length() == 0) {
            if(log.isTraceEnabled()) {
                log.trace("No tenant, will resolve to "+kibanaIndexName);    
            }

            return null;
        }
        
        if(USER_TENANT.equals(requestedTenant)) {
            requestedTenant = user.getName();
        }
        
        if (!user.getName().equals(kibanaserverUsername) 
                && requestedResolvedIndices.size() == 1
                && requestedResolvedIndices.contains(toUserIndexName(kibanaIndexName, requestedTenant))) {
            
            if(isTenantAllowed(request, action, user, tenants, requestedTenant)) {
                return Boolean.FALSE;
            }
            
        }
        
        //intercept when requests are not made by the kibana server and if the kibana index (.kibana) is the only index involved
        if (!user.getName().equals(kibanaserverUsername) 
                && requestedResolvedIndices.contains(kibanaIndexName)
                && requestedResolvedIndices.size() == 1) {
            
            if(log.isDebugEnabled()) {
                log.debug("requestedTenant: "+requestedTenant);
                log.debug("is user tenant: "+requestedTenant.equals(user.getName()));
            }
                        
            if(!isTenantAllowed(request, action, user, tenants, requestedTenant)) {
                return Boolean.TRUE;
            }

            //TODO handle user tenant in that way that this tenant cannot be specified as regular tenant
            //to avoid security issue
            
            replaceIndex(request, kibanaIndexName, toUserIndexName(kibanaIndexName, requestedTenant), action);
            return Boolean.FALSE;

        } else if (!user.getName().equals(kibanaserverUsername)) {

            if (log.isTraceEnabled()) {
                log.trace("not a request to only the .kibana index");
                log.trace(user.getName() + "/" + kibanaserverUsername);
                log.trace(requestedResolvedIndices + " does not contain only " + kibanaIndexName);
            }

        }
        
        return null;
    }
    
    
    private void replaceIndex(final ActionRequest request, final String oldIndexName, final String newIndexName, final String action) {
        boolean kibOk = false;
                
        if(log.isDebugEnabled()) {
            log.debug("{} index will be replaced with {} in this {} request", oldIndexName, newIndexName, request.getClass().getName());
        }
        
        if(request instanceof GetFieldMappingsIndexRequest
                || request instanceof GetFieldMappingsRequest) {
            return;
        }

        //createKibanaUserIndex(oldIndexName, newIndexName, action);
        
        //handle msearch and mget
        //in case of GET change the .kibana index to the userskibanaindex
        //in case of Search add the userskibanaindex
        //if (request instanceof CompositeIndicesRequest) {
        String[] newIndexNames = new String[] { newIndexName };
        
        
        // CreateIndexRequest
        if (request instanceof CreateIndexRequest) {
            ((CreateIndexRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof BulkRequest) {

            for (DocWriteRequest ar : ((BulkRequest) request).requests()) {

                if(ar instanceof DeleteRequest) {
                    ((DeleteRequest) ar).index(newIndexName);
                }
                
                if(ar instanceof IndexRequest) {
                    ((IndexRequest) ar).index(newIndexName);
                }
                
                if(ar instanceof UpdateRequest) {
                    ((UpdateRequest) ar).index(newIndexName);
                }
            }
            
            kibOk = true;

        } else if (request instanceof MultiGetRequest) {

            for (Item item : ((MultiGetRequest) request).getItems()) {
                item.index(newIndexName);
            }
            
            kibOk = true;

        } else if (request instanceof MultiSearchRequest) {

            for (SearchRequest ar : ((MultiSearchRequest) request).requests()) {
                ar.indices(newIndexNames);
            }
            
            kibOk = true;

        } else if (request instanceof MultiTermVectorsRequest) {

            for (TermVectorsRequest ar : (Iterable) () -> ((MultiTermVectorsRequest) request).iterator()) {
                ar.index(newIndexName);
            }

            kibOk = true;
        } else if (request instanceof UpdateRequest) {
            ((UpdateRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof IndexRequest) {
            ((IndexRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof DeleteRequest) {
            ((DeleteRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof SingleShardRequest) {
            ((SingleShardRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof RefreshRequest) {
            (( RefreshRequest) request).indices(newIndexNames); //???
            kibOk = true;
        } else if (request instanceof ReplicationRequest) {
            ((ReplicationRequest) request).index(newIndexName);
            kibOk = true;
        } else if (request instanceof Replaceable) {
            Replaceable replaceableRequest = (Replaceable) request;
            replaceableRequest.indices(newIndexNames);
            kibOk = true;
        } else {
            log.warn("Dont know what to do (1) with {}", request.getClass());
        }

        if (!kibOk) {
            log.warn("Dont know what to do (2) with {}", request.getClass());
        }
    }

    @Override
    public boolean replaceAllowedIndices(final ActionRequest request, final String action, final User user, final Settings config,
            final Map> leftOvers) {

        final boolean enabled = config.getAsBoolean("searchguard.dynamic.kibana.do_not_fail_on_forbidden", false);

        if (!enabled || leftOvers.size() == 0) {
            return false;
        }

        if (!action.startsWith("indices:data/read/") 
                && !action.startsWith("indices:admin/mappings/fields/get")) {
            return false;
        }
        
        Entry> min = null;
        
        //find role with smallest number of leftovers
        //what when two ore more als equal in size??
        
        for(Entry> entry: leftOvers.entrySet()) {
            if(min == null || entry.getValue().size() < min.getValue().size()) {
                min = entry;
            }
        }
        
        if(min == null) {
            log.warn("No valid leftover found");
            return false;
        }

        final Set leftOversIndex = new HashSet();

        for (IndexType indexType: min.getValue()) {
            leftOversIndex.add(indexType.getIndex());
        }

        if(log.isDebugEnabled()) {
            log.debug("handle {}/{} for leftovers {}", action, request.getClass(), leftOversIndex);
        }
        
        if (request instanceof CompositeIndicesRequest) {
            
            if(request instanceof BulkRequest) {

                for(DocWriteRequest ar: ((BulkRequest) request).requests()) {
                    final boolean ok = applyIndexReduce0(ar, action, leftOversIndex);
                    if (!ok) {
                        return false;
                    }
                }
                
            } else if(request instanceof MultiGetRequest) {
                
                for(Item item: ((MultiGetRequest) request).getItems()) {
                    final boolean ok = applyIndexReduce0(item, action, leftOversIndex);
                    if (!ok) {
                        return false;
                    }
                }
                
            } else if(request instanceof MultiSearchRequest) {
                
                for(ActionRequest ar: ((MultiSearchRequest) request).requests()) {
                    final boolean ok = applyIndexReduce0(ar, action, leftOversIndex);
                    if (!ok) {
                        return false;
                    }
                }
                
            } else if(request instanceof MultiTermVectorsRequest) {
                
                for(ActionRequest ar: (Iterable) () -> ((MultiTermVectorsRequest) request).iterator()) {
                    final boolean ok = applyIndexReduce0(ar, action, leftOversIndex);
                    if (!ok) {
                        return false;
                    }
                }
                
            } else if (request instanceof Replaceable) {
                applyIndexReduce0(request, action, leftOversIndex);
            } else {
                log.warn("Can not handle composite request of type '"+request.getClass()+"' here");
            }

            return true;

        } else {
            return applyIndexReduce0(request, action, leftOversIndex);
        }
    }

    private boolean applyIndexReduce0(final Object request, final String action, final Set leftOversIndex) {

        if (request instanceof Replaceable) {
            final Replaceable ir = (Replaceable) request;
            
            if(log.isDebugEnabled()) {
                log.debug("handle Replaceable, indices: {}", Arrays.toString(ir.indices()));
            }
            
            final String[] resolved = resolve(ir.indices(), leftOversIndex);
            
            if(resolved == null) {
                return false;
            }
            
            ir.indices(resolved);
        } else if (request instanceof SingleShardRequest) {
            final SingleShardRequest gr = (SingleShardRequest) request;
            final String[] indices = gr.indices();
            final String index = gr.index();

            final List indicesL = new ArrayList();

            if (index != null) {
                indicesL.add(index);
            }

            if (indices != null && indices.length > 0) {
                indicesL.addAll(Arrays.asList(indices));
            }

            if(log.isDebugEnabled()) {
                log.debug("handle SingleShardRequest, indices: {}", indicesL);
            }
            
            final String[] resolved = resolve(indicesL.toArray(new String[0]), leftOversIndex);
            
            if(resolved == null) {
                return false;
            }
            
            gr.index(resolved[0]);
            
            //if (r.length == 0) {
            //    gr.index(EMPTY_STRING);
            //}

        } else if (request instanceof MultiGetRequest.Item) {
            final MultiGetRequest.Item i = (MultiGetRequest.Item) request;
            
            if(log.isDebugEnabled()) {
                log.debug("handle MultiGetRequest.Item, indices: {}", Arrays.toString(i.indices()));
            }
            
            final String[] resolved = resolve(i.indices(), leftOversIndex);
            
            if(resolved == null) {
                return false;
            }
            
            i.index(resolved[0]);
            
        } else {
            log.error(request.getClass() + " not supported");
            return false;
        }

        return true;
    }
    
    private String[] resolve(final String[] unresolved, final Set leftOversIndex) {

        if (leftOversIndex.contains("*") || leftOversIndex.contains("_all")) {
            
            if(log.isDebugEnabled()) {
                log.debug("resolved {} with {} to [''] because of * leftovers", Arrays.toString(unresolved), leftOversIndex);
            }
            
            return null;
        }

        final String[] concreteIndices = resolver.concreteIndexNames(clusterService.state(), DEFAULT_INDICES_OPTIONS, unresolved);
        final Set survivors = new HashSet(Arrays.asList(concreteIndices));
        survivors.removeAll(leftOversIndex);

        if (survivors.isEmpty()) {
            
            if(log.isDebugEnabled()) {
                log.debug("resolved {} with {} to [''] because of no survivors", Arrays.toString(unresolved), leftOversIndex);
            }
            
            return null;
        }
        
        if(log.isDebugEnabled()) {
            log.debug("resolved {} with {} - survived: {}", Arrays.toString(unresolved), leftOversIndex, survivors);
        }

        return survivors.toArray(new String[0]);
    }


    private String toUserIndexName(final String originalKibanaIndex, final String tenant) {
        
        if(tenant == null) {
            throw new ElasticsearchException("tenant must not be null here");
        }
        
        return originalKibanaIndex+"_"+tenant.hashCode()+"_"+tenant.toLowerCase().replaceAll("[^a-z0-9]+",EMPTY_STRING);
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy