build.rs

// smbclient-sys (rust bindings for libsmbclient from Samba project)
// Copyright (c) 2016 Konstantin Gribov
//
// This file is part of smbclient-sys.
//
// smbclient-sys is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// smbclient-sys is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with smbclient-sys.  If not, see <http://www.gnu.org/licenses/>.

extern crate bindgen;
extern crate pkg_config;

use std::env;
use std::path::PathBuf;

fn main() {
    let inc_path_name = "SMBCLIENT_INCLUDE_PATH";
    let lib_path_name = "SMBCLIENT_LIBRARY_PATH";
    let header = match (env::var(inc_path_name), env::var(lib_path_name)) {
        (Ok(inc_path), Ok(lib_path)) => {
            println!("cargo:rustc-link-lib=smbclient");
            println!("cargo:rustc-link-search=native={}", lib_path);
            let mut path = PathBuf::from(inc_path);
            path.push("libsmbclient.h");
            path.to_str().unwrap().to_string()
        }
        (Ok(_), Err(_)) | (Err(_), Ok(_)) =>
            panic!("Either both {} and {} should be set or none of them", inc_path_name, lib_path_name),
        (Err(_), Err(_)) => {
            let lib = pkg_config::Config::new()
                .probe("smbclient")
                .expect("libsmbclient not found");
            find_header(&lib).unwrap().to_str().unwrap().to_string()
        }
    };

    let bindings = bindgen::Builder::default()
        .rust_target(bindgen::RustTarget::Stable_1_25)
        .header(header)
        .ctypes_prefix("::libc")
        .whitelist_type("(smbc_.*)|(SMBC.*)")
        .whitelist_function("smbc_.*")
        .whitelist_var("SMBC?.*")
        .generate()
        .expect("failed to generate bindings for libsmbclient");

    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
    bindings.write_to_file(out_dir.join("libsmbclient.rs"))
            .expect("failed to write libsmbclient.rs");
}

fn find_header(lib: &pkg_config::Library) -> Option<PathBuf> {
    lib.include_paths.iter()
                     .map(|p| p.join("libsmbclient.h"))
                     .find(|p| p.exists() && p.is_file())
}