792f606a64
Contains the cql3 grammer and supporting classes, which will be converted one by one. From commit bf599fb5b062cbcc652da78b7d699e7a01b949ad.
73 lines
2.9 KiB
Java
73 lines
2.9 KiB
Java
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you 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.apache.cassandra.cql3.statements;
|
|
|
|
import org.apache.cassandra.auth.Auth;
|
|
import org.apache.cassandra.auth.AuthenticatedUser;
|
|
import org.apache.cassandra.config.DatabaseDescriptor;
|
|
import org.apache.cassandra.exceptions.InvalidRequestException;
|
|
import org.apache.cassandra.exceptions.RequestExecutionException;
|
|
import org.apache.cassandra.exceptions.RequestValidationException;
|
|
import org.apache.cassandra.exceptions.UnauthorizedException;
|
|
import org.apache.cassandra.service.ClientState;
|
|
import org.apache.cassandra.transport.messages.ResultMessage;
|
|
|
|
public class DropUserStatement extends AuthenticationStatement
|
|
{
|
|
private final String username;
|
|
private final boolean ifExists;
|
|
|
|
public DropUserStatement(String username, boolean ifExists)
|
|
{
|
|
this.username = username;
|
|
this.ifExists = ifExists;
|
|
}
|
|
|
|
public void validate(ClientState state) throws RequestValidationException
|
|
{
|
|
// validate login here before checkAccess to avoid leaking user existence to anonymous users.
|
|
state.ensureNotAnonymous();
|
|
|
|
if (!ifExists && !Auth.isExistingUser(username))
|
|
throw new InvalidRequestException(String.format("User %s doesn't exist", username));
|
|
|
|
AuthenticatedUser user = state.getUser();
|
|
if (user != null && user.getName().equals(username))
|
|
throw new InvalidRequestException("Users aren't allowed to DROP themselves");
|
|
}
|
|
|
|
public void checkAccess(ClientState state) throws UnauthorizedException
|
|
{
|
|
if (!state.getUser().isSuper())
|
|
throw new UnauthorizedException("Only superusers are allowed to perform DROP USER queries");
|
|
}
|
|
|
|
public ResultMessage execute(ClientState state) throws RequestValidationException, RequestExecutionException
|
|
{
|
|
// not rejected in validate()
|
|
if (ifExists && !Auth.isExistingUser(username))
|
|
return null;
|
|
|
|
// clean up permissions after the dropped user.
|
|
DatabaseDescriptor.getAuthorizer().revokeAll(username);
|
|
Auth.deleteUser(username);
|
|
DatabaseDescriptor.getAuthenticator().drop(username);
|
|
return null;
|
|
}
|
|
}
|