/*
 * Copyright 2003-2004, Waldemar Kornewald <wkornew@gmx.net>
 * Copyright 2006-2017, Haiku, Inc. All rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *   Alexander von Gluck IV <kallisti5@unixzen.com>
 */

#include <cstdio>
#include <String.h>
#include <driver_settings.h>

#include <PPPInterface.h>
#include <PPPManager.h>


static const char sVersion[] = "0.12 pre-alpha";
static const char sPPPInterfaceModuleName[] = PPP_INTERFACE_MODULE_NAME;


static
status_t
print_help()
{
        fprintf(stderr, "Haiku Network Team: pppconfig: sVersion %s\n", sVersion);
        fprintf(stderr, "With pppconfig you can create and manage PPP connections.\n");
        fprintf(stderr, "Usage:\n");
        fprintf(stderr, "pppconfig show | -a\n");
        fprintf(stderr, "pppconfig init <name>\n");
        fprintf(stderr, "pppconfig create <name>\n");
        fprintf(stderr, "pppconfig connect <name|interface|id>\n");
        fprintf(stderr, "pppconfig disconnect <name|interface|id>\n");
        fprintf(stderr, "pppconfig delete <name|interface|id>\n");
        fprintf(stderr, "pppconfig details <name|interface|id>\n");
        fprintf(stderr, "\t<name> must be an interface description file\n");
        
        return -1;
}


static
status_t
show(ppp_interface_filter filter = PPP_REGISTERED_INTERFACES)
{
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        int32 count = 0;
        ppp_interface_id *interfaces = manager.Interfaces(&count, filter);
        
        if (!interfaces || count <= 0) {
                fprintf(stderr, "Error: Could not get interfaces information!\n");
                return -1;
        }

        fprintf(stderr, "Get %" B_PRId32 " ppp interfaces, first is %" B_PRIu32 "!\n",
                count, interfaces[0]);

        ppp_interface_info_t info;
        PPPInterface interface;
        
        printf("Listing PPP interfaces:\n");
        
        // print out information for each interface
        for (int32 index = 0; index < count; index++) {
                interface.SetTo(interfaces[index]);
                if (interface.InitCheck() == B_OK) {
                        interface.GetInterfaceInfo(&info);
                        printf("\n");
                        
                        // type and unit (if it has one)
                        if (info.info.if_unit >= 0) {
                                printf("Type: Visible\n");
                                printf("\tInterface: ppp%" B_PRId32 "\n", info.info.if_unit);
                        } else
                                printf("Type: Hidden\n");
                        
                        printf("\tName: %s\n", info.info.name);
                        
                        // ID
                        printf("\tID: %" B_PRIu32 "\n", interface.ID());
                        
                        // mode
                        printf("\tMode: ");
                        if (info.info.mode == PPP_CLIENT_MODE)
                                printf("Client\n");
                        else if (info.info.mode == PPP_SERVER_MODE)
                                printf("Server\n");
                        else
                                printf("Unknown\n");
                        
                        // status
                        printf("\tStatus: ");
                        switch(info.info.phase) {
                                case PPP_ESTABLISHED_PHASE:
                                        printf("Connected\n");
                                break;
                                
                                case PPP_DOWN_PHASE:
                                        printf("Disconnected\n");
                                break;
                                
                                case PPP_TERMINATION_PHASE:
                                        printf("Disconnecting\n");
                                break;
                                
                                default:
                                        printf("Connecting\n");
                        }
                }
        }
        
        delete interfaces;
        
        return 0;
}


static
status_t
create(const char *name, bool bringUp = true)
{
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.CreateInterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not create interface: %s!\n", name);
                return -1;
        }
        
        printf("Created interface with ID: %" B_PRIu32 "\n", interface.ID());
        
        ppp_interface_info_t info;
        interface.GetInterfaceInfo(&info);
        
        if (info.info.if_unit >= 0)
                printf("Interface: ppp%" B_PRId32 "\n", info.info.if_unit);
        else
                printf("This interface is hidden! You can delete it by typing:\n"
                        "pppconfig delete %" B_PRIu32 "\n", interface.ID());
        
        if (bringUp) {
                interface.Up();
                printf("Connecting in background...\n");
        }
        
        return 0;
}


