C / C++ / MFC

<br />
long GetRemoveableDriveCount()<br />
{<br />
	// Count the units flagged by Windows as REMOVEABLE<br />
	// Accept all USB-connected drives too, even USB-connected har d-drives<br />
	// (REMOVEABLE means it's a unit who's got a removeable media. An<br />
	// USB-harddrive doesn't have such a media. But the unit itself is removeable, so it's considdered OK.<br />
	// Find them too<br />
<br />
	int i;<br />
	char sDrivePath[3];<br />
	UINT uDriveType;<br />
	DWORD dwDrives;<br />
	fRemovableDrives = 0;<br />
	int noOfDrives = 0;<br />
	<br />
	dwDrives = GetLogicalDrives();<br />
<br />
	strcpy(sDrivePath+1, ":\\");<br />
<br />
	// Loopa through all drives in the system<br />
	for(i = 0; i < 32; i++) {<br />
		if(dwDrives & (1 << i)) {<br />
			// Create sDrivePath<br />
			sDrivePath[0] = (char) ('A' + i);<br />
			uDriveType = GetDriveType(sDrivePath);<br />
<br />
			if((uDriveType == DRIVE_REMOVABLE || uDriveType== DRIVE_RAMDISK)) {<br />
				// It's a removable or a RAM-DISK, add it to the list and count it.<br />
				fRemovableDrives |= (1 << i);<br />
				noOfDrives++;<br />
			}<br />
			else {<br />
				if(uDriveType == DRIVE_FIXED) {<br />
					// Accept DRIVE_FIXED if it's a USB-unit<br />
					// DRIVE_FIXED The disk cannot be removed from the drive. <br />
					char szBuf[10];<br />
					HANDLE hDevice;<br />
					PSTORAGE_DEVICE_DESCRIPTOR pDevDesc;<br />
<br />
					sprintf(szBuf, "\\\\?\\%c:", sDrivePath[0]);<br />
//					hDevice = CreateFile(szBuf, GENERIC_READ,<br />
//							FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL);<br />
					// Open the unit with "Zero rights". Is supposed to work at all times<br />
					hDevice = CreateFile(szBuf, 0,<br />
							FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL);<br />
<br />
					if (hDevice != INVALID_HANDLE_VALUE)<br />
					{<br />
						pDevDesc = (PSTORAGE_DEVICE_DESCRIPTOR)new BYTE[sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1];<br />
<br />
						pDevDesc->Size = sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1;<br />
<br />
						if(GetDisksProperty(hDevice, pDevDesc))<br />
						{<br />
							if(pDevDesc->BusType == BusTypeUsb)<br />
							{<br />
<br />
	// It's a USB conneted hard drive. Add it to the list and count it.							<br />
                                                               fRemovableDrives |= (1 << i);<br />
								noOfDrives++;<br />
							}<br />
						}<br />
<br />
						delete pDevDesc;<br />
						CloseHandle(hDevice);<br />
					}<br />
				}<br />
			}<br />
		}<br />
	}<br />
<br />
	return noOfDrives;<br />
}<br />
<br />

kiranin wrote:

How can i detect which drive is alloted to USB drive and how can i know USB drive is inserted or not?

Mridang Agarwal wrote:

It there another library i could use for socket programming.

#include <sys/socket.h>       /*  socket definitions        */
#include <sys/types.h>        /*  socket types              */
#include <arpa/inet.h>        /*  inet (3) funtions         */
#include "helper.h"           /*  our own helper functions  */
#include  <stdlib.h>          /* header file */
#include  <stdio.h>           /* header file */


/*  Global constants  */

#define ECHO_PORT          (2002)
#define MAX_LINE           (1000)


int main(int argc, char *argv[]) {
    int       list_s;                /*  listening socket          */
    int       conn_s;                /*  connection socket         */
    short int port;                  /*  port number               */
    struct    sockaddr_in servaddr;  /*  socket address structure  */
    char      buffer[MAX_LINE];      /*  character buffer          */
    char     *endptr;                /*  for strtol()              */


    /*  Get port number from the command line, and
        set to default port if no arguments were supplied  */

    if ( argc == 2 ) {
	port = strtol(argv[1], &endptr, 0);
	if ( *endptr ) {
	    fprintf(stderr, "ECHOSERV: Invalid port number.\n");
	    exit(EXIT_FAILURE);
	}
    }
    else if ( argc < 2 ) {
	port = ECHO_PORT;
    }
    else {
	fprintf(stderr, "ECHOSERV: Invalid arguments.\n");
	exit(EXIT_FAILURE);
    }


    /*  Create the listening socket  */

    if ( (list_s = socket(AF_INET, SOCK_STREAM, 0)) < 0 ) {
	fprintf(stderr, "ECHOSERV: Error creating listening socket.\n");
	exit(EXIT_FAILURE);
    }


    /*  Set all bytes in socket address structure to
        zero, and fill in the relevant data members   */

    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family      = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port        = htons(port);


    /*  Bind our socket addresss to the
	listening socket, and call listen()  */

    if ( bind(list_s, (struct sockaddr *) &servaddr, sizeof(servaddr)) < 0 ) {
	fprintf(stderr, "ECHOSERV: Error calling bind()\n");
	exit(EXIT_FAILURE);
    }

    if ( listen(list_s, LISTENQ) < 0 ) {
	fprintf(stderr, "ECHOSERV: Error calling listen()\n");
	exit(EXIT_FAILURE);
    }


    /*  Enter an infinite loop to respond
        to client requests and echo input  */

    while ( 1 ) {

	/*  Wait for a connection, then accept() it  */

	if ( (conn_s = accept(list_s, NULL, NULL) ) < 0 ) {
	    fprintf(stderr, "ECHOSERV: Error calling accept()\n");
	    exit(EXIT_FAILURE);
	}


	/*  Retrieve an input line from the connected socket
	    then simply write it back to the same socket.     */

	Readline(conn_s, buffer, MAX_LINE-1);
	Writeline(conn_s, buffer, strlen(buffer));


	/*  Close the connected socket  */

	if ( close(conn_s) < 0 ) {
	    fprintf(stderr, "ECHOSERV: Error calling close()\n");
	    exit(EXIT_FAILURE);
	}
    }
}

Mridang Agarwal wrote:

#include <sys socket.h=""> /* socket definitions */#include <sys types.h=""> /* socket types */#include <arpa inet.h="">

---

<marquee scrollamount="1" scrolldelay="1" direction="up" height="10" step="1">--[ ]--

[My Current Status]

Link2006 wrote:Let's take it outside of CP
Jeremy : Please don't.I would love to see this.I'm making the popcorn already.