utils: Add convenience function for setting fd flags
[asterisk/asterisk.git] / res / res_pktccops.c
index 4f69448..156c49d 100644 (file)
@@ -648,7 +648,7 @@ static struct cops_gate *cops_gate_cmd(int cmd, struct cops_cmts *cmts,
 
 static int cops_connect(char *host, char *port)
 {
-       int s, sfd = -1, flags;
+       int s, sfd = -1;
        struct addrinfo hints;
        struct addrinfo *rp;
        struct addrinfo *result;
@@ -674,8 +674,7 @@ static int cops_connect(char *host, char *port)
                if (sfd == -1) {
                        ast_log(LOG_WARNING, "Failed socket\n");
                }
-               flags = fcntl(sfd, F_GETFL);
-               fcntl(sfd, F_SETFL, flags | O_NONBLOCK);
+               ast_fd_set_flags(sfd, O_NONBLOCK);
 #ifdef HAVE_SO_NOSIGPIPE
                setsockopt(sfd, SOL_SOCKET, SO_NOSIGPIPE, &trueval, sizeof(trueval));
 #endif
@@ -1472,6 +1471,10 @@ static int load_module(void)
        }
        ast_cli_register_multiple(cli_pktccops, sizeof(cli_pktccops) / sizeof(struct ast_cli_entry));
        restart_pktc_thread();
+
+       /* For Optional API. */
+       ast_module_shutdown_ref(AST_MODULE_SELF);
+
        return 0;
 }