Merge pull request #17 from OSDLabs/dev

Allow using multiple accounts!
Fix #14

Author: utkarshme

bitsnet

@@ -35,31 +35,21 @@ while getopts "u:p:dhUof" o; do
 done
 if [[ "$f" == "1" ]]; then
     debug_msg "Forcing login attempt"
-    force_login
 fi
 
 
-debug_msg "Username: $username"
 debug_msg "LoginURL: $login_url"
 
 
 # Check if already logged in
-debug_msg "Checking if already connected"
-if wget -qT 5 -t 1 --spider --no-check-certificate https://google.com
-then
-    send_msg "Already logged into the BITSnet"
-    exit
-fi
-
-
-# Check if username or password are empty
-debug_msg "Checking if username or password is empty"
-if [[ ("$username" == "") || ("$password" == "") ]]; then
-    send_msg "Username or password empty"
-    debug_msg "Exiting"
-    exit
+if [[ "$f" != "1" ]]; then
+    debug_msg "Checking if already connected"
+    if wget -qT 5 -t 1 --spider --no-check-certificate https://google.com
+    then
+        send_msg "Already logged into the BITSnet"
+        exit
+    fi
 fi
-debug_msg "Not empty"
 
 
 # Check if using wifi or ethernet
@@ -69,7 +59,7 @@ dev=$(get_device)
 debug_msg "dev = $dev"
 
 # Exit if dev = 0
-if [[ "$dev" == "0" ]]; then
+if [[ "$f" == "0" && "$dev" == "0" ]]; then
     debug_msg "Exiting because dev is zero"
     send_msg "Network interface unknown."
     exit
@@ -85,14 +75,26 @@ if [[ "$dev" == "2" && "$(hostname -I)" != "10.20"* ]]; then
     debug_msg "Connected to wifi. Sending request to cisco router form"
     router_login
 fi
+
 # Log into the ldap
 debug_msg "Logging into the LDAP"
-reply=$(ldap_login)
+i=1
+while [[ ! -z ${username[$i]} ]]
+do
+    debug_msg "Trying with ID: ${username[$i]}"
+    reply=$(ldap_login ${username[$i]} ${password[$i]})
+
+    # Extract reply
+    reply=$(extract_msg $reply)
+
+    if [[ "$reply" == "You have successfully logged in" ]]; then
+        break
+    else
+        debug_msg "$reply"
+    fi
+    ((i++))
 
-
-# Extract reply
-debug_msg "Extracting reply"
-reply=$(extract_msg $reply)
+done
 
 
 # Display reply or send a notification

bitsnetrc

@@ -1,4 +1,28 @@
-# Set defaults here
-username='f2015022'
-password='myPass'
+#
+# This file is for the user configuration
+# Set some username and passwords according to you priority
+#
+# eg: if you have 2 IDs with 'ID1' being the most used,
+# then write this:
+#
+# username[1]='ID1'
+# password[1]='ID1_password'
+# username[2]='ID2'
+# password[2]='ID2_password'
+#
+# Use can use UNLIMITED ID and password combinations
+
+username[1]='f2015022'
+password[1]='myPass'
+
+username[2]='f2015023'
+password[2]='myPass2'
+
+#
+# Configure features
+
+# Always force login:
+# change to '1' to always use autologin
+f=0
+
 login_url='https://10.1.0.10:8090/httpclient.html'

install

@@ -1,41 +1,49 @@
 #!/bin/sh
 
+echo "Installing BITSnet Login"
+
 # Check if wget installed
+echo "Checking if wget installed"
 if ! which wget > /dev/null ; then
     echo "wget not installed"
     exit
 fi
 # Check if git installed for update
+echo "Checking if git installed"
 if ! which git > /dev/null ; then
     echo "git not installed"
     exit
 fi
 # Check if nmcli installed
+echo "Checking if nmcli installed"
 if ! which nmcli > /dev/null ; then
     echo "nmcli not installed"
     exit
 fi
 
 # Copy config file to the right place
-if ! ls -a ~ | grep .bitsnetrc > /dev/null ; then
+if ! ls -a ~ | grep -x ".bitsnetrc" --quiet ; then
+    echo "Copying bitsnetrc"
     cp ./bitsnetrc ~/.bitsnetrc
 fi
 
 # Copy the script to executables
 if [ ! -d ~/.local/lib/bitsnet ]; then
     mkdir ~/.local/lib/bitsnet
 fi
+echo "Copying script"
 cp ./util.sh ~/.local/lib/bitsnet/util.sh
 cp ./bitsnet ~/.local/bin/bitsnet
 
 # Create a desktop file
+echo "Creating app launcher"
 cp ./bitsnet.desktop ~/.local/share/applications/bitsnet.desktop
 
 # Set to autostart
+echo "Setting up autologin"
 if [ ! -d ~/.config/autostart ]; then
     mkdir -p ~/.config/autostart
 fi
 ln -sf ~/.local/share/applications/bitsnet.desktop ~/.config/autostart/bitsnet.desktop
 
-# Create a man page
-# To Be Done
+# TODO: Create a man page

util.sh

@@ -99,14 +99,6 @@ function router_login() {
 }
 
 function ldap_login() {
-    reply=$(wget -qO- --no-check-certificate --post-data="mode=191&username=$username&password=$password" $login_url)
+    reply=$(wget -qO- --no-check-certificate --post-data="mode=191&username=$1&password=$2" $login_url)
     echo $reply
 }
-
-function force_login() {
-    router_login
-    reply=$(ldap_login)
-    reply=$(extract_msg $reply)
-    send_msg "$reply"
-    exit
-}