Annotation of embedaddon/strongswan/src/libcharon/plugins/kernel_netlink/kernel_netlink_shared.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2014 Martin Willi
        !             3:  * Copyright (C) 2014 revosec AG
        !             4:  *
        !             5:  * Copyright (C) 2008-2020 Tobias Brunner
        !             6:  * HSR Hochschule fuer Technik Rapperswil
        !             7:  *
        !             8:  * This program is free software; you can redistribute it and/or modify it
        !             9:  * under the terms of the GNU General Public License as published by the
        !            10:  * Free Software Foundation; either version 2 of the License, or (at your
        !            11:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
        !            12:  *
        !            13:  * This program is distributed in the hope that it will be useful, but
        !            14:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
        !            15:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
        !            16:  * for more details.
        !            17:  */
        !            18: 
        !            19: /*
        !            20:  * Copyright (C) 2016 secunet Security Networks AG
        !            21:  * Copyright (C) 2016 Thomas Egerer
        !            22:  *
        !            23:  * Permission is hereby granted, free of charge, to any person obtaining a copy
        !            24:  * of this software and associated documentation files (the "Software"), to deal
        !            25:  * in the Software without restriction, including without limitation the rights
        !            26:  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
        !            27:  * copies of the Software, and to permit persons to whom the Software is
        !            28:  * furnished to do so, subject to the following conditions:
        !            29:  *
        !            30:  * The above copyright notice and this permission notice shall be included in
        !            31:  * all copies or substantial portions of the Software.
        !            32:  *
        !            33:  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
        !            34:  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
        !            35:  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
        !            36:  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
        !            37:  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
        !            38:  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
        !            39:  * THE SOFTWARE.
        !            40:  */
        !            41: 
        !            42: #include <sys/socket.h>
        !            43: #include <linux/netlink.h>
        !            44: #include <linux/rtnetlink.h>
        !            45: #include <linux/xfrm.h>
        !            46: #include <errno.h>
        !            47: #include <unistd.h>
        !            48: 
        !            49: #include "kernel_netlink_shared.h"
        !            50: 
        !            51: #include <utils/debug.h>
        !            52: #include <threading/mutex.h>
        !            53: #include <threading/condvar.h>
        !            54: #include <collections/array.h>
        !            55: #include <collections/hashtable.h>
        !            56: 
        !            57: typedef struct private_netlink_socket_t private_netlink_socket_t;
        !            58: 
        !            59: /**
        !            60:  * Private variables and functions of netlink_socket_t class.
        !            61:  */
        !            62: struct private_netlink_socket_t {
        !            63: 
        !            64:        /**
        !            65:         * public part of the netlink_socket_t object.
        !            66:         */
        !            67:        netlink_socket_t public;
        !            68: 
        !            69:        /**
        !            70:         * mutex to lock access entries
        !            71:         */
        !            72:        mutex_t *mutex;
        !            73: 
        !            74:        /**
        !            75:         * Netlink request entries currently active, uintptr_t seq => entry_t
        !            76:         */
        !            77:        hashtable_t *entries;
        !            78: 
        !            79:        /**
        !            80:         * Current sequence number for Netlink requests
        !            81:         */
        !            82:        refcount_t seq;
        !            83: 
        !            84:        /**
        !            85:         * netlink socket
        !            86:         */
        !            87:        int socket;
        !            88: 
        !            89:        /**
        !            90:         * Netlink protocol
        !            91:         */
        !            92:        int protocol;
        !            93: 
        !            94:        /**
        !            95:         * Enum names for Netlink messages
        !            96:         */
        !            97:        enum_name_t *names;
        !            98: 
        !            99:        /**
        !           100:         * Timeout for Netlink replies, in ms
        !           101:         */
        !           102:        u_int timeout;
        !           103: 
        !           104:        /**
        !           105:         * Number of times to repeat timed out queries
        !           106:         */
        !           107:        u_int retries;
        !           108: 
        !           109:        /**
        !           110:         * Buffer size for received Netlink messages
        !           111:         */
        !           112:        u_int buflen;
        !           113: 
        !           114:        /**
        !           115:         * Use parallel netlink queries
        !           116:         */
        !           117:        bool parallel;
        !           118: 
        !           119:        /**
        !           120:         * Ignore errors potentially resulting from a retransmission
        !           121:         */
        !           122:        bool ignore_retransmit_errors;
        !           123: };
        !           124: 
        !           125: /**
        !           126:  * #definable hook to simulate request message loss
        !           127:  */
        !           128: #ifdef NETLINK_MSG_LOSS_HOOK
        !           129: bool NETLINK_MSG_LOSS_HOOK(struct nlmsghdr *msg);
        !           130: #define msg_loss_hook(msg) NETLINK_MSG_LOSS_HOOK(msg)
        !           131: #else
        !           132: #define msg_loss_hook(msg) FALSE
        !           133: #endif
        !           134: 
        !           135: /**
        !           136:  * Request entry the answer for a waiting thread is collected in
        !           137:  */
        !           138: typedef struct {
        !           139:        /** Condition variable thread is waiting */
        !           140:        condvar_t *condvar;
        !           141:        /** Array of hdrs in a multi-message response, as struct nlmsghdr* */
        !           142:        array_t *hdrs;
        !           143:        /** All response messages received? */
        !           144:        bool complete;
        !           145: } entry_t;
        !           146: 
        !           147: /**
        !           148:  * Clean up a thread waiting entry
        !           149:  */
        !           150: static void destroy_entry(entry_t *entry)
        !           151: {
        !           152:        entry->condvar->destroy(entry->condvar);
        !           153:        array_destroy_function(entry->hdrs, (void*)free, NULL);
        !           154:        free(entry);
        !           155: }
        !           156: 
        !           157: /**
        !           158:  * Write a Netlink message to socket
        !           159:  */
        !           160: static bool write_msg(private_netlink_socket_t *this, struct nlmsghdr *msg)
        !           161: {
        !           162:        struct sockaddr_nl addr = {
        !           163:                .nl_family = AF_NETLINK,
        !           164:        };
        !           165:        int len;
        !           166: 
        !           167:        if (msg_loss_hook(msg))
        !           168:        {
        !           169:                return TRUE;
        !           170:        }
        !           171: 
        !           172:        while (TRUE)
        !           173:        {
        !           174:                len = sendto(this->socket, msg, msg->nlmsg_len, 0,
        !           175:                                         (struct sockaddr*)&addr, sizeof(addr));
        !           176:                if (len != msg->nlmsg_len)
        !           177:                {
        !           178:                        if (errno == EINTR)
        !           179:                        {
        !           180:                                continue;
        !           181:                        }
        !           182:                        DBG1(DBG_KNL, "netlink write error: %s", strerror(errno));
        !           183:                        return FALSE;
        !           184:                }
        !           185:                return TRUE;
        !           186:        }
        !           187: }
        !           188: 
        !           189: /**
        !           190:  * Read a single Netlink message from socket, return 0 on error, -1 on timeout
        !           191:  */
        !           192: static ssize_t read_msg(private_netlink_socket_t *this,
        !           193:                                                char *buf, size_t buflen, bool block)
        !           194: {
        !           195:        ssize_t len;
        !           196: 
        !           197:        if (block)
        !           198:        {
        !           199:                fd_set set;
        !           200:                timeval_t tv = {};
        !           201: 
        !           202:                FD_ZERO(&set);
        !           203:                FD_SET(this->socket, &set);
        !           204:                timeval_add_ms(&tv, this->timeout);
        !           205: 
        !           206:                if (select(this->socket + 1, &set, NULL, NULL,
        !           207:                                   this->timeout ? &tv : NULL) <= 0)
        !           208:                {
        !           209:                        return -1;
        !           210:                }
        !           211:        }
        !           212:        len = recv(this->socket, buf, buflen, MSG_TRUNC|(block ? 0 : MSG_DONTWAIT));
        !           213:        if (len > buflen)
        !           214:        {
        !           215:                DBG1(DBG_KNL, "netlink response exceeds buffer size");
        !           216:                return 0;
        !           217:        }
        !           218:        if (len < 0)
        !           219:        {
        !           220:                if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR)
        !           221:                {
        !           222:                        DBG1(DBG_KNL, "netlink read error: %s", strerror(errno));
        !           223:                }
        !           224:                return 0;
        !           225:        }
        !           226:        return len;
        !           227: }
        !           228: 
        !           229: /**
        !           230:  * Queue received response message
        !           231:  */
        !           232: static bool queue(private_netlink_socket_t *this, struct nlmsghdr *buf)
        !           233: {
        !           234:        struct nlmsghdr *hdr;
        !           235:        entry_t *entry;
        !           236:        uintptr_t seq;
        !           237: 
        !           238:        seq = (uintptr_t)buf->nlmsg_seq;
        !           239: 
        !           240:        this->mutex->lock(this->mutex);
        !           241:        entry = this->entries->get(this->entries, (void*)seq);
        !           242:        if (entry)
        !           243:        {
        !           244:                hdr = malloc(buf->nlmsg_len);
        !           245:                memcpy(hdr, buf, buf->nlmsg_len);
        !           246:                array_insert(entry->hdrs, ARRAY_TAIL, hdr);
        !           247:                if (hdr->nlmsg_type == NLMSG_DONE || !(hdr->nlmsg_flags & NLM_F_MULTI))
        !           248:                {
        !           249:                        entry->complete = TRUE;
        !           250:                        entry->condvar->signal(entry->condvar);
        !           251:                }
        !           252:        }
        !           253:        else
        !           254:        {
        !           255:                DBG1(DBG_KNL, "received unknown netlink seq %u, ignored", seq);
        !           256:        }
        !           257:        this->mutex->unlock(this->mutex);
        !           258: 
        !           259:        return entry != NULL;
        !           260: }
        !           261: 
        !           262: /**
        !           263:  * Read and queue response message, optionally blocking, returns TRUE on timeout
        !           264:  */
        !           265: static bool read_and_queue(private_netlink_socket_t *this, bool block)
        !           266: {
        !           267:        struct nlmsghdr *hdr;
        !           268:        char buf[this->buflen];
        !           269:        ssize_t len, read_len;
        !           270:        bool wipe = FALSE;
        !           271: 
        !           272:        len = read_len = read_msg(this, buf, sizeof(buf), block);
        !           273:        if (len == -1)
        !           274:        {
        !           275:                return TRUE;
        !           276:        }
        !           277:        if (len)
        !           278:        {
        !           279:                hdr = (struct nlmsghdr*)buf;
        !           280:                while (NLMSG_OK(hdr, len))
        !           281:                {
        !           282:                        if (this->protocol == NETLINK_XFRM &&
        !           283:                                hdr->nlmsg_type == XFRM_MSG_NEWSA)
        !           284:                        {       /* wipe potential IPsec SA keys */
        !           285:                                wipe = TRUE;
        !           286:                        }
        !           287:                        if (!queue(this, hdr))
        !           288:                        {
        !           289:                                break;
        !           290:                        }
        !           291:                        hdr = NLMSG_NEXT(hdr, len);
        !           292:                }
        !           293:        }
        !           294:        if (wipe)
        !           295:        {
        !           296:                memwipe(buf, read_len);
        !           297:        }
        !           298:        return FALSE;
        !           299: }
        !           300: 
        !           301: CALLBACK(watch, bool,
        !           302:        private_netlink_socket_t *this, int fd, watcher_event_t event)
        !           303: {
        !           304:        if (event == WATCHER_READ)
        !           305:        {
        !           306:                read_and_queue(this, FALSE);
        !           307:        }
        !           308:        return TRUE;
        !           309: }
        !           310: 
        !           311: /**
        !           312:  * Send a netlink request, try once
        !           313:  */
        !           314: static status_t send_once(private_netlink_socket_t *this, struct nlmsghdr *in,
        !           315:                                                  uintptr_t seq, struct nlmsghdr **out, size_t *out_len)
        !           316: {
        !           317:        struct nlmsghdr *hdr;
        !           318:        entry_t *entry;
        !           319:        u_char *ptr;
        !           320:        int i;
        !           321: 
        !           322:        in->nlmsg_seq = seq;
        !           323:        in->nlmsg_pid = getpid();
        !           324: 
        !           325:        if (this->names)
        !           326:        {
        !           327:                DBG3(DBG_KNL, "sending %N %u: %b", this->names, in->nlmsg_type,
        !           328:                         (u_int)seq, in, in->nlmsg_len);
        !           329:        }
        !           330: 
        !           331:        this->mutex->lock(this->mutex);
        !           332:        if (!write_msg(this, in))
        !           333:        {
        !           334:                this->mutex->unlock(this->mutex);
        !           335:                return FAILED;
        !           336:        }
        !           337: 
        !           338:        INIT(entry,
        !           339:                .condvar = condvar_create(CONDVAR_TYPE_DEFAULT),
        !           340:                .hdrs = array_create(0, 0),
        !           341:        );
        !           342:        this->entries->put(this->entries, (void*)seq, entry);
        !           343: 
        !           344:        while (!entry->complete)
        !           345:        {
        !           346:                if (this->parallel &&
        !           347:                        lib->watcher->get_state(lib->watcher) != WATCHER_STOPPED &&
        !           348:                        lib->processor->get_total_threads(lib->processor))
        !           349:                {
        !           350:                        if (this->timeout)
        !           351:                        {
        !           352:                                if (entry->condvar->timed_wait(entry->condvar, this->mutex,
        !           353:                                                                                           this->timeout))
        !           354:                                {
        !           355:                                        break;
        !           356:                                }
        !           357:                        }
        !           358:                        else
        !           359:                        {
        !           360:                                entry->condvar->wait(entry->condvar, this->mutex);
        !           361:                        }
        !           362:                }
        !           363:                else
        !           364:                {       /* During (de-)initialization, no watcher thread is active.
        !           365:                         * collect responses ourselves. */
        !           366:                        if (read_and_queue(this, TRUE))
        !           367:                        {
        !           368:                                break;
        !           369:                        }
        !           370:                }
        !           371:        }
        !           372:        this->entries->remove(this->entries, (void*)seq);
        !           373: 
        !           374:        this->mutex->unlock(this->mutex);
        !           375: 
        !           376:        if (!entry->complete)
        !           377:        {       /* timeout */
        !           378:                destroy_entry(entry);
        !           379:                return OUT_OF_RES;
        !           380:        }
        !           381: 
        !           382:        for (i = 0, *out_len = 0; i < array_count(entry->hdrs); i++)
        !           383:        {
        !           384:                array_get(entry->hdrs, i, &hdr);
        !           385:                *out_len += NLMSG_ALIGN(hdr->nlmsg_len);
        !           386:        }
        !           387:        ptr = malloc(*out_len);
        !           388:        *out = (struct nlmsghdr*)ptr;
        !           389: 
        !           390:        while (array_remove(entry->hdrs, ARRAY_HEAD, &hdr))
        !           391:        {
        !           392:                if (this->names)
        !           393:                {
        !           394:                        DBG3(DBG_KNL, "received %N %u: %b", this->names, hdr->nlmsg_type,
        !           395:                                 hdr->nlmsg_seq, hdr, hdr->nlmsg_len);
        !           396:                }
        !           397:                memcpy(ptr, hdr, hdr->nlmsg_len);
        !           398:                ptr += NLMSG_ALIGN(hdr->nlmsg_len);
        !           399:                free(hdr);
        !           400:        }
        !           401:        destroy_entry(entry);
        !           402:        return SUCCESS;
        !           403: }
        !           404: 
        !           405: /**
        !           406:  * Ignore errors for message types that might have completed previously
        !           407:  */
        !           408: static void ignore_retransmit_error(private_netlink_socket_t *this,
        !           409:                                                                        struct nlmsgerr *err, int type)
        !           410: {
        !           411:        switch (err->error)
        !           412:        {
        !           413:                case -EEXIST:
        !           414:                        switch (this->protocol)
        !           415:                        {
        !           416:                                case NETLINK_XFRM:
        !           417:                                        switch (type)
        !           418:                                        {
        !           419:                                                case XFRM_MSG_NEWPOLICY:
        !           420:                                                case XFRM_MSG_NEWSA:
        !           421:                                                        err->error = 0;
        !           422:                                                        break;
        !           423:                                        }
        !           424:                                        break;
        !           425:                                case NETLINK_ROUTE:
        !           426:                                        switch (type)
        !           427:                                        {
        !           428:                                                case RTM_NEWADDR:
        !           429:                                                case RTM_NEWLINK:
        !           430:                                                case RTM_NEWNEIGH:
        !           431:                                                case RTM_NEWROUTE:
        !           432:                                                case RTM_NEWRULE:
        !           433:                                                        err->error = 0;
        !           434:                                                        break;
        !           435:                                        }
        !           436:                                        break;
        !           437:                        }
        !           438:                        break;
        !           439:                case -ENOENT:
        !           440:                        switch (this->protocol)
        !           441:                        {
        !           442:                                case NETLINK_XFRM:
        !           443:                                        switch (type)
        !           444:                                        {
        !           445:                                                case XFRM_MSG_DELPOLICY:
        !           446:                                                case XFRM_MSG_DELSA:
        !           447:                                                        err->error = 0;
        !           448:                                                        break;
        !           449:                                        }
        !           450:                                        break;
        !           451:                                case NETLINK_ROUTE:
        !           452:                                        switch (type)
        !           453:                                        {
        !           454:                                                case RTM_DELADDR:
        !           455:                                                case RTM_DELLINK:
        !           456:                                                case RTM_DELNEIGH:
        !           457:                                                case RTM_DELROUTE:
        !           458:                                                case RTM_DELRULE:
        !           459:                                                        err->error = 0;
        !           460:                                                        break;
        !           461:                                        }
        !           462:                                        break;
        !           463:                        }
        !           464:                        break;
        !           465:        }
        !           466: }
        !           467: 
        !           468: METHOD(netlink_socket_t, netlink_send, status_t,
        !           469:        private_netlink_socket_t *this, struct nlmsghdr *in, struct nlmsghdr **out,
        !           470:        size_t *out_len)
        !           471: {
        !           472:        uintptr_t seq;
        !           473:        u_int try;
        !           474: 
        !           475:        seq = ref_get(&this->seq);
        !           476: 
        !           477:        for (try = 0; try <= this->retries; ++try)
        !           478:        {
        !           479:                struct nlmsghdr *hdr;
        !           480:                status_t status;
        !           481:                size_t len;
        !           482: 
        !           483:                if (try > 0)
        !           484:                {
        !           485:                        DBG1(DBG_KNL, "retransmitting Netlink request (%u/%u)",
        !           486:                                 try, this->retries);
        !           487:                }
        !           488:                status = send_once(this, in, seq, &hdr, &len);
        !           489:                switch (status)
        !           490:                {
        !           491:                        case SUCCESS:
        !           492:                                break;
        !           493:                        case OUT_OF_RES:
        !           494:                                continue;
        !           495:                        default:
        !           496:                                return status;
        !           497:                }
        !           498:                if (hdr->nlmsg_type == NLMSG_ERROR)
        !           499:                {
        !           500:                        struct nlmsgerr* err;
        !           501: 
        !           502:                        err = NLMSG_DATA(hdr);
        !           503:                        if (err->error == -EBUSY)
        !           504:                        {
        !           505:                                free(hdr);
        !           506:                                try--;
        !           507:                                continue;
        !           508:                        }
        !           509:                        if (this->ignore_retransmit_errors && try > 0)
        !           510:                        {
        !           511:                                ignore_retransmit_error(this, err, in->nlmsg_type);
        !           512:                        }
        !           513:                }
        !           514:                *out = hdr;
        !           515:                *out_len = len;
        !           516:                return SUCCESS;
        !           517:        }
        !           518:        DBG1(DBG_KNL, "Netlink request timed out after %u retransmits",
        !           519:                 this->retries);
        !           520:        return OUT_OF_RES;
        !           521: }
        !           522: 
        !           523: METHOD(netlink_socket_t, netlink_send_ack, status_t,
        !           524:        private_netlink_socket_t *this, struct nlmsghdr *in)
        !           525: {
        !           526:        struct nlmsghdr *out, *hdr;
        !           527:        size_t len;
        !           528: 
        !           529:        if (netlink_send(this, in, &out, &len) != SUCCESS)
        !           530:        {
        !           531:                return FAILED;
        !           532:        }
        !           533:        hdr = out;
        !           534:        while (NLMSG_OK(hdr, len))
        !           535:        {
        !           536:                switch (hdr->nlmsg_type)
        !           537:                {
        !           538:                        case NLMSG_ERROR:
        !           539:                        {
        !           540:                                struct nlmsgerr* err = NLMSG_DATA(hdr);
        !           541: 
        !           542:                                if (err->error)
        !           543:                                {
        !           544:                                        if (-err->error == EEXIST)
        !           545:                                        {       /* do not report existing routes */
        !           546:                                                free(out);
        !           547:                                                return ALREADY_DONE;
        !           548:                                        }
        !           549:                                        if (-err->error == ESRCH)
        !           550:                                        {       /* do not report missing entries */
        !           551:                                                free(out);
        !           552:                                                return NOT_FOUND;
        !           553:                                        }
        !           554:                                        DBG1(DBG_KNL, "received netlink error: %s (%d)",
        !           555:                                                 strerror(-err->error), -err->error);
        !           556:                                        free(out);
        !           557:                                        return FAILED;
        !           558:                                }
        !           559:                                free(out);
        !           560:                                return SUCCESS;
        !           561:                        }
        !           562:                        default:
        !           563:                                hdr = NLMSG_NEXT(hdr, len);
        !           564:                                continue;
        !           565:                        case NLMSG_DONE:
        !           566:                                break;
        !           567:                }
        !           568:                break;
        !           569:        }
        !           570:        DBG1(DBG_KNL, "netlink request not acknowledged");
        !           571:        free(out);
        !           572:        return FAILED;
        !           573: }
        !           574: 
        !           575: METHOD(netlink_socket_t, destroy, void,
        !           576:        private_netlink_socket_t *this)
        !           577: {
        !           578:        if (this->socket != -1)
        !           579:        {
        !           580:                if (this->parallel)
        !           581:                {
        !           582:                        lib->watcher->remove(lib->watcher, this->socket);
        !           583:                }
        !           584:                close(this->socket);
        !           585:        }
        !           586:        this->entries->destroy(this->entries);
        !           587:        this->mutex->destroy(this->mutex);
        !           588:        free(this);
        !           589: }
        !           590: 
        !           591: /*
        !           592:  * Described in header
        !           593:  */
        !           594: u_int netlink_get_buflen()
        !           595: {
        !           596:        u_int buflen;
        !           597: 
        !           598:        buflen = lib->settings->get_int(lib->settings,
        !           599:                                                                "%s.plugins.kernel-netlink.buflen", 0, lib->ns);
        !           600:        if (!buflen)
        !           601:        {
        !           602:                long pagesize = sysconf(_SC_PAGESIZE);
        !           603: 
        !           604:                if (pagesize == -1)
        !           605:                {
        !           606:                        pagesize = 4096;
        !           607:                }
        !           608:                /* base this on NLMSG_GOODSIZE */
        !           609:                buflen = min(pagesize, 8192);
        !           610:        }
        !           611:        return buflen;
        !           612: }
        !           613: 
        !           614: /*
        !           615:  * Described in header
        !           616:  */
        !           617: netlink_socket_t *netlink_socket_create(int protocol, enum_name_t *names,
        !           618:                                                                                bool parallel)
        !           619: {
        !           620:        private_netlink_socket_t *this;
        !           621:        struct sockaddr_nl addr = {
        !           622:                .nl_family = AF_NETLINK,
        !           623:        };
        !           624:        bool force_buf = FALSE;
        !           625:        int rcvbuf_size = 0;
        !           626: 
        !           627:        INIT(this,
        !           628:                .public = {
        !           629:                        .send = _netlink_send,
        !           630:                        .send_ack = _netlink_send_ack,
        !           631:                        .destroy = _destroy,
        !           632:                },
        !           633:                .seq = 200,
        !           634:                .mutex = mutex_create(MUTEX_TYPE_RECURSIVE),
        !           635:                .socket = socket(AF_NETLINK, SOCK_RAW, protocol),
        !           636:                .entries = hashtable_create(hashtable_hash_ptr, hashtable_equals_ptr, 4),
        !           637:                .protocol = protocol,
        !           638:                .names = names,
        !           639:                .buflen = netlink_get_buflen(),
        !           640:                .timeout = lib->settings->get_int(lib->settings,
        !           641:                                                        "%s.plugins.kernel-netlink.timeout", 0, lib->ns),
        !           642:                .retries = lib->settings->get_int(lib->settings,
        !           643:                                                        "%s.plugins.kernel-netlink.retries", 0, lib->ns),
        !           644:                .ignore_retransmit_errors = lib->settings->get_bool(lib->settings,
        !           645:                                                        "%s.plugins.kernel-netlink.ignore_retransmit_errors",
        !           646:                                                        FALSE, lib->ns),
        !           647:                .parallel = parallel,
        !           648:        );
        !           649: 
        !           650:        if (this->socket == -1)
        !           651:        {
        !           652:                DBG1(DBG_KNL, "unable to create netlink socket: %s (%d)",
        !           653:                         strerror(errno), errno);
        !           654:                destroy(this);
        !           655:                return NULL;
        !           656:        }
        !           657:        if (bind(this->socket, (struct sockaddr*)&addr, sizeof(addr)))
        !           658:        {
        !           659:                DBG1(DBG_KNL, "unable to bind netlink socket: %s (%d)",
        !           660:                         strerror(errno), errno);
        !           661:                destroy(this);
        !           662:                return NULL;
        !           663:        }
        !           664:        rcvbuf_size = lib->settings->get_int(lib->settings,
        !           665:                                                "%s.plugins.kernel-netlink.receive_buffer_size",
        !           666:                                                rcvbuf_size, lib->ns);
        !           667:        if (rcvbuf_size)
        !           668:        {
        !           669:                int optname;
        !           670: 
        !           671:                force_buf = lib->settings->get_bool(lib->settings,
        !           672:                                                "%s.plugins.kernel-netlink.force_receive_buffer_size",
        !           673:                                                force_buf, lib->ns);
        !           674:                optname = force_buf ? SO_RCVBUFFORCE : SO_RCVBUF;
        !           675: 
        !           676:                if (setsockopt(this->socket, SOL_SOCKET, optname, &rcvbuf_size,
        !           677:                                           sizeof(rcvbuf_size)) == -1)
        !           678:                {
        !           679:                        DBG1(DBG_KNL, "failed to %supdate receive buffer size to %d: %s",
        !           680:                                        force_buf ? "forcibly " : "", rcvbuf_size, strerror(errno));
        !           681:                }
        !           682:        }
        !           683:        if (this->parallel)
        !           684:        {
        !           685:                lib->watcher->add(lib->watcher, this->socket, WATCHER_READ, watch, this);
        !           686:        }
        !           687: 
        !           688:        return &this->public;
        !           689: }
        !           690: 
        !           691: /*
        !           692:  * Described in header
        !           693:  */
        !           694: void netlink_add_attribute(struct nlmsghdr *hdr, int rta_type, chunk_t data,
        !           695:                                                  size_t buflen)
        !           696: {
        !           697:        struct rtattr *rta;
        !           698: 
        !           699:        if (NLMSG_ALIGN(hdr->nlmsg_len) + RTA_LENGTH(data.len) > buflen)
        !           700:        {
        !           701:                DBG1(DBG_KNL, "unable to add attribute, buffer too small");
        !           702:                return;
        !           703:        }
        !           704: 
        !           705:        rta = (struct rtattr*)(((char*)hdr) + NLMSG_ALIGN(hdr->nlmsg_len));
        !           706:        rta->rta_type = rta_type;
        !           707:        rta->rta_len = RTA_LENGTH(data.len);
        !           708:        memcpy(RTA_DATA(rta), data.ptr, data.len);
        !           709:        hdr->nlmsg_len = NLMSG_ALIGN(hdr->nlmsg_len) + RTA_ALIGN(rta->rta_len);
        !           710: }
        !           711: 
        !           712: /**
        !           713:  * Add an attribute to the given Netlink message
        !           714:  */
        !           715: static struct rtattr *add_rtattr(struct nlmsghdr *hdr, int buflen, int type,
        !           716:                                                                 int len)
        !           717: {
        !           718:        struct rtattr *rta;
        !           719: 
        !           720:        if (NLMSG_ALIGN(hdr->nlmsg_len) + RTA_LENGTH(len) > buflen)
        !           721:        {
        !           722:                DBG1(DBG_KNL, "unable to add attribute, buffer too small");
        !           723:                return NULL;
        !           724:        }
        !           725: 
        !           726:        rta = ((void*)hdr) + NLMSG_ALIGN(hdr->nlmsg_len);
        !           727:        rta->rta_type = type;
        !           728:        rta->rta_len = RTA_LENGTH(len);
        !           729:        hdr->nlmsg_len = NLMSG_ALIGN(hdr->nlmsg_len) + RTA_ALIGN(rta->rta_len);
        !           730:        return rta;
        !           731: }
        !           732: 
        !           733: /*
        !           734:  * Described in header
        !           735:  */
        !           736: void *netlink_nested_start(struct nlmsghdr *hdr, size_t buflen, int type)
        !           737: {
        !           738:        return add_rtattr(hdr, buflen, type, 0);
        !           739: }
        !           740: 
        !           741: /*
        !           742:  * Described in header
        !           743:  */
        !           744: void netlink_nested_end(struct nlmsghdr *hdr, void *attr)
        !           745: {
        !           746:        struct rtattr *rta = attr;
        !           747:        void *end;
        !           748: 
        !           749:        if (attr)
        !           750:        {
        !           751:                end = (char*)hdr + NLMSG_ALIGN(hdr->nlmsg_len);
        !           752:                rta->rta_len = end - attr;
        !           753:        }
        !           754: }
        !           755: 
        !           756: /*
        !           757:  * Described in header
        !           758:  */
        !           759: void *netlink_reserve(struct nlmsghdr *hdr, int buflen, int type, int len)
        !           760: {
        !           761:        struct rtattr *rta;
        !           762: 
        !           763:        rta = add_rtattr(hdr, buflen, type, len);
        !           764:        if (!rta)
        !           765:        {
        !           766:                return NULL;
        !           767:        }
        !           768:        return RTA_DATA(rta);
        !           769: }
        !           770: 
        !           771: /*
        !           772:  * Described in header
        !           773:  */
        !           774: void route_entry_destroy(route_entry_t *this)
        !           775: {
        !           776:        free(this->if_name);
        !           777:        DESTROY_IF(this->src_ip);
        !           778:        DESTROY_IF(this->gateway);
        !           779:        chunk_free(&this->dst_net);
        !           780:        free(this);
        !           781: }
        !           782: 
        !           783: /*
        !           784:  * Described in header
        !           785:  */
        !           786: route_entry_t *route_entry_clone(const route_entry_t *this)
        !           787: {
        !           788:        route_entry_t *route;
        !           789: 
        !           790:        INIT(route,
        !           791:                .if_name = strdupnull(this->if_name),
        !           792:                .src_ip = this->src_ip ? this->src_ip->clone(this->src_ip) : NULL,
        !           793:                .gateway = this->gateway ? this->gateway->clone(this->gateway) : NULL,
        !           794:                .dst_net = chunk_clone(this->dst_net),
        !           795:                .prefixlen = this->prefixlen,
        !           796:                .pass = this->pass,
        !           797:        );
        !           798:        return route;
        !           799: }
        !           800: 
        !           801: /*
        !           802:  * Described in header
        !           803:  */
        !           804: u_int route_entry_hash(const route_entry_t *this)
        !           805: {
        !           806:        return chunk_hash_inc(chunk_from_thing(this->prefixlen),
        !           807:                                                  chunk_hash(this->dst_net));
        !           808: }
        !           809: 
        !           810: /**
        !           811:  * Compare two IP addresses, also accept it if both are NULL
        !           812:  */
        !           813: static bool addrs_null_or_equal(host_t *a, host_t *b)
        !           814: {
        !           815:        return (!a && !b) || (a && b && a->ip_equals(a, b));
        !           816: }
        !           817: 
        !           818: /*
        !           819:  * Described in header
        !           820:  */
        !           821: bool route_entry_equals(const route_entry_t *a, const route_entry_t *b)
        !           822: {
        !           823:        return streq(a->if_name, b->if_name) &&
        !           824:                a->pass == b->pass &&
        !           825:                a->prefixlen == b->prefixlen &&
        !           826:                chunk_equals(a->dst_net, b->dst_net) &&
        !           827:                addrs_null_or_equal(a->src_ip, b->src_ip) &&
        !           828:                addrs_null_or_equal(a->gateway, b->gateway);
        !           829: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>