server_common/wsproxy
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
aozhiwei 2023-04-24 11:40:28 +08:00
parent 4677716f1f
commit 51b01f8e79
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server/wsproxy/downstream.cc
View File

@ -5,6 +5,7 @@
#include "longsessionmgr.h" #include "longsessionmgr.h"
#include "GCListener.h" #include "GCListener.h"
#include "longsession.h" #include "longsession.h"
#include "kcpsession.h"
#include "ss_proto.pb.h" #include "ss_proto.pb.h"
@ -32,14 +33,12 @@ void DownStream::ForwardUpStreamMsg(f8::MsgHdr& hdr)
memmove(buff + sizeof(f8::PackHead), hdr.buf, hdr.buflen); memmove(buff + sizeof(f8::PackHead), hdr.buf, hdr.buflen);
} }
#ifdef USE_KCP auto session = LongSessionMgr::Instance()->GetSession(hdr.socket_handle);
auto session = GetKcpSessionBySocketHandle(hdr.socket_handle);
if (session) { if (session) {
session->SendClientMsg(buff, sizeof(f8::PackHead) + head->packlen); session->GetKcpSession()->SendClientMsg(buff, sizeof(f8::PackHead) + head->packlen);
} } else {
#else
GCListener::Instance()->SendBuf(hdr.socket_handle, buff, sizeof(f8::PackHead) + head->packlen); GCListener::Instance()->SendBuf(hdr.socket_handle, buff, sizeof(f8::PackHead) + head->packlen);
#endif }
free(buff); free(buff);
} }

1
server/wsproxy/downstream.h
View File

@ -17,5 +17,4 @@ class DownStream
private: private:
int socket_handle_ = a8::INVALID_SOCKET_HANDLE; int socket_handle_ = a8::INVALID_SOCKET_HANDLE;
std::weak_ptr<UpStream> up_; std::weak_ptr<UpStream> up_;
int type_ = 0;
}; };