2025年2月17日月曜日

Go言語入門(12)

RPC serverを作る。彼は32bit Windows実行ファイルであり、32bit Windows DLLを使って何か計算する。RPC clientを作る。彼はRPC serverを起動してそこから結果を取得できる。ちょっと前まで64bit Windows実行ファイルだったけど64bit DLLになろうとしている。



もはやGo言語そっちのけ感があるけど、やりたかったことの終着点が来た。それは過去に取り組んだこちら(32bit DLLを64bitアプリから使う)をsocketに代えてRPCでやるって取り組みだったわけだよ、、、で具体的な構造はこんな感じ(以前と同じだけど32bitと64bitの間がsocketからgRPCになってる)
で、client以外は前回(Go言語入門(11))と全くおんなじ。
PATH="$PATH:$HOME/go/bin"
mkdir grpctry4
cd grpctry4
mkdir proto
mkdir server
mkdir client
go mod init grpc/grpctry4

proto/grpctry4.proto
  1. syntax = "proto3";
  2. package grpctry4;
  3. option go_package = "./proto";
  4. service MyAESEncrypter {
  5.   rpc AESEncrypt (Plain_and_Key) returns (Cipher) {}
  6. }
  7.  
  8. message Plain_and_Key {
  9.   bytes plain=1;
  10.   bytes key=2;
  11. }
  12.  
  13. message Cipher {
  14.   bytes result=1;
  15. }
一旦、
protoc --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative proto/grpctry4.proto
go get google.golang.org/grpc
(後回しにするとよく忘れちゃうから)

server/main.go
  1. package main
  2.  
  3. import (
  4.     "context"
  5.     "flag"
  6.     "fmt"
  7.     "log"
  8.     "net"
  9.     "syscall"
  10.     "unsafe"
  11.  
  12.     pb "grpc/grpctry4/proto"
  13.  
  14.     "google.golang.org/grpc"
  15. )
  16.  
  17. var (
  18.     hdll, _ = syscall.LoadLibrary("rijndael.dll")
  19.     aes128encrypt, _ = syscall.GetProcAddress(hdll, "AES128Encrypt")
  20.     aes128decrypt, _ = syscall.GetProcAddress(hdll, "AES128Decrypt")
  21. )
  22.  
  23. func AES128Encrypt(plain *uint8, key *uint8, crypted *uint8) (r uint32) {
  24.     var nargs uintptr = 3
  25.     ret, _, _ := syscall.Syscall9(uintptr(aes128encrypt),
  26.         nargs,
  27.         uintptr(unsafe.Pointer(plain)),
  28.         uintptr(unsafe.Pointer(key)),
  29.         uintptr(unsafe.Pointer(crypted)),
  30.         0, 0, 0, 0, 0, 0)
  31.     r = uint32(ret)
  32.     return
  33. }
  34. func AES128Decrypt(crypted *uint8, key *uint8, plain *uint8) (r uint32) {
  35.     var nargs uintptr = 3
  36.     ret, _, _ := syscall.Syscall9(uintptr(aes128decrypt),
  37.         nargs,
  38.         uintptr(unsafe.Pointer(crypted)),
  39.         uintptr(unsafe.Pointer(key)),
  40.         uintptr(unsafe.Pointer(plain)),
  41.         0, 0, 0, 0, 0, 0)
  42.     r = uint32(ret)
  43.     return
  44. }
  45.  
  46. var (
  47.     port = flag.Int("port", 50051, "The server port")
  48. )
  49.  
  50. type server struct {
  51.     pb.UnimplementedMyAESEncrypterServer
  52. }
  53.  
  54. func (s *server) AESEncrypt(_ context.Context, in *pb.PlainAnd_Key) (*pb.Cipher, error) {
  55.     var d1 []byte = in.GetPlain()
  56.     var d2 []byte = in.GetKey()
  57.     var d1arr [16]byte = *(*[16]byte)(d1[:16])
  58.     var d2arr [16]byte = *(*[16]byte)(d2[:16])
  59.     var d3arr [16]byte
  60.     //defer syscall.FreeLibrary(hdll)
  61.     AES128Encrypt(&d1arr[0], &d2arr[0], &d3arr[0])
  62.     return &pb.Cipher{Result: d3arr[:]}, nil
  63. }
  64. func main() {
  65.     flag.Parse()
  66.     lis, err := net.Listen("tcp", fmt.Sprintf(":%d", *port))
  67.     if err != nil {
  68.         log.Fatalf("failed to listen: %v", err)
  69.     }
  70.     s := grpc.NewServer()
  71.     pb.RegisterMyAESEncrypterServer(s, &server{})
  72.     log.Printf("server listening at %v", lis.Addr())
  73.     if err := s.Serve(lis); err != nil {
  74.         log.Fatalf("failed to serve: %v", err)
  75.     }
  76.     defer syscall.FreeLibrary(hdll)
  77. }

