fix(Server): fix body limit process

hans00 · Jul 22, 2022 · be38816 · be38816
1 parent 1c3a75f
commit be38816
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 1 deletion.
diff --git a/packages/server/js/connection.js b/packages/server/js/connection.js
@@ -42,7 +42,6 @@ class Connection {
       this._on_writable ? this._on_writable(offset) : true)
     this.wsContext = wsContext
     this._remote_address = null
-    this.processBodyData()
   }
 
   static create (app, request, response, wsContext = null) {

diff --git a/packages/server/js/routes.js b/packages/server/js/routes.js
@@ -56,6 +56,7 @@ class Routes {
         const req = Request.create(conn)
         const res = Response.create(conn)
         try {
+          conn.processBodyData()
           await callbacks(req, res, params)
         } catch (e) {
           this.log.error('Server Internal Error', e)

diff --git a/test/cases/http-post-large.js b/test/cases/http-post-large.js
@@ -0,0 +1,13 @@
+const axios = require('axios')
+const FormData = require('form-data')
+
+module.exports = async function ({ HTTP_PORT }) {
+  const res = await axios({
+    method: 'post',
+    url: `http://localhost:${HTTP_PORT}/post`,
+    data: 'A'.repeat(2048000),
+  }).catch(e => e.response)
+  if (res.status !== 413) {
+    throw new Error(`Response ${res.status}`)
+  }
+}
diff --git a/test/prepare/http.js b/test/prepare/http.js
@@ -6,6 +6,7 @@ module.exports = function (port) {
   return new Promise((resolve, reject) => {
     try {
       app(new fastWS({
+        bodySize: '1mb',
         logLevel: 'verbose',
         cache: new LRUCache({
           ttl: 1000 * 60 * 5,

diff --git a/test/prepare/https.js b/test/prepare/https.js
@@ -27,6 +27,7 @@ module.exports = function (port) {
     fs.writeFileSync(cert_file_name, cert.cert)
     try {
       app(new fastWS({
+        bodySize: '1mb',
         logLevel: 'verbose',
         ssl: {
           key_file_name,