From f5a40b695659e449564402b53cb2ed3a6f11c11b Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Tue, 19 Apr 2022 11:09:42 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=9B=E5=BB=BA=E5=92=8C=E4=BF=AE=E6=94=B9?= =?UTF-8?q?=E8=AF=81=E4=B9=A6=E7=9A=84=E6=97=B6=E5=80=99=E6=A3=80=E6=9F=A5?= =?UTF-8?q?=E6=97=B6=E9=97=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/rpc/services/service_ssl_cert.go | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/internal/rpc/services/service_ssl_cert.go b/internal/rpc/services/service_ssl_cert.go index 99b91b5c..5039000c 100644 --- a/internal/rpc/services/service_ssl_cert.go +++ b/internal/rpc/services/service_ssl_cert.go @@ -5,6 +5,7 @@ import ( "encoding/json" "github.com/TeaOSLab/EdgeAPI/internal/db/models" "github.com/TeaOSLab/EdgeAPI/internal/db/models/acme" + "github.com/TeaOSLab/EdgeAPI/internal/errors" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/sslconfigs" "github.com/iwind/TeaGo/types" @@ -23,7 +24,14 @@ func (this *SSLCertService) CreateSSLCert(ctx context.Context, req *pb.CreateSSL return nil, err } - tx := this.NullTx() + var tx = this.NullTx() + + if req.TimeBeginAt < 0 { + return nil, errors.New("invalid TimeBeginAt") + } + if req.TimeEndAt < 0 { + return nil, errors.New("invalid TimeEndAt") + } certId, err := models.SharedSSLCertDAO.CreateCert(tx, adminId, userId, req.IsOn, req.Name, req.Description, req.ServerName, req.IsCA, req.CertData, req.KeyData, req.TimeBeginAt, req.TimeEndAt, req.DnsNames, req.CommonNames) if err != nil { @@ -41,7 +49,14 @@ func (this *SSLCertService) UpdateSSLCert(ctx context.Context, req *pb.UpdateSSL return nil, err } - tx := this.NullTx() + var tx = this.NullTx() + + if req.TimeBeginAt < 0 { + return nil, errors.New("invalid TimeBeginAt") + } + if req.TimeEndAt < 0 { + return nil, errors.New("invalid TimeEndAt") + } // 检查权限 if userId > 0 {