CustomActions/KamActions.cs

﻿/*******************************************************************************
 * Copyright 2017 McGill University All rights reserved.
 *
 * Licensed 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.
 *******************************************************************************/

using System;
using System.Collections.Generic;
using System.Text;
using System.IO;
using Microsoft.Deployment.WindowsInstaller;

namespace CustomActions
{
    public class KamActions
    {
        public static bool CheckIfIsIdaPluginsFolder(String selectedPath)
        {
            bool isCorrectFolder = false;
            DirectoryInfo info = new DirectoryInfo(selectedPath);
            if (!info.Name.Equals("plugins"))
                isCorrectFolder = false;
            else
            {
                var idadir = info.Parent;
                var files = idadir.GetFiles();
                foreach (var file in files)
                {
                    if (file.Name.Equals("idaq.exe"))
                    {
                        isCorrectFolder = true;
                        break;
                    }
                }
            }
            return isCorrectFolder;
        }

        [CustomAction]
        public static ActionResult ValidaIDAPRO(Session session)
        {
            string selectedVar = session["INSTALL_DIR"];
            string selectedPath = session[selectedVar];
            session.Log(selectedPath);
            if (CheckIfIsIdaPluginsFolder(selectedPath))
            {
                session["INSTALL_DIR_VALID"] = "1";
                session["PATH_IDA"] = (new DirectoryInfo(selectedPath)).Parent.FullName;
            }
            else
            {
                session["INSTALL_DIR_VALID"] = "0";
                session["PATH_IDA"] = "";
            }

            return ActionResult.Success;
        }
    }
}