47 lines
1.7 KiB
Go
47 lines
1.7 KiB
Go
package pia
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/qdm12/private-internet-access-docker/internal/constants"
|
|
"github.com/qdm12/private-internet-access-docker/internal/models"
|
|
)
|
|
|
|
func (c *configurator) GetPortForward() (port uint16, err error) {
|
|
c.logger.Info("%s: Obtaining port to be forwarded", logPrefix)
|
|
b, err := c.random.GenerateRandomBytes(32)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
clientID := hex.EncodeToString(b)
|
|
url := fmt.Sprintf("%s/?client_id=%s", constants.PIAPortForwardURL, clientID)
|
|
content, status, err := c.client.GetContent(url)
|
|
if err != nil {
|
|
return 0, err
|
|
} else if status != 200 {
|
|
return 0, fmt.Errorf("status is %d for %s; does your PIA server support port forwarding?", status, url)
|
|
} else if len(content) == 0 {
|
|
return 0, fmt.Errorf("port forwarding is already activated on this connection, has expired, or you are not connected to a PIA region that supports port forwarding")
|
|
}
|
|
body := struct {
|
|
Port uint16 `json:"port"`
|
|
}{}
|
|
if err := json.Unmarshal(content, &body); err != nil {
|
|
return 0, fmt.Errorf("port forwarding response: %w", err)
|
|
}
|
|
c.logger.Info("%s: Port forwarded is %d", logPrefix, body.Port)
|
|
return body.Port, nil
|
|
}
|
|
|
|
func (c *configurator) WritePortForward(filepath models.Filepath, port uint16) (err error) {
|
|
c.logger.Info("%s: Writing forwarded port to %s", logPrefix, filepath)
|
|
return c.fileManager.WriteLinesToFile(string(filepath), []string{fmt.Sprintf("%d", port)})
|
|
}
|
|
|
|
func (c *configurator) AllowPortForwardFirewall(device models.VPNDevice, port uint16) (err error) {
|
|
c.logger.Info("%s: Allowing forwarded port %d through firewall", logPrefix, port)
|
|
return c.firewall.AllowInputTrafficOnPort(device, port)
|
|
}
|