static
status_t
connect(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        if (!interface.Up()) {
                fprintf(stderr, "Error: Could not connect!\n");
                return -1;
        }
        
        printf("Connecting in background...\n");
        
        return 0;
}


static
status_t
setuser(const char *name, const char* user)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        if (!user || strlen(user) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        if (!interface.SetUsername(user)) {
                fprintf(stderr, "Error: Could not SetUsername %s!\n", user);
                return -1;
        }
        
        return 0;
}


static
status_t
setpass(const char *name, const char* pass)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        if (!pass || strlen(pass) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        if (!interface.SetPassword(pass)) {
                fprintf(stderr, "Error: Could not SetUsername %s!\n", pass);
                return -1;
        }
        
        return 0;
}


static
status_t
setaskbeforeconnect(const char *name, const char* connect)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        bool askBeforeConnecting = false;
        if (connect || !strcmp(connect, "true") || !strcmp(connect, "ask") ||
                !strcmp(connect, "yes") || !strcmp(connect, "y"))
                askBeforeConnecting = true;


        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        if (!interface.SetAskBeforeConnecting(askBeforeConnecting)) {
                fprintf(stderr, "Error: Could not connect %s!\n", connect);
                return -1;
        }
        
        return 0;
}


static
status_t
getstatistics(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_statistics pppStatistics;
        if (interface.GetStatistics(&pppStatistics) != B_OK) {
                fprintf(stderr, "Error: Could not getstatistics: %s!\n", name);
                return -1;
        }

        return 0;
}


static
status_t
hassettings(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        driver_settings settings;
        if (interface.HasSettings(&settings) != B_OK) {
                fprintf(stderr, "Error: Could not getstatistics: %s!\n", name);
                return -1;
        }

        return 0;
}


static
status_t
enablereports(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_report_type type = PPP_ALL_REPORTS;
        thread_id thread = 0;
        int32 flags = 0;

        if (interface.EnableReports(type, thread, flags) != true) {
                fprintf(stderr, "Error: Could not EnableReports: %s!\n", name);
                return -1;
        }

        return 0;
}


static
status_t
disablereports(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_report_type type = PPP_ALL_REPORTS;
        thread_id thread = 0;

        if (interface.DisableReports(type, thread) != true) {
                fprintf(stderr, "Error: Could not EnableReports: %s!\n", name);
                return -1;
        }

        return 0;
}


static
status_t
controlchild(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_simple_handler_info_t info;

        if (interface.ControlChild(&info, 0, PPPC_GET_SIMPLE_HANDLER_INFO) != true) {
                fprintf(stderr, "Error: Could not PPPC_GET_SIMPLE_HANDLER_INFO: %s!\n", name);
                return -1;
        }

        printf("LCPExtensionHandler: %s\n", info.info.name);
        printf("isEnabled: %d\n", info.info.isEnabled);

        if (interface.ControlChild(&info, 0, PPPC_ENABLE) != true) {
                fprintf(stderr, "Error: Could not PPPC_ENABLE: %s!\n", name);
                return -1;
        }
        return 0;
}


static
status_t
controllcpextension(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_simple_handler_info_t info;

        if (interface.ControlLCPExtension(&info, 1, PPPC_GET_SIMPLE_HANDLER_INFO) != true) {
                fprintf(stderr, "Error: Could not PPPC_GET_SIMPLE_HANDLER_INFO: %s!\n", name);
                return -1;
        }

        printf("LCPExtensionHandler: %s\n", info.info.name);
        printf("isEnabled: %d\n", info.info.isEnabled);

        if (interface.ControlLCPExtension(&info, 1, PPPC_ENABLE) != true) {
                fprintf(stderr, "Error: Could not PPPC_ENABLE: %s!\n", name);
                return -1;
        }
        return 0;
}


