13e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes/* 23e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * Copyright (C) 2008 The Android Open Source Project 33e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * All rights reserved. 43e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * 53e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * Redistribution and use in source and binary forms, with or without 63e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * modification, are permitted provided that the following conditions 73e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * are met: 83e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * * Redistributions of source code must retain the above copyright 93e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * notice, this list of conditions and the following disclaimer. 103e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * * Redistributions in binary form must reproduce the above copyright 113e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * notice, this list of conditions and the following disclaimer in 123e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * the documentation and/or other materials provided with the 133e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * distribution. 143e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * 153e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 163e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 173e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 183e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 193e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 203e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 213e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS 223e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 233e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 243e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 253e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 263e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes * SUCH DAMAGE. 273e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes */ 283e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 293e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <pthread.h> 303e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 313e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <fcntl.h> 323e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <stdio.h> // For snprintf. 3305fc1d7050d5451aea08dc5f504d2670287b2d43Elliott Hughes#include <string.h> 343e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <sys/prctl.h> 353e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <sys/stat.h> 363e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <sys/types.h> 373e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include <unistd.h> 383e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 393e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#include "private/ErrnoRestorer.h" 40673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui#include "pthread_internal.h" 413e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 423e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes// This value is not exported by kernel headers. 433e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes#define MAX_TASK_COMM_LEN 16 4440eabe24e4e3ae8ebe437f1f4e43cf39cbba2e9eElliott Hughes#define TASK_COMM_FMT "/proc/self/task/%d/comm" 453e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 46a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughesint pthread_setname_np(pthread_t t, const char* thread_name) { 473e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes ErrnoRestorer errno_restorer; 483e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 493e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes size_t thread_name_len = strlen(thread_name); 503e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes if (thread_name_len >= MAX_TASK_COMM_LEN) { 513e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes return ERANGE; 523e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes } 533e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 543e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes // Changing our own name is an easy special case. 55a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughes if (t == pthread_self()) { 5640eabe24e4e3ae8ebe437f1f4e43cf39cbba2e9eElliott Hughes return prctl(PR_SET_NAME, thread_name) ? errno : 0; 573e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes } 583e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 59a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughes // We have to change another thread's name. 60673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui pthread_internal_t* thread = __pthread_internal_find(t); 61673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui if (thread == NULL) { 62673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui return ENOENT; 63a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughes } 64673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui pid_t tid = thread->tid; 65673b15e4ee2c6d99b150aedddc0f389e29f98e1bYabin Cui 663e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes char comm_name[sizeof(TASK_COMM_FMT) + 8]; 67a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughes snprintf(comm_name, sizeof(comm_name), TASK_COMM_FMT, tid); 68f73183f1a34df22b62a3d0bbf82e18d5797c9cdeElliott Hughes int fd = open(comm_name, O_CLOEXEC | O_WRONLY); 693e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes if (fd == -1) { 703e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes return errno; 713e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes } 723e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes ssize_t n = TEMP_FAILURE_RETRY(write(fd, thread_name, thread_name_len)); 733e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes close(fd); 743e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes 753e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes if (n < 0) { 763e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes return errno; 77a41ba2f0bfc4fce1ce8f06a9c289102c440c929dElliott Hughes } else if (n != static_cast<ssize_t>(thread_name_len)) { 783e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes return EIO; 793e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes } 803e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes return 0; 813e898476c7230b60a0f76968e64ff25f475b48c0Elliott Hughes} 82