X-Git-Url: https://code.citadel.org/?p=citadel.git;a=blobdiff_plain;f=citadel%2Fmsgbase.c;fp=citadel%2Fmsgbase.c;h=568cefb2e06f6c57175510351cbf1b05acf5b049;hp=ce52cc11365267a1cdd84efe402972e6f288bb65;hb=0b0f8a2280a8f22877baccd2355d7b41b863bfa0;hpb=6d69258f19307e1d889b4979d57ead0fa73e39c6

diff --git a/citadel/msgbase.c b/citadel/msgbase.c
index ce52cc113..568cefb2e 100644
--- a/citadel/msgbase.c
+++ b/citadel/msgbase.c
@@ -38,6 +38,9 @@
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <regex.h>
+
+#include "md5.h"
+
 #include <libcitadel.h>
 #include "citadel.h"
 #include "server.h"
@@ -3539,6 +3542,73 @@ void quickie_message(const char *from,
 	if (recp != NULL) free_recipients(recp);
 }
 
+void flood_protect_quickie_message(const char *from,
+				   const char *fromaddr,
+				   char *to,
+				   char *room,
+				   const char *text, 
+				   int format_type,
+				   const char *subject,
+				   int nCriterions,
+				   const char **CritStr,
+				   long *CritStrLen)
+{
+	int i;
+	struct UseTable ut;
+	u_char rawdigest[MD5_DIGEST_LEN];
+	struct MD5Context md5context;
+	StrBuf *guid;
+	struct cdbdata *cdbut;
+	char timestamp[64];
+	long tslen;
+	time_t ts = time(NULL);
+	time_t tsday = ts / (8*60*60); /* just care for a day... */
+
+	tslen = snprintf(timestamp, sizeof(timestamp), "%ld", tsday);
+	MD5Init(&md5context);
+
+	for (i = 0; i < nCriterions; i++)
+		MD5Update(&md5context,
+			  (const unsigned char*)CritStr[i], CritStrLen[i]);
+	MD5Update(&md5context,
+		  (const unsigned char*)timestamp, tslen);
+	MD5Final(rawdigest, &md5context);
+
+	guid = NewStrBufPlain(NULL,
+			      MD5_DIGEST_LEN * 2 + 12);
+	StrBufHexEscAppend(guid, NULL, rawdigest, MD5_DIGEST_LEN);
+	StrBufAppendBufPlain(guid, HKEY("_fldpt"), 0);
+	if (StrLength(guid) > 40)
+		StrBufCutAt(guid, 40, NULL);
+	/* Find out if we've already sent a similar message */
+	memcpy(ut.ut_msgid, SKEY(guid));
+	ut.ut_timestamp = ts;
+
+	cdbut = cdb_fetch(CDB_USETABLE, SKEY(guid));
+
+	if (cdbut != NULL) {
+		/* yes, we did. flood protection kicks in. */
+		syslog(LOG_DEBUG,
+		       "not sending message again\n");
+		cdb_free(cdbut);
+	}
+
+	/* rewrite the record anyway, to update the timestamp */
+	cdb_store(CDB_USETABLE,
+		  SKEY(guid),
+		  &ut, sizeof(struct UseTable) );
+
+	if (cdbut != NULL) return;
+	/* no, this message isn't sent recently; go ahead. */
+	quickie_message(from,
+			fromaddr,
+			to,
+			room,
+			text, 
+			format_type,
+			subject);
+}
+
 
 /*
  * Back end function used by CtdlMakeMessage() and similar functions