static
status_t
controloptionhandler(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_simple_handler_info_t info;

        if (interface.ControlOptionHandler(&info, 0, PPPC_GET_SIMPLE_HANDLER_INFO) != true) {
                fprintf(stderr, "Error: Could not PPPC_GET_SIMPLE_HANDLER_INFO: %s!\n", name);
                return -1;
        }

        printf("protocol: %s\n", info.info.name);
        printf("isEnabled: %d\n", info.info.isEnabled);

        if (interface.ControlOptionHandler(&info, 0, PPPC_ENABLE) != true) {
                fprintf(stderr, "Error: Could not PPPC_ENABLE: %s!\n", name);
                return -1;
        }
        return 0;
}


static
status_t
controlprotocol(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_protocol_info_t info;

        if (interface.ControlProtocol(&info, 0, PPPC_GET_PROTOCOL_INFO) != true) {
                fprintf(stderr, "Error: Could not PPPC_GET_PROTOCOL_INFO: %s!\n", name);
                return -1;
        }

        printf("protocol: %s\n", info.info.name);
        printf("type: %s\n", info.info.type);
        printf("activationPhase: %d\n", info.info.activationPhase);
        printf("addressFamily: %" B_PRId32 "\n", info.info.addressFamily);
        printf("flags: %" B_PRId32 "\n", info.info.flags);
        printf("side: %d\n", info.info.side);
        printf("level: %d\n", info.info.level);
        printf("connectionPhase: %d\n", info.info.connectionPhase);
        printf("protocolNumber: %d\n", info.info.protocolNumber);
        printf("isEnabled: %d\n", info.info.isEnabled);
        printf("isUpRequested: %d\n", info.info.isUpRequested);

        if (interface.ControlProtocol(&info, 0, PPPC_ENABLE) != true) {
                fprintf(stderr, "Error: Could not PPPC_ENABLE: %s!\n", name);
                return -1;
        }
        return 0;
}


static
status_t
controldevice(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;

        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        ppp_device_info_t info;

        if (interface.ControlDevice(&info) != true) {
                fprintf(stderr, "Error: Could not ControlDevice: %s!\n", name);
                return -1;
        }

        printf("name: %s\n", info.info.name);
        printf("MTU: %" B_PRIu32 "\n", info.info.MTU);
        printf("inputTransferRate: %" B_PRIu32 "\n", info.info.inputTransferRate);
        printf("outputTransferRate: %" B_PRIu32 "\n", info.info.outputTransferRate);
        printf("outputBytesCount: %" B_PRIu32 "\n", info.info.outputBytesCount);
        printf("isUp: %d\n", info.info.isUp);

        return 0;
}


static
status_t
disconnect(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        PPPInterface interface(manager.InterfaceWithName(name));
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
        
        if (!interface.Down()) {
                fprintf(stderr, "Error: Could not disconnect!\n");
                return -1;
        }
        
        return 0;
}


static
status_t
delete_interface(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        if (!manager.DeleteInterface(name)) {
                fprintf(stderr, "Error: Could not delete interface!\n");
                return -1;
        }
        
        return 0;
}


