YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

// Copyright (c) YugaByte, Inc.

//

// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except

// in compliance with the License.  You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software distributed under the License

// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

// or implied.  See the License for the specific language governing permissions and limitations

// under the License.

#include "yb/util/signal_util.h"

#include <signal.h>

#include "yb/util/errno.h"

namespace yb {

Result<sigset_t> ThreadSignalMaskBlock(const std::vector<int>& signals_to_block) {

  // Note that in case of an error sigaddset sets errno and returns -1

  // while pthread_sigmask returns errno directly.

  sigset_t mask;

  sigemptyset(&mask);

  for (int sig : signals_to_block) {

    int err = sigaddset(&mask, sig);

    if (err != 0) {

      return STATUS(InternalError, "sigaddset failed", Errno(errno));

    }

  }

  sigset_t old_mask;

  int err = pthread_sigmask(SIG_BLOCK, &mask, &old_mask);

  return (err == 0

      ? Result<sigset_t>(old_mask)

      : 
STATUS0
(InternalError, "SIG_BLOCK failed", Errno(err)));

}

Status ThreadSignalMaskRestore(sigset_t old_mask) {

  int err = pthread_sigmask(SIG_SETMASK, &old_mask, NULL /* oldset */);

  return (err == 0

      ? Status::OK()

      : 
STATUS0
(InternalError, "SIG_SETMASK failed", Errno(err)));

}

const std::vector<int> kYsqlHandledSignals{

    // Following handlers are installed in StartBackgroundWorker:

    SIGINT, // StatementCancelHandler

    SIGUSR1, // procsignal_sigusr1_handler

    SIGFPE, // FloatExceptionHandler

    SIGTERM, // bgworker_die

    SIGQUIT // bgworker_quickdie

};

Result<sigset_t> ThreadYsqlSignalMaskBlock() {

  return ThreadSignalMaskBlock(kYsqlHandledSignals);

}

} // namespace yb