cd server
GOOS=windows GOARCH=386 go build -o ../grpctry4_server.exe main.go
cd ..

で、clientだが、前回、関数化するってことまでやってたので、コメントでexportするってのとmainの中身をコメント化するってことだけ。で、
  1. package main
  2.  
  3. /*
  4. #include <stdint.h>
  5. */
  6. import "C"
  7.  
  8. import (
  9.     "context"
  10.     "flag"
  11.     "fmt"
  12.     "log"
  13.     "os/exec"
  14.     "time"
  15.     "unsafe"
  16.  
  17.     pb "grpc/grpctry4/proto"
  18.  
  19.     "google.golang.org/grpc"
  20.     "google.golang.org/grpc/credentials/insecure"
  21. )
  22.  
  23. var (
  24.     addr = flag.String("addr", "localhost:50051", "the address to connect to")
  25. )
  26. var cmd *exec.Cmd
  27.  
  28. func init() {
  29.     srv_start_cmd := "./grpctry4_server.exe"
  30.     src_start_arg := ""
  31.     cmd = exec.Command(srv_start_cmd, src_start_arg)
  32.     cmd.Start()
  33.     fmt.Println(cmd.Process.Pid)
  34. }
  35.  
  36. type arr16 struct {
  37.     v [16]C.uint8_t
  38. }
  39.  
  40. //export AES128Encrypt
  41. func AES128Encrypt(cp *C.uint8_t, ck *C.uint8_t, cc *C.uint8_t) {
  42.     acp := (*arr16)(unsafe.Pointer(cp))
  43.     ack := (*arr16)(unsafe.Pointer(ck))
  44.     acc := (*arr16)(unsafe.Pointer(cc))
  45.     ppp := []byte{}
  46.     kkk := []byte{}
  47.     ccc := []byte{}
  48.     for i := 0; i < 16; i++ {
  49.         ppp = append(ppp, (uint8)((*acp).v[i]))
  50.         kkk = append(kkk, (uint8)((*ack).v[i]))
  51.     }
  52.     defer cmd.Process.Kill()
  54.     // Set up a connection to the server.
  55.     dialOpts := []grpc.DialOption{
  56.         grpc.WithTransportCredentials(insecure.NewCredentials()),
  57.         grpc.WithDefaultCallOptions(grpc.WaitForReady(true)),
  58.     }
  60.     conn, err := grpc.NewClient(*addr, dialOpts...)
  61.     if err != nil {
  62.         log.Fatalf("did not connect: %v", err)
  63.     }
  64.     defer conn.Close()
  65.     c := pb.NewMyAESEncrypterClient(conn)
  67.     // Contact the server and print out its response.
  68.     ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  69.     defer cancel()
  70.     r, err := c.AESEncrypt(ctx, &pb.PlainAnd_Key{Plain: ppp, Key: kkk})
  71.     if err != nil {
  72.         log.Fatalf("could not greet: %v", err)
  73.     }
  74.     ccc = r.GetResult()
  75.     for i := 0; i < 16; i++ {
  76.         (*acc).v[i] = (C.uint8_t)(ccc[i])
  77.     }
  78. }
  79. func main() {
  80.     /*
  81.      p := [...]C.uint8_t{0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff}
  82.      k := [...]C.uint8_t{0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f}
  83.      c := [...]C.uint8_t{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}
  84.      fmt.Printf("% 02X\n", p)
  85.      fmt.Printf("% 02X\n", k)
  86.      AES128Encrypt_grpcexec(&p[0], &k[0], &c[0])
  87.      fmt.Printf("% 02X\n", c)
  88.     */
  89. }

