/* This file is part of VoltDB. * Copyright (C) 2008-2017 VoltDB Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with VoltDB. If not, see <http://www.gnu.org/licenses/>. */ package org.voltdb; import org.voltdb.AuthSystem.AuthUser; import org.voltdb.catalog.Procedure; /** * Check ad hoc query parameters. */ public class AdHocAcceptancePolicy extends InvocationValidationPolicy { public AdHocAcceptancePolicy(boolean isOn) { super(isOn); } @Override public ClientResponseImpl shouldAccept(AuthUser user, StoredProcedureInvocation invocation, Procedure sysProc) { final String procName = invocation.getProcName(); if ( ! procName.equals("@AdHoc") && ! procName.equals("@AdHocSpForTest")) { return null; } ParameterSet params = invocation.getParams(); final Object [] paramsArray = params.toArray(); // Make sure there is at least 1 parameter! ENG-4921 if (paramsArray.length < 1) { return new ClientResponseImpl(ClientResponseImpl.GRACEFUL_FAILURE, new VoltTable[0], "Adhoc system procedure requires at least the query parameter.", invocation.clientHandle); } // check the types are both strings if ((paramsArray[0] instanceof String) == false) { return new ClientResponseImpl(ClientResponseImpl.GRACEFUL_FAILURE, new VoltTable[0], "Adhoc system procedure requires the query parameter to be of String type.", invocation.clientHandle); } return null; } }