go-nd-portal/cmd/main.go

// Package cmd interacts with user
package cmd

import (
	"flag"
	"fmt"
	"net"
	"net/netip"
	"os"
	"runtime"

	"golang.org/x/term"

	"github.com/sirupsen/logrus"

	"github.com/fumiama/go-nd-portal/helper"
	"github.com/fumiama/go-nd-portal/portal"
)

func outip() (net.IP, error) {
	conn, err := net.Dial("udp", "8.8.8.8:53")
	if err != nil {
		return nil, err
	}
	_ = conn.Close()
	return conn.LocalAddr().(*net.UDPAddr).IP.To4(), nil
}

func line() int {
	_, _, fileLine, ok := runtime.Caller(1)
	if ok {
		return fileLine
	}
	return -1
}

const query = "query"

// Main cmd program
func Main() {
	ip, err := outip()
	ipf := ""
	if err != nil {
		ipf = query
	} else {
		ipf = ip.String()
	}
	ips := flag.String("ip", ipf, "public IP")
	n := flag.String("n", query, "username")
	p := flag.String("p", query, "password")
	h := flag.Bool("h", false, "display this help")
	w := flag.Bool("w", false, "only display warn-or-higher-level log")
	d := flag.Bool("d", false, "display debug-level log")
	s := flag.String("s", portal.PortalServerIPQsh, "login host")
	t := flag.String("t", "qsh-edu", "login type [qsh-edu | qsh-dx | qshd-dx | qshd-cmcc]")
	flag.Parse()
	if *h {
		fmt.Println("Usage:")
		flag.PrintDefaults()
		os.Exit(0)
	}
	if *d {
		logrus.SetLevel(logrus.DebugLevel)
	} else if *w {
		logrus.SetLevel(logrus.WarnLevel)
	}
	if *ips == query {
		fmt.Printf("ip: ")
		_, err = fmt.Scanln(ips)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
	}
	if *ips != ip.String() {
		ipaddr, err := netip.ParseAddr(*ips)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
		a4 := ipaddr.As4()
		copy(ip, a4[:])
	}
	if *n == query {
		fmt.Printf("username: ")
		_, err = fmt.Scanln(n)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
	}
	if *p == query {
		fmt.Printf("password: ")
		data, err := term.ReadPassword(int(os.Stdin.Fd()))
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
		*p = helper.BytesToString(data)
		fmt.Println()
	}
	logrus.Debugf("server addr: %s, login type: %s", *s, *t)
	if *s != portal.PortalServerIPQsh {
		// just validate IP here, 
		// dont convert to net.IP because we need only its string later
		_, err := netip.ParseAddr(*s)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
	}
	// n : username 
	// p: password 
	// ip : public ip
	// *t : login type
	ptl, err := portal.NewPortal(*n, *p, ip, portal.LoginType(*t))
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	// input:
	// server IP
	challenge, err := ptl.GetChallenge(*s)
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	// input: 
	// server IP
	// challenge
	err = ptl.Login(*s, challenge)
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	logrus.Infoln("success")
}