/* MiniDLNA media server * Copyright (C) 2008-2009 Justin Maggard * * This program 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 2 of the License, or * (at your option) any later version. * * This program 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 this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include #include #include #include #include #include #include #include #include #include #include #include "upnpglobalvars.h" #include "metadata.h" #include "utils.h" #include "sql.h" #include "scanner.h" #include "log.h" struct virtual_item { int objectID; char parentID[64]; char name[256]; }; int is_video(const char * file) { return (ends_with(file, ".mpg") || ends_with(file, ".mpeg") || ends_with(file, ".avi") || ends_with(file, ".divx") || ends_with(file, ".asf") || ends_with(file, ".wmv") || ends_with(file, ".mp4") || ends_with(file, ".m4v") || ends_with(file, ".mts") || ends_with(file, ".m2ts") || ends_with(file, ".m2t") || ends_with(file, ".mkv") || ends_with(file, ".vob") || ends_with(file, ".ts") || #ifdef TIVO_SUPPORT ends_with(file, ".TiVo") || #endif ends_with(file, ".flv") || ends_with(file, ".xvid")); } int is_audio(const char * file) { return (ends_with(file, ".mp3") || ends_with(file, ".flac") || ends_with(file, ".wma") || ends_with(file, ".asf") || ends_with(file, ".fla") || ends_with(file, ".flc") || ends_with(file, ".m4a") || ends_with(file, ".aac") || ends_with(file, ".mp4") || ends_with(file, ".m4p")); } int is_image(const char * file) { return (ends_with(file, ".jpg") || ends_with(file, ".jpeg")); } sqlite_int64 get_next_available_id(const char * table, const char * parentID) { char * sql; char **result; int ret, rows; sqlite_int64 objectID = 0; asprintf(&sql, "SELECT OBJECT_ID, max(ID) from %s where PARENT_ID = '%s'", table, parentID); ret = sql_get_table(db, sql, &result, &rows, NULL); if( result[2] ) { objectID = strtoll(rindex(result[2], '$')+1, NULL, 16) + 1; } sqlite3_free_table(result); free(sql); return objectID; } long long int insert_container(const char * item, const char * rootParent, const char * refID, const char *class, const char *artist, const char *genre, const char *album_art, const char *art_dlna_pn) { char **result; char **result2; char *sql; int cols, rows, ret; int parentID = 0, objectID = 0; sqlite_int64 detailID = 0; sql = sqlite3_mprintf("SELECT OBJECT_ID from OBJECTS where OBJECT_ID glob '%s$*' and NAME = '%q'", rootParent, item); ret = sql_get_table(db, sql, &result, &rows, &cols); sqlite3_free(sql); if( cols ) { parentID = strtol(rindex(result[1], '$')+1, NULL, 16); objectID = get_next_available_id("OBJECTS", result[1]); } else { parentID = get_next_available_id("OBJECTS", rootParent); if( refID ) { sql = sqlite3_mprintf("SELECT DETAIL_ID from OBJECTS where OBJECT_ID = %Q", refID); ret = sql_get_table(db, sql, &result2, &rows, NULL); if( ret == SQLITE_OK ) { if( rows ) { detailID = strtoll(result2[1], NULL, 10); } sqlite3_free_table(result2); } sqlite3_free(sql); } if( !detailID ) { detailID = GetFolderMetadata(item, NULL, artist, genre, album_art, art_dlna_pn); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, DETAIL_ID, CLASS, NAME) " "VALUES" " ('%s$%X', '%s', %Q, %lld, 'container.%s', '%q')", rootParent, parentID, rootParent, refID, detailID, class, item); ret = sql_exec(db, sql); sqlite3_free(sql); } sqlite3_free_table(result); return (long long)parentID<<32|objectID; } void insert_containers(const char * name, const char *path, const char * refID, const char * class, long unsigned int detailID) { char *sql; char **result; int ret; int cols, row; sqlite_int64 container; if( strstr(class, "imageItem") ) { char *date = NULL, *cam = NULL; char date_taken[13], camera[64]; static struct virtual_item last_date; static struct virtual_item last_cam; static struct virtual_item last_camdate; static sqlite_int64 last_all_objectID = 0; asprintf(&sql, "SELECT DATE, CREATOR from DETAILS where ID = %lu", detailID); ret = sql_get_table(db, sql, &result, &row, &cols); free(sql); if( ret == SQLITE_OK ) { date = result[2]; cam = result[3]; } if( date ) { strncpy(date_taken, date, 10); date_taken[10] = '\0'; } else { strcpy(date_taken, "Unknown Date"); } if( strcmp(last_date.name, date_taken) == 0 ) { last_date.objectID++; //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Using last date item: %s/%s/%X\n", last_date.name, last_date.parentID, last_date.objectID); } else { container = insert_container(date_taken, "3$12", NULL, "album.photoAlbum", NULL, NULL, NULL, NULL); sprintf(last_date.parentID, "3$12$%llX", container>>32); last_date.objectID = (int)container; strcpy(last_date.name, date_taken); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Creating cached date item: %s/%s/%X\n", last_date.name, last_date.parentID, last_date.objectID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_date.parentID, last_date.objectID, last_date.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); if( cam ) { strncpy(camera, cam, 63); camera[63] = '\0'; } else { strcpy(camera, "Unknown Camera"); } if( strcmp(camera, last_cam.name) != 0 ) { container = insert_container(camera, "3$13", NULL, "storageFolder", NULL, NULL, NULL, NULL); sprintf(last_cam.parentID, "3$13$%llX", container>>32); strncpy(last_cam.name, camera, 255); last_camdate.name[0] = '\0'; } if( strcmp(last_camdate.name, date_taken) == 0 ) { last_camdate.objectID++; //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Using last camdate item: %s/%s/%s/%X\n", camera, last_camdate.name, last_camdate.parentID, last_camdate.objectID); } else { container = insert_container(date_taken, last_cam.parentID, NULL, "album.photoAlbum", NULL, NULL, NULL, NULL); sprintf(last_camdate.parentID, "%s$%llX", last_cam.parentID, container>>32); last_camdate.objectID = (int)container; strcpy(last_camdate.name, date_taken); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Creating cached camdate item: %s/%s/%s/%X\n", camera, last_camdate.name, last_camdate.parentID, last_camdate.objectID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_camdate.parentID, last_camdate.objectID, last_camdate.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); /* All Images */ if( !last_all_objectID ) { last_all_objectID = get_next_available_id("OBJECTS", "3$11"); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('3$11$%llX', '3$11', '%s', '%s', %lu, %Q)", last_all_objectID++, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); } else if( strstr(class, "audioItem") ) { asprintf(&sql, "SELECT ARTIST, ALBUM, GENRE, ALBUM_ART, ART_DLNA_PN from DETAILS where ID = %lu", detailID); ret = sql_get_table(db, sql, &result, &row, &cols); free(sql); if( ret != SQLITE_OK ) return; if( !row ) { sqlite3_free_table(result); return; } char *artist = result[5], *album = result[6], *genre = result[7]; char *album_art = result[8], *art_dlna_pn = result[9]; static struct virtual_item last_album; static struct virtual_item last_artist; static struct virtual_item last_artistAlbum; static struct virtual_item last_artistAlbumAll; static struct virtual_item last_genre; static struct virtual_item last_genreArtist; static struct virtual_item last_genreArtistAll; static sqlite_int64 last_all_objectID = 0; if( album ) { if( strcmp(album, last_album.name) == 0 ) { last_album.objectID++; //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Using last album item: %s/%s/%X\n", last_album.name, last_album.parentID, last_album.objectID); } else { strcpy(last_album.name, album); container = insert_container(album, "1$7", NULL, "album.musicAlbum", artist, genre, album_art, art_dlna_pn); sprintf(last_album.parentID, "1$7$%llX", container>>32); last_album.objectID = (int)container; //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Creating cached album item: %s/%s/%X\n", last_album.name, last_album.parentID, last_album.objectID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_album.parentID, last_album.objectID, last_album.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); } if( artist ) { if( strcmp(artist, last_artist.name) != 0 ) { container = insert_container(artist, "1$6", NULL, "person.musicArtist", NULL, genre, NULL, NULL); sprintf(last_artist.parentID, "1$6$%llX", container>>32); strcpy(last_artist.name, artist); last_artistAlbum.name[0] = '\0'; /* Add this file to the "- All Albums -" container as well */ container = insert_container("- All Albums -", last_artist.parentID, NULL, "storageFolder", artist, genre, NULL, NULL); sprintf(last_artistAlbumAll.parentID, "%s$%llX", last_artist.parentID, container>>32); last_artistAlbumAll.objectID = (int)container; } else { last_artistAlbumAll.objectID++; } if( strcmp(album?album:"Unknown Album", last_artistAlbum.name) == 0 ) { last_artistAlbum.objectID++; //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Using last artist/album item: %s/%s/%X\n", last_artist.name, last_artist.parentID, last_artist.objectID); } else { container = insert_container(album?album:"Unknown Album", last_artist.parentID, album?last_album.parentID:NULL, "album.musicAlbum", artist, genre, album_art, art_dlna_pn); sprintf(last_artistAlbum.parentID, "%s$%llX", last_artist.parentID, container>>32); last_artistAlbum.objectID = (int)container; strcpy(last_artistAlbum.name, album?album:"Unknown Album"); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Creating cached artist/album item: %s/%s/%X\n", last_artist.name, last_artist.parentID, last_artist.objectID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_artistAlbum.parentID, last_artistAlbum.objectID, last_artistAlbum.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_artistAlbumAll.parentID, last_artistAlbumAll.objectID, last_artistAlbumAll.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); } if( genre ) { if( strcmp(genre, last_genre.name) != 0 ) { container = insert_container(genre, "1$5", NULL, "genre.musicGenre", NULL, NULL, NULL, NULL); sprintf(last_genre.parentID, "1$5$%llX", container>>32); strcpy(last_genre.name, genre); last_genreArtist.name[0] = '\0'; /* Add this file to the "- All Artists -" container as well */ container = insert_container("- All Artists -", last_genre.parentID, NULL, "storageFolder", NULL, genre, NULL, NULL); sprintf(last_genreArtistAll.parentID, "%s$%llX", last_genre.parentID, container>>32); last_genreArtistAll.objectID = (int)container; } else { last_genreArtistAll.objectID++; } if( strcmp(artist?artist:"Unknown Artist", last_genreArtist.name) == 0 ) { last_genreArtist.objectID++; } else { container = insert_container(artist?artist:"Unknown Artist", last_genre.parentID, artist?last_artist.parentID:NULL, "person.musicArtist", NULL, genre, NULL, NULL); sprintf(last_genreArtist.parentID, "%s$%llX", last_genre.parentID, container>>32); last_genreArtist.objectID = (int)container; strcpy(last_genreArtist.name, artist?artist:"Unknown Artist"); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Creating cached artist/artist item: %s/%s/%X\n", last_artist.name, last_artist.parentID, last_artist.objectID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_genreArtist.parentID, last_genreArtist.objectID, last_genreArtist.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s$%X', '%s', '%s', '%s', %lu, %Q)", last_genreArtistAll.parentID, last_genreArtistAll.objectID, last_genreArtistAll.parentID, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); } /* All Music */ if( !last_all_objectID ) { last_all_objectID = get_next_available_id("OBJECTS", "1$4"); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('1$4$%llX', '1$4', '%s', '%s', %lu, %Q)", last_all_objectID++, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); } else if( strstr(class, "videoItem") ) { static sqlite_int64 last_all_objectID = 0; /* All Videos */ if( !last_all_objectID ) { last_all_objectID = get_next_available_id("OBJECTS", "2$8"); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('2$8$%llX', '2$8', '%s', '%s', %lu, %Q)", last_all_objectID++, refID, class, detailID, name); sql_exec(db, sql); sqlite3_free(sql); return; } else { return; } sqlite3_free_table(result); } int insert_directory(const char * name, const char * path, const char * base, const char * parentID, int objectID) { char * sql; int ret, rows, found = 0; sqlite_int64 detailID = 0; char * refID = NULL; char class[] = "container.storageFolder"; char * id_buf = NULL; char * parent_buf = NULL; char **result; char *dir = NULL; static char last_found[256] = "-1"; if( strcmp(base, BROWSEDIR_ID) != 0 ) asprintf(&refID, "%s%s$%X", BROWSEDIR_ID, parentID, objectID); if( refID ) { dir = strdup(path); dir = dirname(dir); asprintf(&id_buf, "%s%s$%X", base, parentID, objectID); asprintf(&parent_buf, "%s%s", base, parentID); while( !found ) { if( strcmp(id_buf, last_found) == 0 ) break; sql = sqlite3_mprintf("SELECT count(OBJECT_ID) from OBJECTS where OBJECT_ID = '%s'", id_buf); if( (sql_get_table(db, sql, &result, NULL, NULL) == SQLITE_OK) && atoi(result[1]) ) { sqlite3_free_table(result); sqlite3_free(sql); strcpy(last_found, id_buf); break; } sqlite3_free_table(result); sqlite3_free(sql); /* Does not exist. Need to create, and may need to create parents also */ sql = sqlite3_mprintf("SELECT DETAIL_ID from OBJECTS where OBJECT_ID = '%s'", refID); if( (sql_get_table(db, sql, &result, &rows, NULL) == SQLITE_OK) && rows ) { detailID = atoi(result[1]); } sqlite3_free_table(result); sqlite3_free(sql); sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, DETAIL_ID, CLASS, NAME) " "VALUES" " ('%s', '%s', %Q, '%lld', '%s', '%q')", id_buf, parent_buf, refID, detailID, class, rindex(dir, '/')+1); sql_exec(db, sql); sqlite3_free(sql); if( rindex(id_buf, '$') ) *rindex(id_buf, '$') = '\0'; if( rindex(parent_buf, '$') ) *rindex(parent_buf, '$') = '\0'; if( rindex(refID, '$') ) *rindex(refID, '$') = '\0'; dir = dirname(dir); } free(refID); free(parent_buf); free(id_buf); free(dir); return 1; } detailID = GetFolderMetadata(name, path, NULL, NULL, NULL, NULL); sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, DETAIL_ID, CLASS, NAME) " "VALUES" " ('%s%s$%X', '%s%s', %Q, '%lld', '%s', '%q')", base, parentID, objectID, base, parentID, refID, detailID, class, name); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "SQL: %s\n", sql); ret = sql_exec(db, sql); sqlite3_free(sql); if( refID ) free(refID); return -1; } int insert_file(char * name, const char * path, const char * parentID, int object) { char *sql; char class[32]; char objectID[64]; unsigned long int detailID = 0; char base[8]; char * typedir_parentID; int typedir_objectID; char * baseid; char * orig_name = NULL; if( is_image(name) ) { strcpy(base, IMAGE_DIR_ID); strcpy(class, "item.imageItem.photo"); detailID = GetImageMetadata(path, name); } else if( is_video(name) ) { orig_name = strdup(name); strcpy(base, VIDEO_DIR_ID); strcpy(class, "item.videoItem"); detailID = GetVideoMetadata(path, name); if( !detailID ) strcpy(name, orig_name); } if( !detailID && is_audio(name) ) { strcpy(base, MUSIC_DIR_ID); strcpy(class, "item.audioItem.musicTrack"); detailID = GetAudioMetadata(path, name); } if( orig_name ) free(orig_name); if( !detailID ) { DPRINTF(E_WARN, L_SCANNER, "Unsuccessful getting details for %s!\n", path); return -1; } sprintf(objectID, "%s%s$%X", BROWSEDIR_ID, parentID, object); sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s', '%s%s', '%s', %lu, '%q')", objectID, BROWSEDIR_ID, parentID, class, detailID, name); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "SQL: %s\n", sql); sql_exec(db, sql); sqlite3_free(sql); if( *parentID ) { typedir_objectID = 0; typedir_parentID = strdup(parentID); baseid = rindex(typedir_parentID, '$'); if( baseid ) { sscanf(baseid+1, "%X", &typedir_objectID); *baseid = '\0'; } insert_directory(name, path, base, typedir_parentID, typedir_objectID); free(typedir_parentID); } sql = sqlite3_mprintf( "INSERT into OBJECTS" " (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, NAME) " "VALUES" " ('%s%s$%X', '%s%s', '%s', '%s', %lu, '%q')", base, parentID, object, base, parentID, objectID, class, detailID, name); //DEBUG DPRINTF(E_DEBUG, L_SCANNER, "SQL: %s\n", sql); sql_exec(db, sql); sqlite3_free(sql); insert_containers(name, path, objectID, class, detailID); return 0; } int CreateDatabase(void) { int ret, i; char sql_buf[512]; const char * containers[] = { "0","-1", "root", "1", "0", "Music", "1$4", "1", "All Music", "1$5", "1", "Genre", "1$6", "1", "Artist", "1$7", "1", "Album", "1$14", "1", "Folders", "2", "0", "Video", "2$8", "2", "All Video", "2$15", "2", "Folders", "3", "0", "Pictures", "3$11", "3", "All Pictures", "3$12", "3", "Date Taken", "3$13", "3", "Camera", "3$16", "3", "Folders", "64", "0", "Browse Folders", 0 }; sql_exec(db, "pragma page_size = 4096"); sql_exec(db, "pragma journal_mode = OFF"); sql_exec(db, "pragma synchronous = OFF;"); sql_exec(db, "pragma default_cache_size = 8192;"); //JM: Set up a db version number, so we know if we need to rebuild due to a new structure. sprintf(sql_buf, "pragma user_version = %d;", DB_VERSION); sql_exec(db, sql_buf); ret = sql_exec(db, "CREATE TABLE OBJECTS ( " "ID INTEGER PRIMARY KEY AUTOINCREMENT, " "OBJECT_ID TEXT NOT NULL, " "PARENT_ID TEXT NOT NULL, " "REF_ID TEXT DEFAULT NULL, " "CLASS TEXT NOT NULL, " "DETAIL_ID INTEGER DEFAULT NULL, " "NAME TEXT DEFAULT NULL" ");"); if( ret != SQLITE_OK ) goto sql_failed; ret = sql_exec(db, "CREATE TABLE DETAILS ( " "ID INTEGER PRIMARY KEY AUTOINCREMENT, " "PATH TEXT DEFAULT NULL, " "SIZE INTEGER, " "TITLE TEXT COLLATE NOCASE, " "DURATION TEXT, " "BITRATE INTEGER, " "SAMPLERATE INTEGER, " "ARTIST TEXT COLLATE NOCASE, " "ALBUM TEXT COLLATE NOCASE, " "GENRE TEXT COLLATE NOCASE, " "COMMENT TEXT, " "CHANNELS INTEGER, " "TRACK INTEGER, " "DATE DATE, " "RESOLUTION TEXT, " "THUMBNAIL BOOL DEFAULT 0, " "CREATOR TEXT COLLATE NOCASE, " "DLNA_PN TEXT, " "MIME TEXT, " "ALBUM_ART INTEGER DEFAULT 0, " "ART_DLNA_PN TEXT DEFAULT NULL" ")"); if( ret != SQLITE_OK ) goto sql_failed; ret = sql_exec(db, "CREATE TABLE ALBUM_ART ( " "ID INTEGER PRIMARY KEY AUTOINCREMENT, " "PATH TEXT NOT NULL" ")"); if( ret != SQLITE_OK ) goto sql_failed; ret = sql_exec(db, "CREATE TABLE SETTINGS (" "UPDATE_ID INTEGER PRIMARY KEY DEFAULT 0" ")"); if( ret != SQLITE_OK ) goto sql_failed; ret = sql_exec(db, "INSERT into SETTINGS values (0)"); if( ret != SQLITE_OK ) goto sql_failed; for( i=0; containers[i]; i=i+3 ) { sprintf(sql_buf, "INSERT into OBJECTS (OBJECT_ID, PARENT_ID, DETAIL_ID, CLASS, NAME)" " values " "('%s', '%s', %lld, 'container.storageFolder', '%s')", containers[i], containers[i+1], GetFolderMetadata(containers[i+2], NULL, NULL, NULL, NULL, NULL), containers[i+2]); ret = sql_exec(db, sql_buf); if( ret != SQLITE_OK ) goto sql_failed; } sql_exec(db, "create INDEX IDX_OBJECTS_OBJECT_ID ON OBJECTS(OBJECT_ID);"); sql_exec(db, "create INDEX IDX_OBJECTS_PARENT_ID ON OBJECTS(PARENT_ID);"); sql_exec(db, "create INDEX IDX_OBJECTS_DETAIL_ID ON OBJECTS(DETAIL_ID);"); sql_exec(db, "create INDEX IDX_OBJECTS_CLASS ON OBJECTS(CLASS);"); sql_exec(db, "create INDEX IDX_DETAILS_PATH ON DETAILS(PATH);"); sql_exec(db, "create INDEX IDX_DETAILS_ID ON DETAILS(ID);"); sql_exec(db, "create INDEX IDX_ALBUM_ART ON ALBUM_ART(ID);"); sql_failed: if( ret != SQLITE_OK ) fprintf(stderr, "Error creating SQLite3 database!\n"); return (ret != SQLITE_OK); } int filter_audio(const struct dirent *d) { return ( (*d->d_name != '.') && ((d->d_type == DT_DIR) || ((d->d_type == DT_REG) && is_audio(d->d_name) ) ) ); } int filter_video(const struct dirent *d) { return ( (*d->d_name != '.') && ((d->d_type == DT_DIR) || ((d->d_type == DT_REG) && is_video(d->d_name) ) ) ); } int filter_images(const struct dirent *d) { return ( (*d->d_name != '.') && ((d->d_type == DT_DIR) || ((d->d_type == DT_REG) && is_image(d->d_name) ) ) ); } int filter_media(const struct dirent *d) { return ( (*d->d_name != '.') && ((d->d_type == DT_DIR) || ((d->d_type == DT_REG) && (is_image(d->d_name) || is_audio(d->d_name) || is_video(d->d_name) ) )) ); } void ScanDirectory(const char * dir, const char * parent, enum media_types type) { struct dirent **namelist; int i, n, startID=0; char parent_id[PATH_MAX]; char full_path[PATH_MAX]; char * name = NULL; static long long unsigned int fileno = 0; setlocale(LC_COLLATE, ""); if( chdir(dir) != 0 ) return; DPRINTF(parent?E_INFO:E_WARN, L_SCANNER, "Scanning %s\n", dir); switch( type ) { case ALL_MEDIA: n = scandir(".", &namelist, filter_media, alphasort); break; case AUDIO_ONLY: n = scandir(".", &namelist, filter_audio, alphasort); break; case VIDEO_ONLY: n = scandir(".", &namelist, filter_video, alphasort); break; case IMAGES_ONLY: n = scandir(".", &namelist, filter_images, alphasort); break; default: n = -1; break; } if (n < 0) { fprintf(stderr, "Error scanning %s [scandir]\n", dir); return; } if( !parent ) { startID = get_next_available_id("OBJECTS", BROWSEDIR_ID); #ifdef READYNAS FILE * flag = fopen("/ramfs/.upnp-av_scan", "w"); if( flag ) fclose(flag); #endif } for (i=0; i < n; i++) { sprintf(full_path, "%s/%s", dir, namelist[i]->d_name); name = escape_tag(namelist[i]->d_name); if( namelist[i]->d_type == DT_DIR ) { insert_directory(name?name:namelist[i]->d_name, full_path, BROWSEDIR_ID, (parent ? parent:""), i+startID); sprintf(parent_id, "%s$%X", (parent ? parent:""), i+startID); ScanDirectory(full_path, parent_id, type); } else { if( insert_file(name?name:namelist[i]->d_name, full_path, (parent ? parent:""), i+startID) == 0 ) fileno++; } if( name ) free(name); free(namelist[i]); } free(namelist); if( parent ) { chdir(dirname((char*)dir)); } else { #ifdef READYNAS if( access("/ramfs/.rescan_done", F_OK) == 0 ) system("/bin/sh /ramfs/.rescan_done"); unlink("/ramfs/.upnp-av_scan"); #endif DPRINTF(E_WARN, L_SCANNER, "Scanning %s finished (%llu files)!\n", dir, fileno); } } void * start_scanner() { struct media_dir_s * media_path = media_dirs; if (setpriority(PRIO_PROCESS, 0, 15) == -1) DPRINTF(E_WARN, L_INOTIFY, "Failed to reduce scanner thread priority\n"); scanning = 1; freopen("/dev/null", "a", stderr); while( media_path ) { ScanDirectory(media_path->path, NULL, media_path->type); media_path = media_path->next; } freopen("/proc/self/fd/2", "a", stderr); scanning = 0; return 0; }