metron-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From ottobackwards <...@git.apache.org>
Subject [GitHub] incubator-metron pull request #439: METRON-571 add stellar external function...
Date Fri, 10 Mar 2017 03:36:56 GMT
Github user ottobackwards commented on a diff in the pull request:

    https://github.com/apache/incubator-metron/pull/439#discussion_r105325302
  
    --- Diff: metron-platform/metron-common/src/main/java/org/apache/metron/common/dsl/ExternalFunctions.java
---
    @@ -0,0 +1,292 @@
    +/**
    + * 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.metron.common.dsl.functions;
    +
    +import java.io.BufferedReader;
    +import java.io.File;
    +import java.io.IOException;
    +import java.io.InputStream;
    +import java.io.InputStreamReader;
    +import java.io.OutputStream;
    +import java.io.PrintWriter;
    +import java.util.List;
    +import java.lang.ProcessBuilder;
    +import java.lang.ClassLoader;
    +import java.lang.reflect.Method;
    +import java.util.Map;
    +import java.util.regex.Pattern;
    +import com.google.common.base.Joiner;
    +import com.google.common.base.Splitter;
    +import com.google.common.collect.Iterables;
    +import org.apache.metron.common.dsl.Context;
    +import org.apache.metron.common.dsl.StellarFunction;
    +import org.apache.metron.common.dsl.ParseException;
    +import org.apache.metron.common.dsl.Stellar;
    +
    +/**
    + * Executes external script on server via stellar process
    + */
    +public class ExternalFunctions {
    +
    +	public static class ExecuteScript implements StellarFunction {
    +
    +        private ThreadedStreamHandler inStream;
    +        private ThreadedStreamHandler errStream;
    +        private boolean isOnTheList = false;
    +
    +        @Stellar(name="EXEC_SCRIPT",
    +                description = "Executes an external shell function via stellar.",
    +                params = {
    +                        "exec - the executing cmd (ie. bash, sh, python)",
    +                        "name - name of the script, located in /scripts " +
    +                                "Do NOT include any special chars except(_), Do include
file extension"
    +                },
    +                returns = "the return value of the function"
    +        )
    +
    +	    @Override
    +        public Object apply(List<Object> args, Context context) throws ParseException
{
    +            String exec = "";
    +            String name = "";
    +            String path = "";
    +
    +            // if args are provided, get args, only if in whitelist
    +            if (args.size() >= 1) {
    +                Object execObj = args.get(0);
    +                if (!(execObj instanceof String)) { //check if string
    +                    return null;
    +                }
    +                else if (((String) execObj).length() > 0) {
    +                    exec = (String) execObj;
    +                }
    +                else {
    +                    return null;
    +                }
    +
    +                Object nameObj = args.get(1);
    +                if (!(nameObj instanceof String)) { //check if string
    +                    return null;
    +                }
    +                else if (((String) nameObj).length() > 0) {
    +                    name = (String) nameObj;
    +                }
    +                else {
    +                    return null;
    +                }
    +
    +                if (!Pattern.matches("[0-9A-Za-z.]+", name)) {
    +                    return null; //if not on whitelist
    +                }
    +
    +                path = "/scripts" + name;
    +                try {
    +                    File script = new File(path);
    +                    if (!script.exists() || script.isDirectory()) {
    +                        return null;
    +                    }
    +                }
    +                catch (NullPointerException e)  {
    +                    System.err.println("Error: " + e.toString());
    +                    return null;
    +                }
    +
    +                switch (exec) { //check if matches extension
    +                    case "bash":
    +                        if (name.contains(".sh")) {
    +                            isOnTheList = true;
    +                        }
    +                        break;
    +                    case "sh":
    +                        if (name.contains(".sh")) {
    +                            isOnTheList = true;
    +                        }
    +                        break;
    +                    case "python":
    +                        if (name.contains(".py")) {
    +                            isOnTheList = true;
    +                        }
    +                        break;
    +                    case "java":
    +                        if (name.contains(".class")) {
    +                            isOnTheList = true;
    +                            // java cmd needs called w/o .class
    +                            name.replace(".class", "");
    +                        }
    +                        break;
    +                }
    +            }
    +            if (!(isOnTheList)) {
    +                return null;
    +            }
    +
    +            try { // Create and start Process with ProcessBuilder.
    +                ProcessBuilder pb = new ProcessBuilder(exec, name);
    +                //Map<String, String> env = pb.environment();
    +                //env.put("VAR1", "myValue");
    +
    +                File log = new File("/scripts/log");
    +                pb.redirectErrorStream(true);
    +                pb.redirectOutput(ProcessBuilder.Redirect.appendTo(log));
    --- End diff --
    
    I would not mix error and output


---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastructure@apache.org or file a JIRA ticket
with INFRA.
---

Mime
View raw message