main.go

package main

import (
	"toolset/controller"

	_ "toolset/docs"

	"github.com/gin-gonic/gin"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
)

// @title ToolSet API
// @version 1.0
// @description This a toolset server.
// @termsOfService http://swagger.io/terms/

// @contact.name API Support
// @contact.url http://www.swagger.io/support
// @contact.email support@swagger.io

// @license.name Apache 2.0
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html

// @host localhost:8080
// @BasePath /api/v1

func main() {
	r := gin.Default()

	c := controller.NewController()

	v1 := r.Group("/api/v1")
	{
		loader := v1.Group("/loader")
		{
			loader.POST("/shellcode-runner", c.ShellcodeRunner)
			loader.POST("/dll-proxyer", c.DllProxyer)
			loader.GET("/dll-hijack/config", c.DllHijackConfig)
			loader.POST("/dll-hijack", c.DllHijack)
		}
	}

	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
	r.Run(":8080")
}