package main

import (
	"crypto/sha256"
	"fmt"
	"net"

	"golang.org/x/crypto/chacha20"
)

// EncryptIPv4 IPv4暗号化 (32bit→32bit)
func EncryptIPv4(ip net.IP, psk []byte) ([]byte, error) {
	if ip.To4() == nil {
		return nil, fmt.Errorf("invalid ipv4 address")
	}

	key := sha256.Sum256(psk)
	encryptKey := key[:32]
	nonce := make([]byte, 12)
	cipher, err := chacha20.NewUnauthenticatedCipher(encryptKey, nonce)
	if err != nil {
		return nil, err
	}

	// インプレース暗号化（32bit正確な出力）
	ciphertext := make([]byte, net.IPv4len)
	cipher.XORKeyStream(ciphertext, ip.To4())
	return ciphertext, nil
}

// EncryptIPv6 IPv6暗号化 (128bit→128bit)
func EncryptIPv6(ip net.IP, psk []byte) ([]byte, error) {
	if ip.To16() == nil {
		return nil, fmt.Errorf("invalid ipv6 address")
	}

	key := sha256.Sum256(psk)
	encryptKey := key[:32]
	nonce := make([]byte, 12)
	cipher, err := chacha20.NewUnauthenticatedCipher(encryptKey, nonce)
	if err != nil {
		return nil, err
	}

	// 128bit正確な出力
	ciphertext := make([]byte, net.IPv6len)
	cipher.XORKeyStream(ciphertext, ip.To16())
	return ciphertext, nil
}

// 復号関数（暗号化/復号に同じ関数を使用します）
var DecryptIPv4 = EncryptIPv4
var DecryptIPv6 = EncryptIPv6

func main() {
	psk := []byte("test-datakey")

	// IPv4テスト
	ipv4 := net.ParseIP("192.168.1.1")
	encV4, err := EncryptIPv4(ipv4, psk)
	if err != nil {
		fmt.Println("encrypt ipv4 error:", err)
		return
	}
	decV4, err := DecryptIPv4(encV4, psk)
	if err != nil {
		fmt.Println("decrypt ipv4 error:", err)
		return
	}
	fmt.Printf("IPv4:\\n  元IP: %v\\n  暗号文: %08x\\n  復号: %v\\n",
		ipv4, encV4, net.IP(decV4))

	// IPv6テスト
	ipv6 := net.ParseIP("2001:db8::a1b2")
	encV6, err := EncryptIPv6(ipv6, psk)
	if err != nil {
		fmt.Println("encrypt ipv6 error:", err)
		return
	}
	decV6, err := DecryptIPv6(encV6, psk)
	if err != nil {
		fmt.Println("decrypt ipv6 error:", err)
		return
	}
	fmt.Printf("\\nIPv6:\\n  元IP: %v\\n  暗号文: %032x\\n  復号: %v\\n",
		ipv6, encV6, net.IP(decV6))
}

package com.android.tencentvpn.util;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import java.security.Key;
import java.util.Date;
import javax.crypto.spec.SecretKeySpec;

public class SignGenerate {
    private static final long EXPIRE_TIME = 72 * 60 * 60 * 1000; // 有効期限、単位はミリ秒です。
    /**
     * @param deviceName   業務側でデバイスの一意な識別子を管理します。
     * @param SecretKey   TencentCloud apiが返すアプリケーションキー。
     * @return   マルチネットワークsdkに渡す必要がある署名文字列。最終的にsetSignを通じて呼び出されます。
     *
     */
    public static String generateSign(String deviceName, String SecretKey) {
        try {
            Date now = new Date();
            Date expireDate = new Date(now.getTime() + EXPIRE_TIME);
            Key key =
                    new SecretKeySpec(SecretKey.getBytes(), SignatureAlgorithm.HS256.getJcaName());
            return Jwts.builder()
                    .claim("deviceName", deviceName)
                    .expiration(expireDate)
                    .signWith(key)
                    .compact();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }
}

//podfileにpod 'JWT'を追加し、JWTライブラリを統合します。 
// .hファイル：
@interface SignGenerator : NSObject
/**
{
  "type": "sentence",
  "analyse_result": "原文为技术文档中的注释语句、属于省略主语的陈述句。分析如下：一、句子判定：该文本为句子、主语隐含。二、成分分析：1、句型为陈述句、单句。2、主语与语态分析：表层无显性主语、深层逻辑主语为‘业务方’（隐含）、具备执行‘控制’动作的能力。‘@deviceName’为注释标记及参数名、作提示性成分。‘业务自己’为状语、限定动作主体。‘控制’为谓语核心动词。‘设备唯一标识’为宾语。全句意为‘业务方自行控制设备唯一标识’、主动语态、逻辑自洽。3、主干：主语（隐含：业务方）＋谓语‘控制’＋宾语‘设备唯一标识’。4、修饰语：‘业务自己’修饰谓语‘控制’、表示动作由业务方自身执行、‘@deviceName’为注释标记、提示该参数含义、不参与主干结构。5、检查验证：深层逻辑与语法自洽、无施受矛盾。"
}
* @param SecretKey TencentCloud apiが返すアプリケーションキー。
* @return マルチネットワークsdkに渡す必要がある署名文字列。最終的にsetSignを通じて呼び出されます。
*/
 + (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey;
@end
// .mファイル
#import "SignGenerator.h"
#import <JWT/JWT.h>
@implementation SignGenerator
+ (NSString *)generateSignWithDeviceName:(NSString *)deviceName secretKey:(NSString *)secretKey {
    // 1. 有効期限を設定します（72時間後）。
        NSDate *currentDate = [NSDate date];
        NSDate *expireDate = [currentDate dateByAddingTimeInterval:72 * 60 * 60];
        // 2. Claims（ペイロード）を構築します。
        NSDictionary *claims = @{
            @"deviceName": deviceName,
            @"exp": @((long)[expireDate timeIntervalSince1970])   // Expiration
        };
        NSData *keyData = [secretKey dataUsingEncoding:NSUTF8StringEncoding];
        // 3. HS256アルゴリズムとキーを使用して署名します。
        JWTBuilder *builder = [JWTBuilder encodePayload:claims]
            .secretData(keyData)
            .algorithmName(@"HS256"); //アルゴリズム名
        NSString *jwt = builder.encode;
        // 4. エラー処理。
        if (builder.jwtError) {
            NSLog(@"JWT生成失敗: %@", builder.jwtError);
            return @"";
        }  
        return jwt;
}

package main
import (
	"fmt"
	"github.com/dgrijalva/jwt-go/v4"
	"time"
)
func main() {
	sign, expireTime, err := GenerateSign("tencent-test")
	if err != nil {
		panic(err)
	}
	println(sign, expireTime)
}
// SignClaims JwtCustomClaims
type SignClaims struct {
	jwt.StandardClaims
	DeviceName string `json:"deviceName"`
}
var (
	ExpireTime = 72                       // 有効期限、単位は時間です。
	SecretKey = []byte("your secretkey") // APPキー
)
// GenerateSignは署名を生成します。
func GenerateSign(deviceName string) (string, int64, error) {
	expireTime := time.Now().Add(time.Duration(ExpireTime) * time.Hour)
	claims := &SignClaims{
		StandardClaims: jwt.StandardClaims{
			NotBefore: jwt.Now(),
			ExpiresAt: jwt.At(expireTime),
		},
		DeviceName: deviceName,
	}
	println(fmt.Sprintf("%#v", claims))
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	sign, err := token.SignedString(SecretKey)
	if err != nil {
		return "", 0, err
    }
	return sign, expireTime.Unix(), nil
}