で、-buildmode=c-sharedを付けてbuildすればいいんだけど、オプションがめっちゃ長いので、Makefileを作っちゃう。何度もやり直す羽目になった時用。

client/Makefile
  1. OPT_CC= CC=x86_64-w64-mingw32-gcc
  2. #OPT_CC= CC=i686-w64-mingw32-gcc
  3. OPT_CGO= CGO_ENABLED=1
  4. OPT_OS= GOOS=windows
  5. OPT_ARCH= GOARCH=amd64
  6. #OPT_ARCH= GOARCH=386
  7. GOFLAGS=build -buildmode=c-shared
  8. #GOFLAGS=build -ldflags "-s -w" -buildmode=c-shared
  9. GO=go
  10. #SRCS=dll64wdllmain.go dllmain.go
  11. SRCS=main.go
  12. TARGET=../grpctry4_client.dll
  13.  
  14. $(TARGET) : $(SRCS)
  15.     $(OPT_CC) $(OPT_CGO) $(OPT_OS) $(OPT_ARCH) $(GO) $(GOFLAGS) -o $(TARGET) $(SRCS)
  16.  
  17. clean :
  18.     rm $(TARGET)
で、
cd client
make
うまくいくはず。
cd ..
でclientのDLLを呼び出す実行ファイルを作成する。これはC言語でいかせてください。こんなかんじ。
grpctry4_test_64.c
  1. #include <stdio.h>
  2. #include <stdint.h>
  3. #include <windows.h>
  4. typedef int (__stdcall *AES128Encrypt_type)(uint8_t* a,uint8_t* b,uint8_t* c);
  5. HMODULE dll;
  6. AES128Encrypt_type AES128Encrypt;
  7.   
  8. int main(int argc,char** argv){
  9.     int r,i;
  10.     uint8_t aa[16]={0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff};
  11.     uint8_t bb[16]={0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f};
  12.     uint8_t cc[16]={0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
  13.     dll=LoadLibrary("grpctry4_client.dll");
  14.     AES128Encrypt=(AES128Encrypt_type)GetProcAddress(dll,"AES128Encrypt");
  15.     for(i=0;i<16;i++){
  16.         printf("%02x",aa[i]);
  17.     }
  18.     printf("\n");
  19.     for(i=0;i<16;i++){
  20.         printf("%02x",bb[i]);
  21.     }
  22.     printf("\n");
  23.     r=AES128Encrypt(aa,bb,cc);
  24.     for(i=0;i<16;i++){
  25.         printf("%02x",cc[i]);
  26.     }
  27.     printf("\n");
  28.     return 0;
  29. }
で、コンパイルする。
x86_64-w64-mingw32-gcc grpctry4_test_64.c
で、こうなる。
  • .
  • ├── a.exe
  • ├── client
  • │ ├── Makefile
  • │ ├── dllmain.h
  • │ └── main.go
  • ├── go.mod
  • ├── go.sum
  • ├── grpctry4_client.dll
  • ├── grpctry4_client.h
  • ├── grpctry4_server.exe
  • ├── grpctry4_test_64.c
  • ├── proto
  • │ ├── grpctry4.pb.go
  • │ ├── grpctry4.proto
  • │ └── grpctry4_grpc.pb.go
  • ├── rijndael.dll
  • └── server
  •     └── main.go

で、実行。
./a.exe
すごいぞ。64bitアプリから32bitDLLを呼び出せているように見える。Go言語使うってことを除けばsocketよりも全然簡単。C++でもC#でもできるんだろうけどGo言語の方がたぶん簡単。
at
Labels: , ,

1 件のコメント:

  1. ヒマナッツ2025年2月17日 16:44

    テスト用CコードにFreeLibraryが無いのは良くない

    返信削除

登録: コメントの投稿 (Atom)