/* get only the local part of the hostname and set it in cache entry */
static void
setLocalHostName(dnscache_entry_t *etry)
{
uchar *fqdnLower;
uchar *p;
int i;
uchar hostbuf[NI_MAXHOST];
if(glbl.GetPreserveFQDN()) {
prop.AddRef(etry->fqdnLowerCase);
etry->localName = etry->fqdnLowerCase;
goto done;
}
/* strip domain, if configured for this entry */
fqdnLower = propGetSzStr(etry->fqdnLowerCase);
p = (uchar*)strchr((char*)fqdnLower, '.'); /* find start of domain name "machine.example.com" */
if(p == NULL) { /* do we have a domain part? */
prop.AddRef(etry->fqdnLowerCase); /* no! */
etry->localName = etry->fqdnLowerCase;
goto done;
}
i = p - fqdnLower; /* length of hostname */
memcpy(hostbuf, fqdnLower, i);
hostbuf[i] = '\0';
/* at this point, we have not found anything, so we again use the
* already-created complete full name property.
*/
prop.AddRef(etry->fqdnLowerCase);
etry->localName = etry->fqdnLowerCase;
done: return;
}
/* preprocess a batch of messages, that is ready them for actual processing. This is done
* as a first stage and totally in parallel to any other worker active in the system. So
* it helps us keep up the overall concurrency level.
* rgerhards, 2010-06-09
*/
static inline rsRetVal
preprocessBatch(batch_t *pBatch, int *pbShutdownImmediate) {
prop_t *ip;
prop_t *fqdn;
prop_t *localName;
prop_t *propFromHost = NULL;
prop_t *propFromHostIP = NULL;
int bIsPermitted;
msg_t *pMsg;
int i;
rsRetVal localRet;
DEFiRet;
for(i = 0 ; i < pBatch->nElem && !*pbShutdownImmediate ; i++) {
pMsg = pBatch->pElem[i].pMsg;
if((pMsg->msgFlags & NEEDS_ACLCHK_U) != 0) {
DBGPRINTF("msgConsumer: UDP ACL must be checked for message (hostname-based)\n");
if(net.cvthname(pMsg->rcvFrom.pfrominet, &localName, &fqdn, &ip) != RS_RET_OK)
continue;
bIsPermitted = net.isAllowedSender2((uchar*)"UDP",
(struct sockaddr *)pMsg->rcvFrom.pfrominet, (char*)propGetSzStr(fqdn), 1);
if(!bIsPermitted) {
DBGPRINTF("Message from '%s' discarded, not a permitted sender host\n",
propGetSzStr(fqdn));
pBatch->eltState[i] = BATCH_STATE_DISC;
} else {
/* save some of the info we obtained */
MsgSetRcvFrom(pMsg, localName);
CHKiRet(MsgSetRcvFromIP(pMsg, ip));
pMsg->msgFlags &= ~NEEDS_ACLCHK_U;
}
}
if((pMsg->msgFlags & NEEDS_PARSING) != 0) {
if((localRet = parser.ParseMsg(pMsg)) != RS_RET_OK) {
DBGPRINTF("Message discarded, parsing error %d\n", localRet);
pBatch->eltState[i] = BATCH_STATE_DISC;
}
}
}
finalize_it:
if(propFromHost != NULL)
prop.Destruct(&propFromHost);
if(propFromHostIP != NULL)
prop.Destruct(&propFromHostIP);
RETiRet;
}
/* get only the local part of the hostname and set it in cache entry */
static void
setLocalHostName(dnscache_entry_t *etry)
{
uchar *fqdnLower;
uchar *p;
int count;
int i;
uchar hostbuf[NI_MAXHOST];
if(glbl.GetPreserveFQDN()) {
prop.AddRef(etry->fqdnLowerCase);
etry->localName = etry->fqdnLowerCase;
goto done;
}
/* strip domain, if configured for this entry */
fqdnLower = propGetSzStr(etry->fqdnLowerCase);
p = (uchar*)strchr((char*)fqdnLower, '.'); /* find start of domain name "machine.example.com" */
if(p == NULL) { /* do we have a domain part? */
prop.AddRef(etry->fqdnLowerCase); /* no! */
etry->localName = etry->fqdnLowerCase;
goto done;
}
i = p - fqdnLower; /* length of hostname */
memcpy(hostbuf, fqdnLower, i);
hostbuf[i] = '\0';
/* now check if we belong to any of the domain names that were specified
* in the -s command line option. If so, remove and we are done.
*/
if(glbl.GetStripDomains() != NULL) {
count=0;
while(glbl.GetStripDomains()[count]) {
if(strcmp((char*)(p + 1), glbl.GetStripDomains()[count]) == 0) {
prop.CreateStringProp(&etry->localName, hostbuf, i);
goto done;
}
count++;
}
}
/* if we reach this point, we have not found any domain we should strip. Now
* we try and see if the host itself is listed in the -l command line option
* and so should be stripped also. If so, we do it and return. Please note that
* -l list FQDNs, not just the hostname part. If it did just list the hostname, the
* door would be wide-open for all kinds of mixing up of hosts. Because of this,
* you'll see comparison against the full string (pszHostFQDN) below.
*/
if(glbl.GetLocalHosts() != NULL) {
count=0;
while(glbl.GetLocalHosts()[count]) {
if(!strcmp((char*)fqdnLower, (char*)glbl.GetLocalHosts()[count])) {
prop.CreateStringProp(&etry->localName, hostbuf, i);
goto done;
}
count++;
}
}
/* at this point, we have not found anything, so we again use the
* already-created complete full name property.
*/
prop.AddRef(etry->fqdnLowerCase);
etry->localName = etry->fqdnLowerCase;
done: return;
}