static
status_t
show_details(const char *name)
{
        if (!name || strlen(name) == 0)
                return -1;
        
        PPPManager manager;
        if (manager.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not load interface manager!\n");
                return -1;
        }
        
        ppp_interface_id ID = manager.InterfaceWithName(name);
        if (ID <= 0) {
                fprintf(stderr, "Error: Could not find interface: %s!\n", name);
                return -1;
        }
                
        PPPInterface interface(ID);
        if (interface.InitCheck() != B_OK) {
                fprintf(stderr, "Error: Could not find interface ID: %" B_PRIu32 "!\n", ID);
                return -1;
        }
        
        ppp_interface_info_t info;
        // printf("ppp_interface_info_t addr:%p\n", &info);
        interface.GetInterfaceInfo(&info);
        
        // type and name (if it has one)
        if (info.info.if_unit >= 0) {
                printf("Type: Visible\n");
                printf("Interface: ppp%" B_PRId32 "\n", info.info.if_unit);
        } else
                printf("Type: Hidden\n");
        
        printf("Name: %s\n", info.info.name);
        
        // ID
        printf("ID: %" B_PRIu32 "\n", interface.ID());
        
        // ConnectOnDemand
        printf("ConnectOnDemand: %s\n", info.info.doesConnectOnDemand ?
                "Enabled" : "Disabled");
        
        // AutoReconnect
        printf("AutoReconnect: %s\n", info.info.doesAutoReconnect ?
                "Enabled" : "Disabled");
        
        // MRU and interfaceMTU
        printf("MRU: %" B_PRIu32 "\n", info.info.MRU);
        printf("Interface MTU: %" B_PRIu32 "\n", info.info.interfaceMTU);
        
        // mode
        printf("Mode: ");
        if (info.info.mode == PPP_CLIENT_MODE)
                printf("Client\n");
        else if (info.info.mode == PPP_SERVER_MODE)
                printf("Server\n");
        else
                printf("Unknown\n");
        
        // status
        printf("\tStatus: ");
        switch(info.info.phase) {
                case PPP_ESTABLISHED_PHASE:
                        printf("Connected\n");
                break;
                
                case PPP_DOWN_PHASE:
                        printf("Disconnected\n");
                break;
                
                case PPP_TERMINATION_PHASE:
                        printf("Disconnecting\n");
                break;
                
                default:
                        printf("Connecting\n");
        }
        
        return 0;
}


int
main(int argc, char *argv[])
{
        if (argc == 2) {
                if (!strcmp(argv[1], "show") || !strcmp(argv[1], "-a"))
                        return show(PPP_ALL_INTERFACES);
                else
                        return print_help();
        }if (argc == 3) {
                if (!strcmp(argv[1], "init"))
                        return create(argv[2], false);
                else if (!strcmp(argv[1], "create"))
                        return create(argv[2], true);
                else if (!strcmp(argv[1], "connect"))
                        return connect(argv[2]);
                else if (!strcmp(argv[1], "disconnect"))
                        return disconnect(argv[2]);
                else if (!strcmp(argv[1], "delete"))
                        return delete_interface(argv[2]);
                else if (!strcmp(argv[1], "details"))
                        return show_details(argv[2]);
                else if (!strcmp(argv[1], "getstatistics"))
                        return getstatistics(argv[2]);
                else if (!strcmp(argv[1], "hassettings"))
                        return hassettings(argv[2]);
                else if (!strcmp(argv[1], "enablereports"))
                        return enablereports(argv[2]);
                else if (!strcmp(argv[1], "disablereports"))
                        return disablereports(argv[2]);
                else if (!strcmp(argv[1], "controldevice"))
                        return controldevice(argv[2]);
                else if (!strcmp(argv[1], "controlprotocol"))
                        return controlprotocol(argv[2]);
                else if (!strcmp(argv[1], "controloptionhandler"))
                        return controloptionhandler(argv[2]);
                else if (!strcmp(argv[1], "controllcpextension"))
                        return controllcpextension(argv[2]);
                else if (!strcmp(argv[1], "controlchild"))
                        return controlchild(argv[2]);
                else
                        return print_help();
        } if (argc == 4) {
                if (!strcmp(argv[1], "setuser"))
                        return setuser(argv[2], argv[3]);
                else if (!strcmp(argv[1], "setpass"))
                        return setpass(argv[2], argv[3]);
                else if (!strcmp(argv[1], "setaskbeforeconnect"))
                        return setaskbeforeconnect(argv[2], argv[3]);
                else
                        return print_help();
        } else
                return print_help();
}