mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-29 04:22:24 -05:00 
			
		
		
		
	penis
This commit is contained in:
		
					parent
					
						
							
								c7c8edd982
							
						
					
				
			
			
				commit
				
					
						db4a6e746c
					
				
			
		
					 7 changed files with 200 additions and 111 deletions
				
			
		|  | @ -499,7 +499,6 @@ var Start action.GTSAction = func(ctx context.Context) error { | |||
| 	s2sLimit := middleware.RateLimit(rlLimit, exceptions)       // server-to-server (AP) | ||||
| 	fsMainLimit := middleware.RateLimit(rlLimit, exceptions)    // fileserver / web templates | ||||
| 	fsEmojiLimit := middleware.RateLimit(rlLimit*2, exceptions) // fileserver (emojis only, use high limit) | ||||
| 	nollamas := middleware.NoLLaMas(instanceAccount) | ||||
| 
 | ||||
| 	// throttling | ||||
| 	cpuMultiplier := config.GetAdvancedThrottlingMultiplier() | ||||
|  | @ -545,7 +544,7 @@ var Start action.GTSAction = func(ctx context.Context) error { | |||
| 	nodeInfoModule.Route(route, s2sLimit, s2sThrottle, gzip) | ||||
| 	activityPubModule.Route(route, s2sLimit, s2sThrottle, robotsDisallowAll, gzip) | ||||
| 	activityPubModule.RoutePublicKey(route, s2sLimit, pkThrottle, robotsDisallowAll, gzip) | ||||
| 	webModule.Route(route, fsMainLimit, fsThrottle, robotsDisallowAIOnly, nollamas, gzip) | ||||
| 	webModule.Route(route, fsMainLimit, fsThrottle, robotsDisallowAIOnly, gzip) | ||||
| 
 | ||||
| 	// Finally start the main http server! | ||||
| 	if err := route.Start(); err != nil { | ||||
|  |  | |||
|  | @ -18,6 +18,9 @@ | |||
| package middleware | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"crypto/rand" | ||||
| 	"crypto/rsa" | ||||
| 	"crypto/sha256" | ||||
| 	"crypto/sha512" | ||||
| 	"crypto/subtle" | ||||
|  | @ -29,15 +32,23 @@ import ( | |||
| 
 | ||||
| 	"codeberg.org/gruf/go-byteutil" | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||
| 	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" | ||||
| 	apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util" | ||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtserror" | ||||
| 	"github.com/superseriousbusiness/gotosocial/internal/oauth" | ||||
| ) | ||||
| 
 | ||||
| func NoLLaMas(instanceAcc *gtsmodel.Account) gin.HandlerFunc { | ||||
| 	// Generate seed hash from | ||||
| 	// this instance private key. | ||||
| 	priv := instanceAcc.PrivateKey | ||||
| 	bpriv := x509.MarshalPKCS1PrivateKey(priv) | ||||
| func NoLLaMas( | ||||
| 	getInstance func(ctx context.Context) (*apimodel.InstanceV1, gtserror.WithCode), | ||||
| ) gin.HandlerFunc { | ||||
| 	privKey, err := rsa.GenerateKey(rand.Reader, 2048) | ||||
| 	if err != nil { | ||||
| 		panic(err) | ||||
| 	} | ||||
| 
 | ||||
| 	// Generate seed hash | ||||
| 	// from this private key. | ||||
| 	bpriv := x509.MarshalPKCS1PrivateKey(privKey) | ||||
| 	seed := sha512.Sum512(bpriv) | ||||
| 
 | ||||
| 	// Configure nollamas. | ||||
|  | @ -45,6 +56,7 @@ func NoLLaMas(instanceAcc *gtsmodel.Account) gin.HandlerFunc { | |||
| 	nollamas.seed = seed[:] | ||||
| 	nollamas.ttl = time.Hour | ||||
| 	nollamas.diff = 4 | ||||
| 	nollamas.getInstance = getInstance | ||||
| 	return nollamas.Serve | ||||
| } | ||||
| 
 | ||||
|  | @ -57,9 +69,10 @@ const encodedHashLen = 2 * hashLen | |||
| func newHash() hash.Hash { return sha256.New() } | ||||
| 
 | ||||
| type nollamas struct { | ||||
| 	seed []byte // securely hashed instance private key | ||||
| 	ttl  time.Duration | ||||
| 	diff uint8 | ||||
| 	seed        []byte // securely hashed private key | ||||
| 	ttl         time.Duration | ||||
| 	diff        uint8 | ||||
| 	getInstance func(ctx context.Context) (*apimodel.InstanceV1, gtserror.WithCode) | ||||
| } | ||||
| 
 | ||||
| func (m *nollamas) Serve(c *gin.Context) { | ||||
|  | @ -169,10 +182,26 @@ func (m *nollamas) renderChallenge(c *gin.Context, challenge string) { | |||
| 	// our challenge page. | ||||
| 	c.Abort() | ||||
| 
 | ||||
| 	instance, errWithCode := m.getInstance(c.Request.Context()) | ||||
| 	if errWithCode != nil { | ||||
| 		apiutil.ErrorHandler(c, errWithCode, m.getInstance) | ||||
| 		return | ||||
| 	} | ||||
| 
 | ||||
| 	// Write the templated challenge HTML response to client. | ||||
| 	c.HTML(http.StatusOK, "nollamas.tmpl", map[string]any{ | ||||
| 		"challenge":  challenge, | ||||
| 		"difficulty": m.diff, | ||||
| 	apiutil.TemplateWebPage(c, apiutil.WebPage{ | ||||
| 		Template: "nollamas.tmpl", | ||||
| 		Instance: instance, | ||||
| 		Extra: map[string]any{ | ||||
| 			"challenge":  challenge, | ||||
| 			"difficulty": m.diff, | ||||
| 		}, | ||||
| 		Javascript: []apiutil.JavascriptEntry{ | ||||
| 			{ | ||||
| 				Src:   "/assets/dist/nollamas.js", | ||||
| 				Defer: true, | ||||
| 			}, | ||||
| 		}, | ||||
| 	}) | ||||
| } | ||||
| 
 | ||||
|  |  | |||
|  | @ -99,12 +99,16 @@ func (m *Module) Route(r *router.Router, mi ...gin.HandlerFunc) { | |||
| 
 | ||||
| 	// Handlers that serve profiles and statuses should use | ||||
| 	// the SignatureCheck middleware, so that requests with | ||||
| 	// content-type application/activity+json can be served | ||||
| 	// content-type application/activity+json can be served, | ||||
| 	// and (if enabled) the nollamas middleware, to protect | ||||
| 	// against scraping by shitty LLM bullshit. | ||||
| 	profileGroup := r.AttachGroup(profileGroupPath) | ||||
| 	profileGroup.Use(mi...) | ||||
| 	profileGroup.Use(middleware.SignatureCheck(m.isURIBlocked), middleware.CacheControl(middleware.CacheControlConfig{ | ||||
| 		Directives: []string{"no-store"}, | ||||
| 	})) | ||||
| 	nollamas := middleware.NoLLaMas(m.processor.InstanceGetV1) | ||||
| 	profileGroup.Use(nollamas) | ||||
| 	profileGroup.Handle(http.MethodGet, "", m.profileGETHandler) // use empty path here since it's the base of the group | ||||
| 	profileGroup.Handle(http.MethodGet, statusPath, m.threadGETHandler) | ||||
| 
 | ||||
|  |  | |||
|  | @ -73,6 +73,24 @@ skulk({ | |||
| 				["babelify", { global: true }] | ||||
| 			], | ||||
| 		}, | ||||
| 		nollamas: { | ||||
| 			entryFile: "nollamas", | ||||
| 			outputFile: "nollamas.js", | ||||
| 			preset: ["js"], | ||||
| 			prodCfg: prodCfg, | ||||
| 			transform: [ | ||||
| 				["babelify", { global: true }] | ||||
| 			], | ||||
| 		}, | ||||
| 		nollamasworker: { | ||||
| 			entryFile: "nollamasworker", | ||||
| 			outputFile: "nollamasworker.js", | ||||
| 			preset: ["js"], | ||||
| 			prodCfg: prodCfg, | ||||
| 			transform: [ | ||||
| 				["babelify", { global: true }] | ||||
| 			], | ||||
| 		}, | ||||
| 		settings: { | ||||
| 			entryFile: "settings", | ||||
| 			outputFile: "settings.js", | ||||
|  |  | |||
							
								
								
									
										52
									
								
								web/source/nollamas/index.js
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								web/source/nollamas/index.js
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,52 @@ | |||
| /* | ||||
| 	GoToSocial | ||||
| 	Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||
| 	SPDX-License-Identifier: AGPL-3.0-or-later | ||||
| 
 | ||||
| 	This program is free software: you can redistribute it and/or modify | ||||
| 	it under the terms of the GNU Affero General Public License as published by | ||||
| 	the Free Software Foundation, either version 3 of the License, or | ||||
| 	(at your option) any later version. | ||||
| 
 | ||||
| 	This program is distributed in the hope that it will be useful, | ||||
| 	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| 	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| 	GNU Affero General Public License for more details. | ||||
| 
 | ||||
| 	You should have received a copy of the GNU Affero General Public License | ||||
| 	along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | ||||
| */ | ||||
| 
 | ||||
| document.addEventListener('DOMContentLoaded', function() { | ||||
| 	// Read the challenge and difficulty from
 | ||||
| 	// data attributes on the nollamas section.
 | ||||
| 	const nollamas = document.querySelector(".nollamas"); | ||||
| 	const challenge = nollamas.dataset.nollamasChallenge; | ||||
| 	const difficulty = nollamas.dataset.nollamasDifficulty; | ||||
| 	 | ||||
| 	console.log('challenge:', challenge); | ||||
| 	console.log('difficulty:', difficulty); | ||||
| 	 | ||||
| 	// Not sure what this is for. Kim help??
 | ||||
| 	const jsOnlyElements = document.querySelectorAll('.hidden'); | ||||
| 	jsOnlyElements.forEach(el => { | ||||
| 		el.classList.remove('hidden'); | ||||
| 	}); | ||||
| 
 | ||||
| 	// Prepare the worker with task function.
 | ||||
| 	const worker = new Worker("/assets/dist/nollamasworker.js"); | ||||
| 	worker.postMessage({ | ||||
| 		challenge: challenge, | ||||
| 		difficulty: difficulty, | ||||
| 	}); | ||||
| 
 | ||||
| 	// Set the main worker function.
 | ||||
| 	worker.onmessage = function (e) { | ||||
| 		if (e.data.done) { | ||||
| 			console.log('solution found for:', e.data.nonce); | ||||
| 			let url = new URL(window.location.href); | ||||
| 			url.searchParams.append('nollamas_solution', e.data.nonce); | ||||
| 			window.location.href = url.toString(); | ||||
| 		} | ||||
| 	}; | ||||
| }); | ||||
							
								
								
									
										52
									
								
								web/source/nollamasworker/index.js
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								web/source/nollamasworker/index.js
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,52 @@ | |||
| /* | ||||
| 	GoToSocial | ||||
| 	Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||
| 	SPDX-License-Identifier: AGPL-3.0-or-later | ||||
| 
 | ||||
| 	This program is free software: you can redistribute it and/or modify | ||||
| 	it under the terms of the GNU Affero General Public License as published by | ||||
| 	the Free Software Foundation, either version 3 of the License, or | ||||
| 	(at your option) any later version. | ||||
| 
 | ||||
| 	This program is distributed in the hope that it will be useful, | ||||
| 	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| 	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| 	GNU Affero General Public License for more details. | ||||
| 
 | ||||
| 	You should have received a copy of the GNU Affero General Public License | ||||
| 	along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | ||||
| */ | ||||
| 
 | ||||
| onmessage = async function(e) { | ||||
| 	console.log('worker started'); | ||||
| 
 | ||||
| 	const challenge = e.data.challenge; | ||||
| 	const textEncoder = new TextEncoder(); | ||||
| 
 | ||||
| 	// Get difficulty and generate the expected
 | ||||
| 	// zero ASCII prefix to check for in hashes.
 | ||||
| 	const difficultyStr = e.data.difficulty; | ||||
| 	const difficulty = parseInt(difficultyStr, 10); | ||||
| 	const zeroPrefix = '0'.repeat(difficulty); | ||||
| 
 | ||||
| 	let nonce = 0; | ||||
| 	while (true) { | ||||
| 		// Create possible solution string from challenge + nonce.
 | ||||
| 		const solution = textEncoder.encode(challenge + nonce.toString()); | ||||
| 
 | ||||
| 		// Generate SHA256 hashsum of solution string and hex encode the result.
 | ||||
| 		const hashBuffer = await crypto.subtle.digest('SHA-256', solution); | ||||
| 		const hashArray = Array.from(new Uint8Array(hashBuffer)); | ||||
| 		const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join(''); | ||||
| 
 | ||||
| 		// Check if the hex encoded hash has
 | ||||
| 		// difficulty defined zeroes prefix.
 | ||||
| 		if (hashHex.startsWith(zeroPrefix)) { | ||||
| 			postMessage({ nonce: nonce, done: true }); | ||||
| 			break; | ||||
| 		} | ||||
| 
 | ||||
| 		// Iter.
 | ||||
| 		nonce++; | ||||
| 	} | ||||
| }; | ||||
|  | @ -1,97 +1,32 @@ | |||
| <!DOCTYPE html> | ||||
| <html> | ||||
| {{- /* | ||||
| // GoToSocial | ||||
| // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||
| // SPDX-License-Identifier: AGPL-3.0-or-later | ||||
| // | ||||
| // This program is free software: you can redistribute it and/or modify | ||||
| // it under the terms of the GNU Affero General Public License as published by | ||||
| // the Free Software Foundation, either version 3 of the License, or | ||||
| // (at your option) any later version. | ||||
| // | ||||
| // This program is distributed in the hope that it will be useful, | ||||
| // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| // GNU Affero General Public License for more details. | ||||
| // | ||||
| // You should have received a copy of the GNU Affero General Public License | ||||
| // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
| */ -}} | ||||
| 
 | ||||
| <head> | ||||
|     <title>Verifying...</title> | ||||
|     <meta charset="utf-8"> | ||||
|     <meta name="viewport" content="width=device-width, initial-scale=1"> | ||||
| </head> | ||||
| 
 | ||||
| <body> | ||||
|     <script> | ||||
|         document.addEventListener('DOMContentLoaded', function() { | ||||
|             const jsOnlyElements = document.querySelectorAll('.hidden'); | ||||
|             jsOnlyElements.forEach(el => { | ||||
|                 el.classList.remove('hidden'); | ||||
|             }); | ||||
|         }); | ||||
| 
 | ||||
|         // Define our worker task func. | ||||
|         const workerTask = function() { | ||||
|             onmessage = async function(e) { | ||||
|                 console.log('worker started'); | ||||
| 
 | ||||
|                 const challenge = e.data.challenge; | ||||
|                 const textEncoder = new TextEncoder(); | ||||
| 
 | ||||
|                 // Get difficulty and generate the expected | ||||
|                 // zero ASCII prefix to check for in hashes. | ||||
|                 const difficultyStr = e.data.difficulty; | ||||
|                 const difficulty = parseInt(difficultyStr, 10); | ||||
|                 const zeroPrefix = '0'.repeat(difficulty); | ||||
| 
 | ||||
|                 let nonce = 0; | ||||
|                 while (true) { | ||||
|                     // Create possible solution string from challenge + nonce. | ||||
|                     const solution = textEncoder.encode(challenge + nonce.toString()); | ||||
| 
 | ||||
|                     // Generate SHA256 hashsum of solution string and hex encode the result. | ||||
|                     const hashBuffer = await crypto.subtle.digest('SHA-256', solution); | ||||
|                     const hashArray = Array.from(new Uint8Array(hashBuffer)); | ||||
|                     const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join(''); | ||||
| 
 | ||||
|                     // Check if the hex encoded hash has | ||||
|                     // difficulty defined zeroes prefix. | ||||
|                     if (hashHex.startsWith(zeroPrefix)) { | ||||
|                         postMessage({ nonce: nonce, done: true }); | ||||
|                         break; | ||||
|                     } | ||||
| 
 | ||||
|                     // Iter. | ||||
|                     nonce++; | ||||
|                 } | ||||
|             } | ||||
|         }; | ||||
| 
 | ||||
|         // Convert the worker task function to call-able base64 blob URL. | ||||
|         const workerTaskBlob = new Blob(['(',workerTask.toString(),')()'], | ||||
|             { type: 'application/javascript' }); | ||||
|         const workerTaskURL = URL.createObjectURL(workerTaskBlob); | ||||
| 
 | ||||
|         const challenge = '{{ .challenge }}'; | ||||
|         const difficulty = '{{ .difficulty }}'; | ||||
|         console.log('challenge:', challenge); | ||||
|         console.log('difficulty:', difficulty); | ||||
| 
 | ||||
|         // Prepare the worker with task function. | ||||
|         const worker = new Worker(workerTaskURL); | ||||
|         worker.postMessage({ | ||||
|             challenge: challenge, | ||||
|             difficulty: difficulty, | ||||
|         }); | ||||
| 
 | ||||
|         // Set the main worker function. | ||||
|         worker.onmessage = function (e) { | ||||
|             if (e.data.done) { | ||||
|                 console.log('solution found for:', e.data.nonce); | ||||
|                 let url = new URL(window.location.href); | ||||
|                 url.searchParams.append('nollamas_solution', e.data.nonce); | ||||
|                 window.location.href = url.toString(); | ||||
|             } | ||||
|         }; | ||||
|     </script> | ||||
|     <div style="display: flex; align-items: center; justify-content: center; min-width: 100%; min-height: 100%;"> | ||||
|         <div style="display: flex; flex-direction: column; align-items: center; justify-content: center; width: 75%; text-align: center;"> | ||||
|             <p class="hidden" style="margin-bottom: 0.25rem;"><svg fill="var(--color-1)" style="width: clamp(64px, 15%, 96px); height: auto;" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><style>.spinner_d9Sa{transform-origin:center}.spinner_qQQY{animation:spinner_ZpfF 9s linear infinite}.spinner_pote{animation:spinner_ZpfF .75s linear infinite}@keyframes spinner_ZpfF{100%{transform:rotate(360deg)}}</style><path d="M12,1A11,11,0,1,0,23,12,11,11,0,0,0,12,1Zm0,20a9,9,0,1,1,9-9A9,9,0,0,1,12,21Z"/><rect class="spinner_d9Sa spinner_qQQY" x="11" y="6" rx="1" width="2" height="7"/><rect class="spinner_d9Sa spinner_pote" x="11" y="11" rx="1" width="2" height="9"/></svg></p> | ||||
|             <p class="hidden" style="margin-top: 0.5rem; max-width: 24rem;">One moment while we verify your connection...</p> | ||||
|             <noscript> | ||||
|                 <p style="display: flex; align-items: center; gap: 0.5rem;"> | ||||
|                     <svg width="24px" height="24px" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg" stroke="#000000"><g id="SVGRepo_bgCarrier" stroke-width="0"></g><g id="SVGRepo_tracerCarrier" stroke-linecap="round" stroke-linejoin="round"></g><g id="SVGRepo_iconCarrier"><path d="M12 10V13" stroke="#ff7800" stroke-width="2" stroke-linecap="round"></path><path d="M12 16V15.9888" stroke="#ff7800" stroke-width="2" stroke-linecap="round"></path><path d="M10.2518 5.147L3.6508 17.0287C2.91021 18.3618 3.87415 20 5.39912 20H18.6011C20.126 20 21.09 18.3618 20.3494 17.0287L13.7484 5.147C12.9864 3.77538 11.0138 3.77538 10.2518 5.147Z" stroke="#ff7800" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></g></svg> | ||||
|                     Javascript must be enabled to verify your browser. | ||||
|                 </p> | ||||
|             </noscript> | ||||
|         </div> | ||||
|     </div> | ||||
| </body> | ||||
| 
 | ||||
| </html> | ||||
| {{- with . }} | ||||
| <main> | ||||
|     <section class="nollamas" | ||||
|         data-nollamas-challenge="{{ .challenge }}" | ||||
|         data-nollamas-difficulty="{{ .difficulty }}" | ||||
|     > | ||||
|         <p>One moment while we verify your connection...</p> | ||||
|         <noscript> | ||||
|             <p>Javascript must be enabled to verify your browser!</p> | ||||
|         </noscript> | ||||
|     </section> | ||||
| </main> | ||||
| {{- end }} | ||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue